protobuf lib庫的使用

問題記錄：

1、在使用protobuf反射機制動態加載解析proto文件時，發現當proto文件中含有import系統proto文件的語句時，無法解析文件，解決方法是添加路徑映射。

google::protobuf::compiler::DiskSourceTree sourceTree;
sourceTree.MapPath("data", "./data");
sourceTree.MapPath("", "D:\\Documents\\Program\\Tools\\protobuf-3.0.2\\install\\x86\\debug\\include");
google::protobuf::compiler::Importer importer(&sourceTree, NULL);
const google::protobuf::FileDescriptor *fileDescriptor = importer.Import("data/test.proto");

　　代碼如上，其中的第3行為解決方案，增加之后才能正確解析。分析其原因是，Importer對象用於導入並解析proto文件，當proto文件中import了其他proto文件時，Importer對象遞歸導入並解析該proto文件；第二行告訴了Importer去哪里找test.proto，但是卻沒有告訴Importer去哪里找系統自帶的proto文件，因此需要加上第3行，並且別名應該留空！

 

2、jsoncpp的下載和使用

　　jsoncpp源碼可以從github上得到：jsoncpp-master.zip

　　解壓后使用python執行根目錄下的 amalgamate.py ，這個腳本將jsoncpp的頭文件和源代碼進行了合並，最終合並成了三個文件：

　　dist\json\json.h　　dist\json\json-forwards.h　　dist\jsoncpp.cpp

　　使用時把 jsoncpp.cpp文件連同json文件夾一起拷貝到工程目錄下，兩者保持同級，代碼中包含 json\json.h 即可。

 

3、遍歷proto文件中的所有消息以及所有字段

#include <iostream>
#include <google/protobuf/compiler/importer.h>
#include <google/protobuf/dynamic_message.h>
#include <google/protobuf/util/json_util.h>

int parseProtoFile()
{
    // 准備配置好文件系統
    google::protobuf::compiler::DiskSourceTree sourceTree;
    // 將當前路徑映射為項目根目錄 ， project_root 僅僅是個名字，你可以你想要的合法名字.
    sourceTree.MapPath("data", "./data");
    sourceTree.MapPath("", "D:\\Documents\\Tools\\protobuf-3.0.2\\install\\x86\\debug\\include");
    // 配置動態編譯器.
    google::protobuf::compiler::Importer importer(&sourceTree, NULL);
    // 動態編譯proto源文件。 源文件在./source/proto/test.proto .
    auto fileDescriptor = importer.Import("data/complex.proto");


    std::cout << fileDescriptor->message_type_count() << std::endl;
    for (auto i = 0; i < fileDescriptor->message_type_count(); i++)
    {
        auto descriptor = fileDescriptor->message_type(i);
        
        std::cout << descriptor->name() << " " << descriptor->field_count() << " " << descriptor->nested_type_count() << std::endl;

        auto descriptor1 = descriptor->containing_type();

        if (descriptor1)
        {
            std::cout << descriptor1->name() << std::endl;
        }
    }
    
    std::cout << fileDescriptor->name() << std::endl;
    

    auto descriptor = fileDescriptor->message_type(1);
    for (auto i = 0; i < descriptor->field_count(); i++)
    {
        auto fieldDes = descriptor->field(i);
        google::protobuf::SourceLocation outLocation;
        if (fieldDes->GetSourceLocation(&outLocation))
        {
            printf("%s: %d %d %d %d\nleading_comments:%s\ntrailing_comments:%s\n", 
                fieldDes->full_name().c_str(),
                outLocation.start_line, outLocation.start_column, outLocation.end_line, outLocation.end_column, 
                outLocation.leading_comments.c_str(), outLocation.trailing_comments.c_str());
            for (auto comment : outLocation.leading_detached_comments)
            {
                printf("leading_detached_comments:%s\n", comment.c_str());
            }
        }
        else
        {
            std::cout << "fail" << std::endl;
        }
    }
    
#if 0
    // 現在可以從編譯器中提取類型的描述信息.
    auto descriptor1 = importer.pool()->FindMessageTypeByName("T.Test.InMsg");
    
    // 創建一個動態的消息工廠.
    google::protobuf::DynamicMessageFactory factory;
    // 從消息工廠中創建出一個類型原型.
    auto proto1 = factory.GetPrototype(descriptor1);
    // 構造一個可用的消息.
    auto message1 = proto1->New();
    // 下面是通過反射接口給字段賦值.
    auto reflection1 = message1->GetReflection();
    auto filed1 = descriptor1->FindFieldByName("id");
    reflection1->SetUInt32(message1, filed1, 1);

    // 打印看看
    std::cout << message1->DebugString() << std::endl;

    std::string output;
    google::protobuf::util::MessageToJsonString(*message1, &output);
    std::cout << output << std::endl;

    // 刪除消息.
    delete message1;
#endif
    return 0;
}

