C++实战(二)——基于多态的职工管理系统完整代码
本文转载自：https://github.com/AnkerLeng/Cpp-0-1-Resource.git
视频见：https://www.bilibili.com/video/av41559729?p=31
workerManager.h
#pragma once // 防止头文件重复包含
#includeworkerManager.cpp
#include "workerManager.h"
WorkerManager::WorkerManager()
{
ifstream ifs;
ifs.open(FILENAME, ios::in);
//1、文件不存在情况
if (!ifs.is_open())
{
cout << "文件不存在" << endl; //测试输出
this->m_EmpNum = 0; //初始化人数
this->m_FileIsEmpty = true; //初始化文件为空标志
this->m_EmpArray = NULL; //初始化数组
ifs.close(); //关闭文件
return;
}
//2、文件存在，并且没有记录
char ch;
ifs >> ch;
if (ifs.eof())
{
cout << "文件为空!" << endl;
this->m_EmpNum = 0;
this->m_FileIsEmpty = true;
this->m_EmpArray = NULL;
ifs.close();
return;
}
//3、文件存在，且有数据
int num = this->get_EmpNum();
cout << "职工个数为：" << num << endl; //测试代码
this->m_EmpNum = num; //更新成员属性
//根据职工数创建数组
this->m_EmpArray = new Worker *[this->m_EmpNum];
//初始化职工，将文件中的数据存储到数组中
init_Emp();
this ->m_Fileisempty = false;
//测试代码
for (int i = 0; i < m_EmpNum; i++)
{
cout << "职工号： " << this->m_EmpArray[i]->m_Id
<< " 职工姓名： " << this->m_EmpArray[i]->m_Name
<< " 部门编号： " << this->m_EmpArray[i]->m_DeptId << endl;
}
}
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();
}
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++;
}
ifs.close();
return num;
}
void WorkerManager::init_Emp()
{
ifstream ifs;
ifs.open(FILENAME, ios::in);
int id;
string name;
int dId;
int index = 0;
while (ifs >> id && ifs >> name && ifs >> dId)
{
Worker * worker = NULL;
//根据不同的部门Id创建不同对象
if (dId == 1) // 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[index] = worker;
index++;
}
}
//显示职工
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");
}
int WorkerManager::IsExist(int id)
{
int index = -1;
for (int i = 0; i < this->m_EmpNum; i++)
{
if (this->m_EmpArray[i]->m_Id == id)
{
index = i;
break;
}
}
return index;
}
//删除职工
void WorkerManager::Del_Emp()
{
if (this->m_FileIsEmpty)
{
cout << "文件不存在或记录为空！" << endl;
}
else
{
//按职工编号删除
cout << "请输入想要删除的职工号：" << endl;
int id = 0;
cin >> id;
int index = this->IsExist(id);
if (index != -1) //说明index上位置数据需要删除
{
for (int i = index; 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");
}
//修改职工
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)
{
case1:
worker = new Employee(newId, newName, dSelect);
break;
case2:
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 minOrMax = i;
for (int j = i + 1; j < m_EmpNum; j++)
{
if (select == 1) //升序
{
if (m_EmpArray[minOrMax]->m_Id > m_EmpArray[j]->m_Id)
{
minOrMax = j;
}
}
else //降序
{
if (m_EmpArray[minOrMax]->m_Id < m_EmpArray[j]->m_Id)
{
minOrMax = j;
}
}
}
if (i != minOrMax)
{
Worker * temp = m_EmpArray[i];
m_EmpArray[i] = m_EmpArray[minOrMax];
m_EmpArray[minOrMax] = 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)
{
//打开模式 ios::trunc 如果存在删除文件并重新创建
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;
}
}职工管理系统.cpp
#includeworker.h，员工抽象类
#pragma once
#includeemployee.h，创建普通员工类
#pragma once
#includeemployee.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.h，创建经理类
#pragma once
#includemanager.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("经理");
}
boss.h，创建老板类
#pragma once
#includeboss.cpp
#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("总裁");
}