python 處理protobuf協議

背景：需要用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 import openrtb_pb2
from google.protobuf.json_format import MessageToJson, Parse

# 將pbStringRequest【protobuf string類型的請求body】轉化為json string以便解析請求中的某個指定參數
def pb_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)
　　return jsonStringRequest } # 將jsonStringResponse轉化為pbString返回 

def json_to_pb(jsonStringResponse){ 　　
　　bidResponse = openrtb_pb2.BidResponse() 　　
　　Parse(jsonStringResponse, bidResponse) 　　
　　pbStringResponse = bidResponse.SerializeToString() 　　
　　return pbStringResponse }