基于多态的职工管理系统

写在开头-
参考《黑马程序员c++核心编程》

//boss.h
#pragma once
#include <bits/stdc++.h>
#include "worker.h"
using namespace std;
class Boss :public Worker
{
public:
	Boss(int id, string name, int dId);
	virtual void showInfo();
	virtual string getDeptName();
};

//employee.h
#pragma once
#include <bits/stdc++.h>
#include "worker.h"
using namespace std;
//员工类
class Employee :public Worker
{
public:
	//构造函数
	Employee(int id, string name, int dId);
	//显示个人信息
	virtual void showInfo();//重写纯虚函数
	//获取职工岗位名称
	virtual string getDeptName();//重写纯虚函数
};

//manager.h
#pragma once
#include <bits/stdc++.h>
#include "worker.h"
using namespace std;
//经理类
class Manager :public Worker
{
public:
	Manager(int id, string name, int dId);
	//显示个人信息
	virtual void showInfo();
	//获取岗位名称
	virtual string getDeptName();
};

//worker.h
#pragma once
#include <bits/stdc++.h>
using namespace std;
//职工抽象基类
class Worker
{
public:
	//显示个人信息
	virtual void showInfo() = 0;
	//获取岗位名称
	virtual string getDeptName() = 0;
	int m_Id;//职工编号
	string m_Name;//职工姓名
	int m_DeptId;//职工所在部门名称编号
};

//workerManager.h
#pragma once//#pragma once一般由编译器提供保证：同一个文件不会被包含多次
#include <bits/stdc++.h>
#include "worker.h"
#include <fstream>
#define FILENAME "empFile.txt"
using namespace std;
class WorkerManager
{
public:
	WorkerManager();//构造函数
	void Show_Menu();//展示菜单
	void exitSystem();//退出系统
	void Add_Emp();
	void save();//保存文件
	int get_EmpNum();//统计文件中人数
	void init_Emp();//初始化员工
	void Show_Emp();//显示员工
	void Del_Emp();//删除员工
	int IsExist(int id);//判断职工是否存在
	void Mod_Emp();//修改职工
	void Find_Emp();//查找职工
	void Sort_Emp();//排序
	void Clean_File();//清空文件
	~WorkerManager();//析构函数
public:
	int m_EmpNum;//记录文件中的人数个数
	//员工数组的指针
	Worker** m_EmpArray;
	bool m_FileIsEmpty;//判断文件是否为空
};

//boss.cpp
#include <bits/stdc++.h>
#include "boss.h"
Boss::Boss(int id, string name, int dId)
{
	this->m_Id = id;
	this->m_Name = name;
	this->m_DeptId = dId;
}
void Boss::showInfo()
{
	cout << "职工编号： " << this->m_Id
		<< " \t职工姓名： " << this->m_Name
		<< " \t岗位：" << this->getDeptName()
		<< " \t岗位职责：管理公司所有事物" << endl;
}
string Boss::getDeptName()
{
	return string("总裁");
}

//employee.cpp
#include "employee.h"
Employee::Employee(int id, string name, int dId)
{
	this->m_Id = id;
	this->m_Name = name;
	this->m_DeptId = dId;
}
void Employee::showInfo()
{
	cout << "职工编号： " << this->m_Id
		<< " \t职工姓名： " << this->m_Name
		<< " \t岗位：" << this->getDeptName()
		<< " \t岗位职责：完成经理交给的任务" << endl;
}
string Employee::getDeptName()
{
	return string("员工");
}

//manager.cpp
#include "manager.h"
Manager::Manager(int id, string name, int dId)
{
	this->m_Id = id;
	this->m_Name = name;
	this->m_DeptId = dId;
}
void Manager::showInfo()
{
	cout << "职工编号： " << this->m_Id
		<< " \t职工姓名： " << this->m_Name
		<< " \t岗位：" << this->getDeptName()
		<< " \t岗位职责：完成老板交给的任务,并下发任务给员工" << endl;
}
string Manager::getDeptName()
{
	return string("经理");
}

