protobuf使用

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;
}

通過上面代碼,進行下語法學習:

  1. 第一行 syntax = “proto3”; 用來聲明protobuf的版本,目前有 proto2proto3 兩個版本;
  2. message 爲關鍵字,後面爲定義的model名字,大括號中爲model具體的字段;
  3. 字段定義格式爲 數據類型 字段名 = 分配標識號;
    支持的數據類型有:double、float、int32、int64、uint32、uint64、sint32、sint64、fixed32、fixed64、sfixed32、sfixed64、bool、string、bytes
  4. 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就可以得到我們傳過來的值了

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章