背景:需要用django基于python3模拟一个http接口,请求是post方式,body是protobuf string,返回也是protobuf string
设计:django获取pb string的post body反序列化成json,通过json里面的指定字段的值获取对应的返回json, 并将返回的json序列化成pb string
重点:需要用python实现基于protobuf协议将json转化为pb string,将pb string转成json
步骤:
1.安装protobuf
安装protobuf,可直接下载编译后的压缩包直接解压后配置环境变量使用,我这里使用源码包安装方式【mac or linux】
下载源码包 https://github.com/protocolbuffers/protobuf/release
直接点击下载图示包,或者右键复制下载链接再通过wget下载
解压并安装
tar -zxvf protobug-all-3.6.1.tar.gz
cd protobuf-3.6.1 && ./configure && make && make check && make install
执行完毕之后通过命令 protoc --version校验是否安装成功,能显示版本则表示安装成功
2.python安装protobuf,直接通过pip安装
pip3 install protobuf
安装完成之后能成功导入 google.protobuf表示成功
3.根据协议生成python文件
将protobuf协议原文件openrtb.proto放在example目录包下,并执行命令则会生成对应的py文件openrtb_pb2.py
protoc --python_out=. openrtb.proto
4.实现pb和json的相互转换
from example importopenrtb_pb2from google.protobuf.json_format importMessageToJson, Parse#将pbStringRequest【protobuf string类型的请求body】转化为json string以便解析请求中的某个指定参数
defpb_to_json(pbStringRequest){
req=openrtb_pb2.BidRequest()
req.ParseFromString(pbStringRequest)
# preserveing_proto_field_name 设置为 True 可以保留 protobuf 的原有字段名,不然会自动转驼峰,如 request_id 会被自动转化为 requestId
jsonStringRequest=MessageToJson(req,preserving_proto_field_name=True)returnjsonStringRequest }#将jsonStringResponse转化为pbString返回
defjson_to_pb(jsonStringResponse){
bidResponse=openrtb_pb2.BidResponse()
Parse(jsonStringResponse, bidResponse)
pbStringResponse=bidResponse.SerializeToString()returnpbStringResponse }