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就可以得到我们传过来的值了

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