14.9 文件交互-读文件

14.9 文件交互-读文件

功能描述：将文件中的内容读取到程序中
虽然我们实现了添加职工后保存到文件的操作，但是每次开始运行程序，并没有将文件中数据读取到程序中
而我们的程序功能中还有清空文件的需求
因此构造函数初始化数据的情况分为三种

1. 第一次使用，文件未创建
2. 文件存在，但是数据被用户清空
3. 文件存在，并且保存职工的所有数据

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();