C++实例2--职工管理系统

职工管理系统

1.  头文件

1.1 workerManager.h

　　系统类

#pragma once  // 防止头文件重复包含
#include<iostream>  // 包含输入输出流头文件
#include<fstream>  // 文件读写头文件
using namespace std;  // 使用标准命名空间
#include "worker.h"
#include "employee.h"
#include "manager.h"
#include "boss.h"

#define FILENAME "employee.txt"

class WorkerManager {
public:

    // 构造函数
    WorkerManager();

    // 展示菜单
    void Show_Menu();

    // 退出系统
    void exitSystem();

    // 记录文件中人数的个数
    int m_EmpNum;

    // 员工数组的指针
    Worker** m_EmpArray;

    // 增加职工
    void Add_Emp();

    // 保存文件
    void save();

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

    // 统计人数
    int get_EmpNum();

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

    // 显示职工
    void Show_Emp();

    // 判断职工是否存在（未删除做准备）
    int IsExit(int id);

    // 删除职工
    void Del_Emp();

    // 修改职工
    void Mod_Emp();

    // 查找职工
    void Find_Emp();

    // 排序职工
    void Sort_Emp();

    // 清空文件
    void Clean_File();

    // 析构函数
    ~WorkerManager();


};

 

1.2 worker.h

　　员工基类（抽象类）

#pragma once
#include <iostream>
#include <string>

using namespace std;

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

    // 显示个人信息
    virtual void showInfo() = 0;

    // 获取岗位信息
    virtual string getDeptName() = 0;

    int m_Id;  // 职工编号
    string m_Name;  // 职工姓名
    int m_DeptId;  // 职工所在部门名称编号
};

 

1.3 employee.h

　　员工类

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

#include "worker.h"

// 员工类
class Employee : public Worker {
public:
    // 构造函数
    Employee(int id, string name, int dId);

    // 显示个人信息
    virtual void showInfo();

    // 获取职工岗位名称
    virtual string getDeptName();

};

 

1.4 manager.h

　　经理类

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

#include "worker.h"

// 经理类
class Manager: public Worker {
public:
    // 构造函数
    Manager(int id, string name, int dId);

    // 显示个人信息
    virtual void showInfo();

    // 获取职工岗位名称
    virtual string getDeptName();

};

 

1.5 boss.h

　　老板类

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

#include "worker.h"

// 经理类
class Boss : public Worker {
public:
    // 构造函数
    Boss(int id, string name, int dId);

    // 显示个人信息
    virtual void showInfo();

    // 获取职工岗位名称
    virtual string getDeptName();

};

 

2. 源文件

2.1 workerManager.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();


}

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

// 展示菜单
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::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::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");

}

// 获取当前职工数量
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 {  // 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();
        }
    }

    system("pause");
    system("cls");
}

// 判断职工是否存在
int WorkerManager::IsExit(int id) {
    // 按照职工编号判断职工是否存在
    // 存在则返回职工在数组中的位置，不存在返回-1
    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->IsExit(id);

        if (index != -1) {  // 说明职工存在，并且要删除掉index位置上的职工
            for (int i = 0; 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->IsExit(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 << "修改成功！" << this->m_EmpArray[ret]->m_DeptId << 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 = IsExit(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 << "1.按照职工工号降序" << 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");
}

 

2.2 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("员工");
}

 

2.3 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("经理");
}

 

2.4 Boss.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("老板");
}

2.5 职工管理系统.cpp

　　主函数

#include<iostream>
#include<string>

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

using namespace std;

// 职工管理系统


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;
        case 7:  // 清空文件
            wm.Clean_File();
            break;
        default:
            system("cls");
            break;
        }
    }

    system("pause");

    return 0;

}

 

 

参考：《黑马程序员》C++教程