#define Log(format, ...) printf(format, __VA_ARGS__)

void printOneField(const google::protobuf::FieldDescriptor *fieldDescriptor)
{
    Log("  field[%d]: name %s, full name %s, json name %s, type %s, cpp type %s\n",
        fieldDescriptor->index(), fieldDescriptor->name().c_str(), fieldDescriptor->full_name().c_str(), fieldDescriptor->json_name().c_str(),
        fieldDescriptor->type_name(), fieldDescriptor->cpp_type_name());
    Log("  debug string:%s\n", fieldDescriptor->DebugString().c_str());
}

void printOneMessage(const google::protobuf::Descriptor *descriptor)
{
    // 消息的總體信息
    Log("msg[%d]: name %s, full name %s, field count %d, nested type count %d\n",
        descriptor->index(), descriptor->name().c_str(), descriptor->full_name().c_str(), descriptor->field_count(),
        descriptor->nested_type_count());
    Log("\tdebug string: %s\n", descriptor->DebugString().c_str());

    // 遍歷消息的所有字段
    for (int fieldLoop = 0; fieldLoop < descriptor->field_count(); fieldLoop++)
    {
        const google::protobuf::FieldDescriptor *fieldDescriptor = descriptor->field(fieldLoop);

        printOneField(fieldDescriptor);
    }

    // 遍歷消息的所有嵌套消息
    for (int nestedLoop = 0; nestedLoop < descriptor->nested_type_count(); nestedLoop++)
    {
        const google::protobuf::Descriptor *nestedDescriptor = descriptor->nested_type(nestedLoop);

        printOneMessage(nestedDescriptor);
    }
}

void printOneFile(const google::protobuf::FileDescriptor *fileDescriptor)
{
    Log("******** message info in proto file, msg count %d ********\n", fileDescriptor->message_type_count());

    // 遍歷文件中的所有頂層消息
    for (int msgLoop = 0; msgLoop < fileDescriptor->message_type_count(); msgLoop++)
    {
        const google::protobuf::Descriptor *descriptor = fileDescriptor->message_type(msgLoop);

        printOneMessage(descriptor);
    }
}

bool testProto(const char *protoIncludePath, const char *testProtoPath, const char *testProtoFile)
{
    // 配置文件系統
    google::protobuf::compiler::DiskSourceTree sourceTree;
    sourceTree.MapPath("", protoIncludePath);
    sourceTree.MapPath("data", testProtoPath);
    //sourceTree.MapPath("data", "./data");
    //sourceTree.MapPath("", "D:\\Documents\\Tools\\protobuf-3.0.2\\install\\x86\\debug\\include");
    // 配置動態編譯器
    google::protobuf::compiler::Importer importer(&sourceTree, NULL);
    // 動態編譯proto源文件
    const google::protobuf::FileDescriptor *fileDescriptor = importer.Import("data/" + std::string(testProtoFile));

    if (fileDescriptor == NULL)
    {
        printf("import \"%s\" failed, last error msg: %s\n", testProtoFile, sourceTree.GetLastErrorMessage().c_str());
        return false;
    }

    printOneFile(fileDescriptor);

    return true;
}

int main()
{
    const char *protoIncludePath = "D:\\Documents\\Tools\\protobuf-3.0.2\\install\\x86\\debug\\include";
    const char *testProtoPath = "C:\\Users\\Administrator\\Desktop\\Document\\C++\\protobufTest\\protobufTest\\data";
    const char *testProtoFile = "complex.proto";

    testProto(protoIncludePath, testProtoPath, testProtoFile);

    //parseProtoFile();
    //printf("Hello world!\n");
    return 0;
}