C++实现Planet类对象向量的文件读写与数据持久化

错误原因 #

编译器提示operator<<不匹配，是因为C++标准库没有为自定义的Planet类实现输出流运算符。要将对象写入文件，必须手动重载std::ostream& operator<<；同理，从文件读取对象需要重载std::istream& operator>>。

步骤1：在Planet类中添加运算符重载的友元声明 #

为了让运算符能访问类的私有成员，需要将它们声明为友元：

class Planet
{
    private:
        string name;
        double mass, diameter;

    public:
        // ... 原有成员函数 ...

        // 添加友元声明
        friend std::ostream& operator<<(std::ostream& os, const Planet& planet);
        friend std::istream& operator>>(std::istream& is, Planet& planet);
    
        // ... 构造函数 ...
};

步骤2：实现输出/输入运算符重载 #

写入文件的输出运算符 #

这个函数定义如何将Planet对象的字段序列化到流中：

#include <ostream>
// 写入格式：name mass diameter（用空格分隔，假设name不含空格）
std::ostream& operator<<(std::ostream& os, const Planet& planet) {
    os << planet.name << " " << planet.mass << " " << planet.diameter << "\n";
    return os;
}

读取文件的输入运算符 #

对应写入格式，将流中的数据反序列化为Planet对象：

#include <istream>
std::istream& operator>>(std::istream& is, Planet& planet) {
    is >> planet.name >> planet.mass >> planet.diameter;
    return is;
}

注意：如果行星名称可能包含空格，需要修改序列化格式（比如用引号包裹名称、用逗号分隔字段），避免读取时解析错误。示例如下：

// 写入带空格的名称
std::ostream& operator<<(std::ostream& os, const Planet& planet) {
    os << "\"" << planet.name << "\"," << planet.mass << "," << planet.diameter << "\n";
    return os;
}

// 读取带空格的名称
#include <sstream>
std::istream& operator>>(std::istream& is, Planet& planet) {
    std::string line;
    if (!std::getline(is, line)) return is;

    std::istringstream iss(line);
    std::string name;
    char comma;

    // 读取引号包裹的名称
    std::getline(iss, name, '"');
    std::getline(iss, name, '"');
    iss >> comma; // 跳过逗号

    iss >> planet.mass >> comma >> planet.diameter;
    planet.name = name;
    return is;
}

步骤3：修改写入文件的代码 #

使用重载后的运算符写入，同时注意文件打开模式：

#include <fstream>
#include <vector>

// 覆盖写入文件（如果要追加用ios::app，但读取时要处理已有数据）
std::fstream file("data", std::ios::out | std::ios::trunc);
if (!file.is_open()) {
    std::cerr << "无法打开文件进行写入\n";
    return;
}

for (const auto& planet : listOfPlanets) {
    file << planet;
}
file.close();

步骤4：实现读取文件的代码 #

程序启动时从文件加载数据：

std::vector<Planet> listOfPlanets;
std::fstream file("data", std::ios::in);
if (!file.is_open()) {
    std::cerr << "无法打开文件进行读取（文件可能不存在）\n";
    // 文件不存在时，listOfPlanets保持空，不影响后续操作
} else {
    Planet planet;
    while (file >> planet) {
        listOfPlanets.push_back(planet);
    }
    file.close();
}

额外注意事项 #

• 确保Planet的默认构造函数能正确初始化成员（比如name为空字符串，mass和diameter为0），避免读取时出现未定义行为。
• 每次写入前如果需要覆盖旧数据，使用ios::out | ios::trunc模式；如果需要追加，使用ios::app，但读取时要读取所有历史数据。

内容的提问来源于stack exchange，提问作者kzina