使用protobuf的反射来动态生成Message并赋值

C++ 专栏收录该内容
1 篇文章 0 订阅

前言:

公司做了个项目,使用protobuf,各功能号会不断增加,message也会一直变,我做为测试人员,需要做一个测试工具,前期可以通过mfc画图,就是画出message的各个属性对应的输入框,之后按照格式,遍历对应的message,读取输入框的数据来一个个赋值。这样一来,功能号少的时候,还没什么,多画点输入框就是了。后来,message嵌套越来越复杂,其中还夹杂着repeated的属性,repeated中还有repeated、子message,直接崩溃。于是就决心查资料做了这个动态生成的,用CMarku从xml读取数据,感觉挺好用,留个档,也希望对其他人能有个帮助。

预期读者:

对protobuf有一定了解,至少已经跑通过程序,只是遇到了和我类似的问题,现在要使用protobuf的动态生成message并赋值、发送消息给服务端。

理论方面,可以查看大牛的http://blog.csdn.net/solstice/article/details/6300108介绍,说的很细。

正文:

如果我在这里大谈理论,可能就是班门弄斧了,那么废话不多说,直接上代码,上核心代码。

根据message name来生成message


Message* createMessage(const std::string& typeName)
{
    google::protobuf::Message* message = NULL;
    const google::protobuf::Descriptor* descriptor = google::protobuf::DescriptorPool::generated_pool()->FindMessageTypeByName(typeName);
    if (descriptor)
    {
        const google::protobuf::Message* prototype = google::protobuf::MessageFactory::generated_factory()->GetPrototype(descriptor);
        if (prototype)
        {
            message = prototype->New();
        }
    }
    return message;
}

typeName就是下面定义的 Role

message Role
{  
  required int32 RoleId= 1;   
  required bytes RoleName= 2;

  required string Status= 3; 

repeated Permission p=4;

}

message Permission

{

required id=1;

required pname=XXX;

}

总结一下,我是这么理解的,如要根据message name动态生成message,先生成message的descriptor,再根据descriptor生成Message类,类再new一个message对象,即可使用。有理解的不对的地方,请大家指教。关于message中的属性项,如上面Role的RoleId、RoleName、Status,也是一个流程,只是descriptor变成了field 的descriptor等。

由于message内部属性的各种嵌套repeated message,而要给一个message对象的属性赋值,使用Role的反射只能赋值一层,比如上面例子Role,只能给message  ROle 的RoleId、RoleName、Status、p赋值,但是不能为Permission的id、pname赋值,如果要给id、pname赋值,需要使用Permission的反射,道理同Role。这里还涉及到一个repeated,先说反射,再来说这个repeated的处理。

pMsg=createMessage(packageName);
                pDescriptor = google::protobuf::DescriptorPool::generated_pool()->FindMessageTypeByName(packageName);
                pReflection = pMsg->GetReflection();

上面的pMsg、pDescriptor、pReflection在嵌套赋值的使用要用到,先不考虑冗余,暂且弄个

struct returnStruct
{
    const google::protobuf::Descriptor* pDescriptor;
    const google::protobuf::Reflection* pReflection;
    google::protobuf::Message* pMsg;
};

struct returnStruct re={pDescriptor,pReflection,pMsg}

参照后面的代码,可能会好理解一点。大概说一下,把pDescriptor,pReflection,pMsg搜集起来,在对Role属性赋值的时候,带上这个returnStruct,这样在子message Permission的赋值处理时候,Permission赋值完毕,可以直接使用returnStruct的Permission的父级Role的反射来把Permission对象赋值给Role的p,递归的时候特别好用。

pFieldDescriptor=re->pDescriptor->FindFieldByName(tag_name);//tag_name 是属性项的名字,就是Role中的RoleId、RoleName之类。

        if(pFieldDescriptor->is_repeated()){//考虑到googel对protobuf 中属性为repeated的赋值接口不同,这里分开
                switch(pFieldDescriptor->type())
                {
                case google::protobuf::FieldDescriptor::TYPE_DOUBLE:
                    re->pReflection->AddDouble(re->pMsg,pFieldDescriptor,atof(tag_value.c_str()));
                    break;
                case google::protobuf::FieldDescriptor::TYPE_FLOAT:
                    re->pReflection->AddFloat(re->pMsg,pFieldDescriptor,atof(tag_value.c_str()));
                    break;

                  .........

                 case google::protobuf::FieldDescriptor::TYPE_MESSAGE://嵌套的message的处理
                    {
                        pDescriptor = google::protobuf::DescriptorPool::generated_pool()->FindMessageTypeByName(packageName);
                        if(xml.IntoElem())   //xml是CMarkUP的xml当前节点的读取位置,xml数据格式和message.proto文件一模一样,当前情况相当于对Role的前三项赋值完毕,来处理permission项
                        {   
                            pMsg=re->pReflection->AddMessage(re->pMsg,pFieldDescriptor);//注意这里是AddMessage,和下面的MutableMessage对应,处理repeated类型的
                            pReflection = pMsg->GetReflection();
                            struct returnStruct ret={pDescriptor,pReflection,pMsg};
                            parseXml(xml,pTestLogger,"",&ret);    
                            xml.OutOfElem();
                        }
                        break;
                    }

                }

}else //属性项不是repeated的处理

{

//这里主要就是个接口不同

....

pMsg=re->pReflection->MutableMessage(re->pMsg,pFieldDescriptor);//注意这里是MutableMessage

....

}


今天股票大跌,我已经么有心思再多写什么了,有什么问题,可以问,吃面吧

  • 2
    点赞
  • 0
    评论
  • 4
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值