上述代码再加一个读取功能
时间: 2024-03-07 13:46:47 浏览: 23
好的,下面是一个增加了读取txt文件的功能的C++代码,它是在上面的职工工作量管理系统的基础上进行扩展的。
```cpp
#include <iostream>
#include <fstream>
#include <string>
#include <vector>
using namespace std;
class Employee {
public:
Employee(string n, int w) : name(n), work(w) {}
string getName() const { return name; }
int getWork() const { return work; }
private:
string name;
int work;
};
class WorkloadManager {
public:
void addEmployee(Employee e) {
employees.push_back(e);
}
void printWorkload() const {
for (auto e : employees) {
cout << e.getName() << " : " << e.getWork() << endl;
}
}
void saveToFile(string filename) const {
ofstream outfile(filename);
if (!outfile.is_open()) {
cout << "Error: cannot open file " << filename << endl;
return;
}
for (auto e : employees) {
outfile << e.getName() << "," << e.getWork() << endl;
}
outfile.close();
cout << "Workload saved to file " << filename << endl;
}
void loadFromFile(string filename) {
ifstream infile(filename);
if (!infile.is_open()) {
cout << "Error: cannot open file " << filename << endl;
return;
}
employees.clear();
string line;
while (getline(infile, line)) {
string name;
int work;
size_t pos = line.find(",");
if (pos != string::npos) {
name = line.substr(0, pos);
work = stoi(line.substr(pos + 1));
employees.push_back(Employee(name, work));
}
}
infile.close();
cout << "Workload loaded from file " << filename << endl;
}
private:
vector<Employee> employees;
};
int main() {
WorkloadManager manager;
manager.loadFromFile("workload.txt");
manager.addEmployee(Employee("Dave", 35));
manager.printWorkload();
manager.saveToFile("workload.txt");
return 0;
}
```
在这个例子中,我们在 `WorkloadManager` 类中增加了一个名为 `loadFromFile` 的函数,用于从指定的文件中读取工作量列表。
这个函数首先尝试打开指定的文件,如果打开失败则输出错误信息并返回。如果打开成功,则清空原有的职工列表,并逐行读取文件内容。对于每一行,我们使用 `find` 函数查找逗号的位置,然后使用 `substr` 函数分别提取姓名和工作量,并将它们作为参数创建一个新的 `Employee` 对象,并添加到列表中。
最后,我们在 `main` 函数中先调用 `loadFromFile` 函数从文件中读取工作量列表,然后添加了一个新的职工,再打印工作量列表,并通过调用 `saveToFile` 函数将工作量列表保存到文件中。
请注意,这个例子中使用了 `stoi` 函数将字符串转换为整数类型,因此需要包含 `#include <string>` 头文件。另外,如果需要支持更复杂的操作,比如对读取的工作量列表进行排序或筛选,可以进一步扩展代码。
相关推荐
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)