我是protobuf的新手,我遇到了简单的任务:我需要遍历消息字段并检查它的类型。如果type是消息,我将以递归方式对此消息执行相同的操作。
例如,我有这样的消息:
package MyTool;
message Configuration {
required GloablSettings globalSettings = 1;
optional string option1 = 2;
optional int32 option2 = 3;
optional bool option3 = 4;
}
message GloablSettings {
required bool option1 = 1;
required bool option2 = 2;
required bool option3 = 3;
}
现在,要在C ++中显式访问字段值,我可以这样做:
MyTool::Configuration config;
fstream input("config", ios::in | ios::binary);
config.ParseFromIstream(&input);
bool option1val = config.globalSettings().option1();
bool option2val = config.globalSettings().option2();
等等。如果有大量的字段,这种方法不方便。
我可以通过迭代执行此操作并获取字段的名称和类型吗?我知道有类型的描述符,有些叫做反射,但我的尝试没有成功。 如果有可能,有人可以给我一些代码示例吗?
谢谢!
答案 0 :(得分:18)
这是旧的,但也许有人会受益。这是一个打印protobuf消息内容的方法:
void Example::printMessageContents(std::shared_ptr<google::protobuf::Message> m)
{
const Descriptor *desc = m->GetDescriptor();
const Reflection *refl = m->GetReflection();
int fieldCount= desc->field_count();
fprintf(stderr, "The fullname of the message is %s \n", desc->full_name().c_str());
for(int i=0;i<fieldCount;i++)
{
const FieldDescriptor *field = desc->field(i);
fprintf(stderr, "The name of the %i th element is %s and the type is %s \n",i,field->name().c_str(),field->type_name());
}
}
您可以在FieldDescriptor Enum Values中找到从field->type
获得的可能值。例如,对于消息类型,您必须检查类型是否等于FieldDescriptor::TYPE_MESSAGE
。
此功能打印protobuf消息的所有“元数据”。但是,您需要单独检查每个值的类型,然后使用Reflection调用相应的getter函数。
因此,使用这个条件,我们可以提取字符串:
if(field->type() == FieldDescriptor::TYPE_STRING && !field->is_repeated())
{
std::string g= refl->GetString(*m, field);
fprintf(stderr, "The value is %s ",g.c_str());
}
然而,可以重复或不重复字段,并且对两种字段类型使用不同的方法。因此,这里使用检查来确保我们使用正确的方法。对于重复的字段,我们有例如字符串的这种方法:
GetRepeatedString(const Message & message, const FieldDescriptor * field, int index
)
因此需要考虑重复场的索引。
对于Message类型的FieldDescriptor,提供的功能只会打印消息的名称,我们也会更好地打印它的内容。
if(field->type()==FieldDescriptor::TYPE_MESSAGE)
{
if(!field->is_repeated())
{
const Message &mfield = refl->GetMessage(*m, field);
Message *mcopy = mfield.New();
mcopy->CopyFrom(mfield);
void *ptr = new std::shared_ptr<Message>(mcopy);
std::shared_ptr<Message> *m =
static_cast<std::shared_ptr<Message> *>(ptr);
printMessageContents(*m);
}
}
最后,如果重复该字段,则必须在反射上调用FieldSize
方法并迭代所有重复的字段。
答案 1 :(得分:8)
看一下Protobuf库如何实现TextFormat::Printer
类,它使用描述符和反射迭代字段并将它们转换为文本:
https://github.com/google/protobuf/blob/master/src/google/protobuf/text_format.cc#L1473