如何从protobuf描述符中获取方法选项?

时间:2016-12-15 13:19:51

标签: protocol-buffers protoc

我尝试使用protobuf编译器解析.proto文件。但这是一个令人困惑的问题,我无法获得方法的选项。

似乎我的选项被视为“未知字段”,但不是选项。

有什么方法可以解决这个问题吗?感谢。

(我讨厌在这里粘贴很多代码,但我认为完全描述问题是必不可少的。很抱歉。)

(环境:g ++ 4.7,Ubuntu 16.04,Protobuf 3.0.0)

#include <google/protobuf/descriptor.h>
#include <google/protobuf/descriptor.pb.h>
#include <google/protobuf/dynamic_message.h>
#include <google/protobuf/compiler/importer.h>

using namespace std;
using namespace google::protobuf;
using namespace google::protobuf::compiler;

#define print(x) std::cout << x << std::endl
#define input(x) std::cin >> x

int main() {
    DiskSourceTree sourceTree;
    sourceTree.MapPath("", "./");
    sourceTree.MapPath("", "./protobuf/include/");
    Importer importer(&sourceTree, NULL);

    auto fd = importer.Import("example.proto");
    assert(fd);

    int service_count = fd->service_count();

    for (int i = 0; i < service_count; i++) {
        auto service_d = fd->service(i);
        int method_count = service_d->method_count();
        for (int j = 0; j < method_count; j++) {
            auto method_d = service_d->method(j);
            print(method_d->options().unknown_fields().field_count());$ 
            print(">> " << method_d->options().uninterpreted_option_size());
        }
    }
    return 0;
}
// lrpc.proto
syntax = "proto3";
package lrpc;
import "google/protobuf/descriptor.proto";


extend google.protobuf.MethodOptions {
    int32 CmdID      = 50000;
    string OptString = 50001;
    string Usage     = 50002;
}
// example.proto
syntax = "proto3";

package foobar;

import "google/protobuf/wrappers.proto";
import "google/protobuf/empty.proto";

import "lrpc.proto";

message SearchRequest {
    // ... 
}

message SearchResponse {
    // ...
}

service SearchService {
    rpc Search( SearchRequest ) returns( SearchResponse ) {
        option( lrpc.CmdID ) = 1;
    }
}

1 个答案:

答案 0 :(得分:2)

选项不是未知字段,因为它们是extensions!声称在proto3中删除了扩展,但是当您使用.proto动态解析Importer文件时,无论您声明的语法版本如何,都会启用扩展。

如果在内部for()循环中添加一行,如:

print(method_d->options().DebugString());

您将获得如下输出:

[lrpc.CmdID]: 1

您可以使用protobuf反射枚举扩展值 - 当您调用Reflection::ListFields()时,它们会显示。