RapidJSON 读取带注释文件，添加值后并保存

RapidJSON不直接支持读取带注释的JSON文件，但是你可以使用RapidJSON中的`Reader`类来读取JSON文件，然后将其转换为RapidJSON的DOM对象，向其中添加值，最后再将其转换回JSON字符串并保存到文件中。

以下是一个示例代码：

#include "rapidjson/document.h"
#include "rapidjson/writer.h"
#include "rapidjson/stringbuffer.h"
#include "rapidjson/reader.h"
#include <iostream>
#include <fstream>

using namespace rapidjson;

class MyHandler : public BaseReaderHandler<UTF8<>, MyHandler> {
public:
    bool Null() { return true; }
    bool Bool(bool b) { return true; }
    bool Int(int i) { return true; }
    bool Uint(unsigned u) { return true; }
    bool Int64(int64_t i) { return true; }
    bool Uint64(uint64_t u) { return true; }
    bool Double(double d) { return true; }
    bool RawNumber(const char* str, SizeType length, bool copy) { return true; }
    bool String(const char* str, SizeType length, bool copy) { return true; }

    bool StartObject() {
        // 当开始解析一个对象时，将其压入堆栈
        Value* obj = new Value(kObjectType);
        if (!stack_.empty()) {
            Value& parent = *stack_.top();
            if (parent.IsArray()) {
                parent.PushBack(*obj, allocator_);
            } else {
                parent.AddMember(StringRef(last_key_.c_str()), *obj, allocator_);
            }
        }
        stack_.push(obj);
        return true;
    }

    bool Key(const char* str, SizeType length, bool copy) {
        // 记录当前的key
        last_key_ = std::string(str, length);
        return true;
    }

    bool EndObject(SizeType memberCount) {
        // 当结束解析一个对象时，将其从堆栈中弹出并释放内存
        Value* obj = stack_.top();
        stack_.pop();
        if (stack_.empty()) {
            document_.CopyFrom(*obj, document_.GetAllocator());
        }
        delete obj;
        return true;
    }

    bool StartArray() {
        // 当开始解析一个数组时，将其压入堆栈
        Value* arr = new Value(kArrayType);
        if (!stack_.empty()) {
            Value& parent = *stack_.top();
            if (parent.IsArray()) {
                parent.PushBack(*arr, allocator_);
            } else {
                parent.AddMember(StringRef(last_key_.c_str()), *arr, allocator_);
            }
        }
        stack_.push(arr);
        return true;
    }

    bool EndArray(SizeType elementCount) {
        // 当结束解析一个数组时，将其从堆栈中弹出并释放内存
        Value* arr = stack_.top();
        stack_.pop();
        if (stack_.empty()) {
            document_.CopyFrom(*arr, document_.GetAllocator());
        }
        delete arr;
        return true;
    }

    Document& GetDocument() { return document_; }

private:
    Document document_;
    std::stack<Value*> stack_;
    std::string last_key_;
    MemoryPoolAllocator<>* allocator_ = &document_.GetAllocator();
};

void read_json_with_comments(const std::string& filename, Document& doc) {
    std::ifstream ifs(filename);
    std::string line;
    std::string json_str;
    while (std::getline(ifs, line)) {
        // 去掉注释
        auto pos = line.find("//");
        if (pos != std::string::npos) {
            line = line.substr(0, pos);
        }
        json_str += line;
    }

    MyHandler handler;
    Reader reader;
    StringStream ss(json_str.c_str());
    reader.Parse(ss, handler);
    doc.CopyFrom(handler.GetDocument(), doc.GetAllocator());
}

void write_json(const std::string& filename, const Document& doc) {
    std::ofstream ofs(filename);
    StringBuffer buffer;
    Writer<StringBuffer> writer(buffer);
    doc.Accept(writer);
    ofs << buffer.GetString() << std::endl;
}

int main() {
    Document doc;
    read_json_with_comments("input.json", doc);

    // 向JSON中添加值
    Value arr(kArrayType);
    for (int i = 0; i < 3; i++) {
        Value obj(kObjectType);
        obj.AddMember("name", "foo", doc.GetAllocator());
        obj.AddMember("value", i, doc.GetAllocator());
        arr.PushBack(obj, doc.GetAllocator());
    }
    doc.AddMember("new_array", arr, doc.GetAllocator());

    write_json("output.json", doc);

    return 0;
}

这个示例代码中，我们首先定义了一个自定义的`Reader`类`MyHandler`，它可以读取带注释的JSON文件并将其转换为RapidJSON的DOM对象。然后我们从`input.json`文件中读取JSON并将其转换为DOM对象。接着我们向DOM对象中添加一个新的数组，并将其保存到`new_array`字段中。最后我们将DOM对象转换回JSON字符串并保存到文件中。