C++案例——职工管理系统（黑马程序员C++ p147-p165）

核心--WorkerManager.cpp

WorkerManager.cpp(核心)

#include "workerManager.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;
    }

    //2.文件存在 数据为空
    char ch;
    ifs>>ch;
    if(ifs.eof())
    {
        //文件为空
        //cout<<"文件为空"<<endl;
        //初始化记录人数
        this->m_EmpNum=0;
        //初始化指针
        this->m_EmpArray=NULL;
        //初始化文件是否为空
        this->m_FileIsEmpty=true;
        ifs.close();
        return;
    }

    //3.文件存在不为空，并且记录数据
    int num= this->get_EmpNum();
    this->m_EmpNum=num;
    //cout<<"职工人数:"<<num<<endl;
    //开辟空间
    this->m_EmpArray = new Worker*[this->m_EmpNum];
    //将文件中的数据，存到数组中
    this->init_Emp();
    
}

//展示菜单
void WorkerManager::Show_Menu()
{
    cout<<"-------------------欢迎使用职工管理系统----------------"<<endl;
    cout<<"-------------------0.退出管理系统---------------------"<<endl;
    cout<<"-------------------1.增加职工信息---------------------"<<endl;
    cout<<"-------------------2.显示职工信息---------------------"<<endl;
    cout<<"-------------------3.删除离职职工---------------------"<<endl;
    cout<<"-------------------4.修改职工信息---------------------"<<endl;
    cout<<"-------------------5.查找职工信息---------------------"<<endl;
    cout<<"-------------------6.按照编号排序---------------------"<<endl;
    cout<<endl;
}

void WorkerManager::ExitSystem()
{
    cout<<"欢迎下次使用"<<endl;
    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;

        //成功添加后 保存到文件中
        this->save();

        //提示添加成功
        cout<<"成功添加"<<addNum<<"名新职工!"<<endl;

        
    }
    else
    {
        cout<<"输入数据有误"<<endl;
    }
    
    //按任意键后 清屏回到上级目录
    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_DepId<<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++;
        }

        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;

        if (dId == 1) //普通员工
        {
            worker = new Employee(id,name,dId);
        }
        else if (dId == 2)//经理
        {
            worker = new Manager(id,name,dId);
        }
        else if (dId == 3) //老板
        {
            worker = new Boss(id,name,dId);
        }
        this->m_EmpArray[index] = worker;
        index++;
        
    }
    //关闭文件
    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();
        }
        
    }
    
}

//判断职工是否存在 如果存在返回职工所在数组中的位置，不存在返回-1
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;
        }
        
    } 
    
}

    //修改职工
    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;
            }
            
        }
        
    }

    //查找职工
    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 (this->m_EmpArray[i]->m_name == name)
                    {
                        cout<<"查找成功！职工编号为:"<<this->m_EmpArray[i]->m_Id
                            <<"号的职工信息如下:"<<endl;

                        flag=true;

                        //调用显示信息接口
                        this->m_EmpArray[i]->showInfo();
                    }
                    
                }
                if (flag == false)
                {
                    cout<<"查找失败！查无此人!"<<endl;
                }
                
            }
            else
            {
                cout<<"输入选项错误"<<endl;
            }
            
        }
         
    }

//排序职工
    void WorkerManager::Sort_Emp()
    {
        if (this->m_FileIsEmpty)
        {
            cout << "文件不存在或记录为空" << endl;
        }
        else
        {
            cout << "请选择排序方式" << endl;
            cout << "1.按照职工号进行升序" << endl;
            cout << "2.按照职工号进行降序" << endl;
            int select = 0;
            cin >> select;
            for (int i = 0; i < this->m_EmpNum; i++)
            {
                int MinOrMax = i;//声明最大值或最小值下标
                for (int j = i + 1; j < this->m_EmpNum; j++)
                {
                    if (select == 1)
                    {
                        //升序
                        if (this->m_EmpArray[MinOrMax]->m_Id > this->m_EmpArray[j]->m_Id)
                        {
                            MinOrMax = j;
                        }
                    }
                    else
                    {
                        //降序
                        if (this->m_EmpArray[MinOrMax]->m_Id < this->m_EmpArray[j]->m_Id)
                        {
                            MinOrMax = j;
                        }
                    }
                }
                //判断一开始认定的最大值或最小值是不是计算的最大值或最小值，如果不是就交换
                if (i != MinOrMax)
                {
                    Worker* temp = this->m_EmpArray[i];
                    this->m_EmpArray[i] = this->m_EmpArray[MinOrMax];
                    this->m_EmpArray[MinOrMax] = temp;
                }
            }
            cout << "排序成功！排序后的结果为:" << endl;
            this->Show_Emp();//展示所有职工
        }
    }

WorkerManager::~WorkerManager()
{
    if (this->m_EmpArray != NULL)
    {
        delete[] this->m_EmpArray;
        this->m_EmpArray=NULL;
    }
    
}

