14.9 文件交互-读文件
功能描述:将文件中的内容读取到程序中
虽然我们实现了添加职工后保存到文件的操作,但是每次开始运行程序,并没有将文件中数据读取到程序中
而我们的程序功能中还有清空文件的需求
因此构造函数初始化数据的情况分为三种
- 第一次使用,文件未创建
- 文件存在,但是数据被用户清空
- 文件存在,并且保存职工的所有数据
1、文件不存在或为空
在workerManager.h添加新属性,标志文件是否为空
//判断文件是否为空
bool m_FileIsEmpty;
所以在添加职工后要把文件为空的属性设置为false:
this->m_FileIsEmpty = false;
重新写WorkerManager类的构造函数,用来初始化存放数据的数组:
WorkerManager::WorkerManager()
{
ifstream ifs;
ifs.open(FILENAME, ios::in);
//1、文件不存在
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;
}
}
2、文件存在且不为空
这种情况就要正常去读取文件中的数据了。
- 首先要判断文件中有多少条记录,好创建一个对于大小的数组去读取:
在workerManager.h中添加成员函数:
//统计文件中的人数
int get_EmpNum();
在workerManager.cpp中实现函数:
//统计文件中人数
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;
}
- 再写一个用来初始化数组的函数
在workerManager.h中添加成员函数:
//初始化员工
void init_Emp();
在workerManager.cpp中实现函数:
//初始化员工
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();
}
- 在构造函数中加入第三种情况
//3、文件存在,并且记录数据
int num = this->get_EmpNum();
this->m_EmpNum = num;
this->m_EmpArray = new Worker *[this->m_EmpNum];
this->init_Emp();