//workerManager.cpp
#include "workerManager.h"
#include "employee.h"
#include "boss.h"
#include "manager.h"
WorkerManager::WorkerManager()
{
	//文件不存在
	ifstream ifs;
	ifs.open(FILENAME, ios::in);//读文件
	if (!ifs.is_open())
	{
		//cout << "文件不存在" << endl;
		this->m_EmpNum = 0;
		this->m_EmpArray = NULL;
		this->m_FileIsEmpty = true;
		ifs.close();
		return;
	}
	//文件存在，数据为空
	char ch;
	ifs >> ch;
	if (ifs.eof())
	{
		//文件为空
		//cout << "文件为空" << endl;
		this->m_EmpNum = 0;
		this->m_EmpArray = NULL;
		this->m_FileIsEmpty = true;
		ifs.close();
		return;
	}
	//当文件存在，并且记录数据
	int num = this->get_EmpNum();
	//cout << "职工的人数为 " << num << endl;
	this->m_EmpNum = num;
	//开辟空间
	this->m_EmpArray = new Worker * [this->m_EmpNum];
	this->init_Emp();
	//测试代码
	//for (int i = 0; i < this->m_EmpNum; i++)
	//{
		//cout << "职工号： " << this->m_EmpArray[i]->m_Id
			//<< " 职工姓名： " << this->m_EmpArray[i]->m_Name
			//<< " 部门编号： " << this->m_EmpArray[i]->m_DeptId << endl;
	//}
}
int WorkerManager::get_EmpNum()//职工数量
{
	ifstream ifs;
	ifs.open(FILENAME, ios::in);
	int id;
	string name;
	int dId;
	int num = 0;
	while (ifs >> id && ifs >> name && ifs >> dId)
		num++;
	return num;
}
void WorkerManager::Show_Menu()
{
	cout << "********************************************" << endl;
	cout << "*********  欢迎使用职工管理系统！ **********" << endl;
	cout << "*************  0.退出管理程序  *************" << endl;
	cout << "*************  1.增加职工信息  *************" << endl;
	cout << "*************  2.显示职工信息  *************" << endl;
	cout << "*************  3.删除离职职工  *************" << endl;
	cout << "*************  4.修改职工信息  *************" << endl;
	cout << "*************  5.查找职工信息  *************" << endl;
	cout << "*************  6.按照编号排序  *************" << endl;
	cout << "*************  7.清空所有文档  *************" << endl;
	cout << "********************************************" << endl;
	cout << endl;
}
void WorkerManager::exitSystem()//退出
{
	cout << "欢迎下次使用" << endl;
	system("pause");
	exit(0);
}
void WorkerManager::Add_Emp()//增加职工
{
	cout << "请输入增加职工的数量：" << endl;
	int addNum = 0;
	cin >> addNum;
	if (addNum > 0)
	{
		//计算新空间大小
		int newSize = this->m_EmpNum + addNum;
		//开辟新空间
		Worker** newSpace = new Worker * [newSize];
		if (this->m_EmpArray != NULL)
		{
			for (int i = 0; i < this->m_EmpNum; i++)
			{
				newSpace[i] = this->m_EmpArray[i];
			}
		}
		for (int i = 0; i < addNum; i++)
		{
			int id;
			string name;
			int dSelect;
			cout << "请输入第" << i + 1 << " 个新职工编号：" << endl;
			cin >> id;
			cout << "请输入第 " << i + 1 << " 个新职工姓名：" << endl;
			cin >> name;
			cout << "请选择该职工的岗位：" << endl;
			cout << "1、普通职工" << endl;
			cout << "2、经理" << endl;
			cout << "3、老板" << endl;
			cin >> dSelect;
			Worker* worker = NULL;
			switch (dSelect)
			{
			case 1://普通员工
				worker = new Employee(id, name, 1);
				break;
			case 2://经理
				worker = new Manager(id, name, 2);
				break;
			case 3://老板
				worker = new Boss(id, name, 3);
				break;
			default:
				break;
			}
			newSpace[this->m_EmpNum + i] = worker;
		}
		//释放原有空间
		delete[] this->m_EmpArray;
		//更改新空间的指向
		this->m_EmpArray = newSpace;
		//更新新的个数
		this->m_EmpNum = newSize;
		//提示信息
		this->m_FileIsEmpty = false;//职工不为空
		cout << "成功添加" << addNum << "名新职工" << endl;
		this->save();
	}
	else cout << "输入有误" << endl;
	system("pause");
	system("cls");
}
void WorkerManager::save()
{
	ofstream ofs;
	ofs.open(FILENAME,ios::out);//用输出的方式打开文件  --写文件
	//将每个人数据写到文件中
	for (int i = 0; i < this->m_EmpNum; i++)
	{
		ofs << this->m_EmpArray[i]->m_Id << " "
			<< this->m_EmpArray[i]->m_Name << " "
			<< this->m_EmpArray[i]->m_DeptId << endl;
	}
	ofs.close();
}
void WorkerManager::init_Emp()
{
	ifstream ifs;
	ifs.open(FILENAME, ios::in);
	int id;
	string name;
	int dId;
	int idx = 0;
	while (ifs >> id && ifs >> name && ifs >> dId)
	{
		Worker* worker = NULL;//普通员工
		if (dId == 1)
			worker = new Employee(id, name, dId);
		else if (dId == 2) //2经理
			worker = new Manager(id, name, dId);
		else //总裁
			worker = new Boss(id, name, dId);
		this->m_EmpArray[idx++] = worker;
	}
	ifs.close();
}
void WorkerManager::Show_Emp()
{
	if (this->m_FileIsEmpty)
	{
		cout << "文件不存在或者记录为空" << endl;
	}
	else
	{
		for (int i = 0; i < m_EmpNum; i++)
		{
			//利用多态调用程序接口
			this->m_EmpArray[i]->showInfo();
		}
		system("pause");
		system("cls");
	}
}
void WorkerManager::Del_Emp()
{
	if (this->m_FileIsEmpty)
	{
		cout << "文件不存在或记录为空!" << endl;
	}
	else
	{
		//按职工编号删除
		cout << "请输入想要删除的职工号" << endl;
		int id = 0;
		cin >> id;
		int idx = this->IsExist(id);
		if (idx != -1)//说明员工存在
		{
			for (int i = idx; i < this->m_EmpNum - 1; i++)
			{
				this->m_EmpArray[i] = this->m_EmpArray[i + 1];
			}
			this->m_EmpNum--;
			this->save();
			cout << "删除成功" << endl;
		}
		else cout << "删除失败，未找到该职工" << endl;
	}
	system("pause");
	system("cls");
}
int WorkerManager::IsExist(int id)
{
	int idx = -1;
	for (int i = 0; i < this->m_EmpNum; i++)
	{
		if (this->m_EmpArray[i]->m_Id == id)
		{
			idx = i;
			break;
		}
	}
	return idx;
}
void WorkerManager::Mod_Emp()//修改职工
{
	if (this->m_FileIsEmpty)
	{
		cout << "文件不存在或者数据为空" << endl;
	}
	else
	{
		cout << "请输入修改职工的编号" << endl;
		int id;
		cin >> id;
		int ret=this->IsExist(id);
		if (ret != -1)
		{
			//查找到编号的员工
			delete this->m_EmpArray[ret];
			int newId = 0;
			string newName = "";
			int dSelect = 0;
			cout << "查找到 " << id << "号职工,请输入新的职工号" << endl;
			cin >> newId;
			cout << "请输入新的姓名" << endl;
			cin >> newName;
			cout << "请输入新的岗位" << endl;
			cout << "1、普通职工" << endl;
			cout << "2、经理" << endl;
			cout << "3、老板" << endl;
			cin >> dSelect;
			Worker* worker = NULL;
			switch (dSelect)
			{
			case 1:
				worker = new Employee(newId, newName, dSelect);
				break;
			case 2:
				worker = new Manager(newId, newName, dSelect);
				break;
			case 3:
				worker = new Boss(newId, newName, dSelect);
				break;
			default:
				break;
			}
			//更改数据到数组中
			this->m_EmpArray[ret] = worker;
			cout << "修改成功" << endl;
			this->save();
		}
		else cout << "修改失败，查无此人" << endl;
	}
	system("pause");
	system("cls");
}
void WorkerManager::Find_Emp()//查找职工
{
	if (this->m_FileIsEmpty)
	{
		cout << "文件不存在或记录为空" << endl;
	}
	else
	{
		cout << "请输入查找方式" << endl;
		cout << "1 按职工编号查找" << endl;
		cout << "2 按姓名查找" << endl;
		int select = 0;
		cin >> select;
		if (select == 1)
		{
			int id;
			cout << "请输入查找的职工编号" << endl;
			cin >> id;
			int ret = IsExist(id);
			if (ret != -1)
			{
				cout << "查找成功~，该职工信息如下" << endl;
				this->m_EmpArray[ret]->showInfo();
			}
			else cout << "查找失败，查无此人" << endl;
		}
		else if (select == 2)
		{
			string name;
			cout << "请输入需要查找的姓名" << endl;
			cin >> name;
			bool flag = false;
			for (int i = 0; i < m_EmpNum; i++)
			{
				if (m_EmpArray[i]->m_Name == name)
				{
					cout << "查找成功，职工编号为:" << m_EmpArray[i]->m_Id << " 号的信息如下" << endl;
					flag = true;
					this->m_EmpArray[i]->showInfo();
				}
			}
			if (flag == false)cout << "查找失败，查无此人" << endl;
		}
		else cout << "输入选项有误" << endl;
	}
	system("pause");
	system("cls");
}
void WorkerManager::Sort_Emp()
{
	if (this->m_FileIsEmpty)
	{
		cout << "文件不存在或记录为空!" << endl;
		system("pause");
		system("cls");
	}
	else
	{
		cout << "请选择排序方式： " << endl;
		cout << "1、按职工号进行升序" << endl;
		cout << "2、按职工号进行降序" << endl;
		int select = 0;
		cin >> select;
		for (int i = 0; i < m_EmpNum; i++)
		{
			int minv = i;
			for (int j = i + 1; j < m_EmpNum; j++)
			{
				if (select == 1)
				{
					if (m_EmpArray[minv]->m_Id > m_EmpArray[j]->m_Id)
						minv = j;
				}
				else
				{
					if (m_EmpArray[minv]->m_Id < m_EmpArray[j]->m_Id)
						minv = j;
				}
			}
			if (i != minv)
			{
				Worker* temp = m_EmpArray[i];
				m_EmpArray[i] = m_EmpArray[minv];
				m_EmpArray[minv] = temp;
			}
		}
		cout << "排序成功，排序后结果为:" << endl;
		this->save();
		this->Show_Emp();
	}
}
void WorkerManager::Clean_File()//清空文件
{
	cout << "确认清空？" << endl;
	cout << "1 确认" << endl;
	cout << "2 返回" << endl;
	int select = 0;
	cin >> select;
	if (select == 1)
	{
		ofstream ofs(FILENAME, ios::trunc);
		ofs.close();
		if (this->m_EmpArray != NULL)
		{
			for (int i = 0; i < this->m_EmpNum; i++)
			{
				if (this->m_EmpArray[i] != NULL)
					delete this->m_EmpArray[i];
			}
			this->m_EmpNum = 0;
			delete[] this->m_EmpArray;
			this->m_EmpArray = NULL;
			this->m_FileIsEmpty = true;
		}
		cout << "清空成功" << endl;
	}
	system("pause");
	system("cls");
}
WorkerManager::~WorkerManager()
{
	if (this->m_EmpArray != NULL)
	{
		delete[] this->m_EmpArray;
		this->m_EmpArray = NULL;
	}
}

//职工管理系统.cpp
#include <bits/stdc++.h>
#include "workerManager.h"
#include "worker.h"
#include "employee.h"
#include "manager.h"
#include "boss.h"
using namespace std;
int main()
{
	WorkerManager wm;
	int choice = 0;
	while (1)
	{
		wm.Show_Menu();
		cout << "请输入选择：" << endl;
		cin >> choice;
		switch (choice)
		{
		case 0:wm.exitSystem();
			break;//退出系统
		case 1:
			wm.Add_Emp();
			break;//添加职工
		case 2:
			wm.Show_Emp();
			break;//显示职工
		case 3:
			wm.Del_Emp();
			break;//删除职工
		case 4:
			wm.Mod_Emp();
			break;//修改职工
		case 5:
			wm.Find_Emp();
			break;//查找职工
		case 6:
			wm.Sort_Emp();
			break;//排序职工
		case 7:
			wm.Clean_File();
			break;//清空文件
		default:
			system("cls");
			break;
		}
	}
	system("pause");
	return 0;
}