0 背景
先交代下背景,說明下本案例在項目中確實可以使用。
我這對外提供了一個fcat/a這樣的接口,接口中對數據進行處理然後下發到fcat/b中,開始使用json傳輸數據,而從開始下發到fcat/b接收到的過程用了7s,針對這個耗時操作選用protobuf嘗試進行優化。
1 安裝protobuf編譯環境
我使用的是Mac,直接使用brew安裝即可
brew install protobuf
2 protobuf語法
首先,創建一個.proto的文件,裏面進行我們對象的定義:
syntax = "proto3";
message SplitBean {
float ratio = 1;
float x = 2;
float y = 3;
repeated float img_arr = 4;
}
通過上面代碼,進行下語法學習:
- 第一行 syntax = “proto3”; 用來聲明protobuf的版本,目前有 proto2 和 proto3 兩個版本;
- message 爲關鍵字,後面爲定義的model名字,大括號中爲model具體的字段;
- 字段定義格式爲 數據類型 字段名 = 分配標識號;
支持的數據類型有:double、float、int32、int64、uint32、uint64、sint32、sint64、fixed32、fixed64、sfixed32、sfixed64、bool、string、bytes - repeated 表示該字段爲數組,不過proto的似乎不支持多維數組
3 protobuf的編譯
model定義完後還需要編譯,使用下面命令可以完成編譯
protoc proto文件路徑 –python_out=輸出路徑
4 protobuf使用
4.1 protobuf 賦值
以上面定義的SplitBean爲例
split = split_pb2.SplitBean()
split_bean.ratio = 1.0
split_bean.x = 1.0
split_bean.y = 1.0
# 這句會報錯
split_bean.img_arr = arr
# 可以通過以下方式賦值
for item in arr:
split_bean.img_arr.append(item)
protobuf裏的數組直接賦值會報錯,需要把每個元素add進去。
賦值完成後可以通過調用**SerializeToString()**進行傳值
request_data = split_bean.SerializeToString()
async with aiohttp.ClientSession() as session:
async with session.post(root_url + "fcat/b/" data=request_data) as response:
4.2 protobuf取值
上面已經把protobuf放入請求中,要完成protobuf的使用,我們還需要從中取值:
data = await request.get_data()
split_bean = split_pb2.SplitBean()
split_bean.ParseFromString(data)
這樣通過split_bean就可以得到我們傳過來的值了