WorkerManager.h
#pragma once//防止头文件重复包含
#include<iostream>//包含输入输出流头文件

#include"worker.h"
#include"employee.h"
#include"manager.h"
#include"boss.h"

#include<fstream>
#define FILENAME "empFile.txt"

using namespace std;//使用标准命名空间

class WorkerManager
{
    public:

    //构造函数
    WorkerManager();

    //展示菜单
    void Show_Menu();

    //退出系统
    void ExitSystem();

    //记录职工人数
    int m_EmpNum;

    //职工数组指针
    Worker ** m_EmpArray;

    //保存文件
    void save();

    //添加员工函数声明
    void Add_Emp();

    //判断文件是否为空 标志
    bool m_FileIsEmpty;

    //统计文件中的人数
    int get_EmpNum();

    //初始化员工
    void init_Emp();

    //显示职工
    void Show_Emp();

    //判断职工是否存在 如果存在返回职工所在数组中的位置，不存在返回-1
    int IsExist(int id);

    //删除职工
    void Del_Emp();

    //修改职工
    void Mod_Emp();

    //查找职工
    void Find_Emp();

    //排序职工
    void Sort_Emp();

    //析构函数
    ~WorkerManager();

};

worker.h
#pragma once
#include<iostream>
using namespace std;
#include<string>

//职工抽象类
class Worker
{
    public:

    //显示个人信息
    virtual void showInfo()=0;
    //获取岗位名称
    virtual string getDeptName()=0;
    //职工编号
    int m_Id;
    //职工姓名
    string m_name;
    //部门编号
    int m_DepId;
};

manager.h
#pragma once
#include<iostream>
#include"worker.h"

//经理类
class Manager :public Worker
{
    public:
    //构造函数
    Manager(int id,string name,int dId);
    //显示个人信息
    virtual void showInfo();
    //获取岗位名称
    virtual string getDeptName();
};

manager.cpp
#include "manager.h"

    //构造函数
    Manager::Manager(int id,string name,int dId)
    {
        this->m_Id=id;
        this->m_name=name;
        this->m_DepId=dId;
    }
    //显示个人信息
    void Manager::showInfo()
    {
        cout<<"职工编号:"<<this->m_Id
            <<"\t职工姓名:"<<this->m_name
            <<"\t岗位:"<<this->getDeptName()
            <<"\t岗位职责:完成老板交给的任务，并下发任务给员工"<<endl;
    }
    //获取岗位名称
    string Manager::getDeptName()
    {
        return string("经理");
    }

employee.h
//普通员工文件
#pragma once
#include<iostream>
#include"worker.h"
using namespace std;

class Employee :public Worker
{
public:

    //构造函数
    Employee(int id,string name,int dId);

    //显示个人信息
    virtual void showInfo();
    //获取岗位名称
    virtual string getDeptName();
};

employee.cpp
#include"employee.h"
    //构造函数
    Employee::Employee(int id,string name,int dId)
    {
        this->m_Id=id;
        this->m_name=name;
        this->m_DepId=dId;
    }

    //显示个人信息
    void Employee::showInfo()
    {
        cout<<"职工编号:"<<this->m_Id
            <<"\t职工姓名:"<<this->m_name
            <<"\t岗位:"<<this->getDeptName()
            <<"\t岗位职责:完成经理交给的任务"<<endl;
    }
    //获取岗位名称
    string Employee::getDeptName()
    {
        return string("员工");
    }


boss.h
#pragma once
#include<iostream>
#include"worker.h"

//老板类
class Boss :public Worker
{
    public:
    //构造函数
    Boss(int id,string name,int dId);
    //显示个人信息
    virtual void showInfo();
    //获取岗位名称
    virtual string getDeptName();
};

boss.cpp
#include "boss.h"

    //构造函数
    Boss::Boss(int id,string name,int dId)
    {
        this->m_Id=id;
        this->m_name=name;
        this->m_DepId=dId;
    }
    //显示个人信息
    void Boss::showInfo()
    {
        cout<<"职工编号:"<<this->m_Id
            <<"\t职工姓名:"<<this->m_name
            <<"\t岗位:"<<this->getDeptName()
            <<"\t岗位职责:管理公司所有事务"<<endl;
    }
    //获取岗位名称
    string Boss::getDeptName()
    {
        return string("老板");
    }


System.cpp(实现)

#include<iostream>
using namespace std;

#include "workerManager.h"
#include"worker.h"
#include"employee.h"
#include"manager.h"
#include"boss.h"

int main()
{
    //实例化管理者对象
    WorkerManager wm;

    int choice=0;//用来存储用户的选项
    while (true)
    {
        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;        
        default:
            break;
        }

    }
    
    //调用展示菜单成员函数
    wm.Show_Menu();

    return 0;
}