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