protobuf及grpc的client请求

背景：
需要对rpc服务进行压测，需要构造rpc请求。

protobuf 简介
Protocol Buffer (简称Protobuf) 是Google出品的性能优异、跨语言、跨平台的序列化库。

文档结构
protobuf 使用 .proto 文件来保存文档。

syntax = "proto3";
package cw.videoanalyze.alg_plugin;

message AlgPluginObject {
    string plugin_name = 1;             // 插件名称(必须要填写)
    uint64 UOID = 2;                    // UOID (非必须)
    string plugin_func = 3;             // 插件方法
}

message AlgPluginRequest {
    AlgPluginObject plugin = 1;         // 插件对象
    bytes serialize_data = 2;           // 序列化数据（无关数据）
}


message AlgPluginResponse {
    AlgPluginObject plugin = 1;         // 插件对象
    bool ret_flag = 2;                  // 响应结果
    int32 error_no = 3;                 // 错误码，当ret_flag = false 时有用
    string message = 4;                 // 返回消息
    bytes serialize_data = 5;           // 返回的序列化数据
}

/*
 * 插件上报数据
 * 说明: 该数据中包换了插件上报的所有的图片，抓拍坐标，属性等数据
 */
message AlgPluginReport {
    AlgPluginObject plugin = 1;         // 插件对象
    string camera_id = 2;               // 摄像头ID
    bytes serialize_data = 3;           // 序列化数据（无关数据）
}

/*
 * 插件上报RPC接口
 */
service AlgReportPush {
    rpc Report (AlgPluginReport) returns (AlgPluginResponse);
    rpc Request (AlgPluginRequest) returns (AlgPluginResponse);
}

protobuf 文档的第一行，为版本申明，不填写默认为版本2.
package 定义proto的包名，包名可以避免对message 类型之间的名字冲突。也可省略。
message 关键字定义消息。
service 关键字定义服务。

定义消息
protobuf使用message定义消息，例如：

message AlgPluginObject {
    string plugin_name = 1;             // 插件名称(必须要填写)
    uint64 UOID = 2;                    // UOID (非必须)
    string plugin_func = 3;             // 插件方法
}

AlgPluginObject消息定义了三个字段，分别为插件名称，U0ID及插件方法。
消息定义中的每个字段都有唯一的编号。这些字段编号用于以消息二进制格式标识字段，并且在使用消息类型后不应更改

定义服务
定义RPC方法，需要使用service关键字定义服务，并在里面定义方法。

/*
 * 插件上报RPC接口
 */
service AlgReportPush {
    rpc Report (AlgPluginReport) returns (AlgPluginResponse);
    rpc Request (AlgPluginRequest) returns (AlgPluginResponse);
}

此处定义了一个AlgReportPush服务，包含两个rpc方法：

Reprot 方法，入参为AlgPluginReport，返回参数为 AlgPluginResponse
Request 方法，入参为AlgPluginRequest，返回参数为 AlgPluginResponse

下面针对上面的proto文件，写一个grpc client 进行rpc请求。

import grpc
import va_plugin_proto_pb2_grpc
import va_plugin_proto_pb2
import VideoObjectPush_pb2

def get_bytes():
    img = 'D:\\10w\\1.jpg'
    with open(img,'rb') as f:
        bytes_data = f.read()
    return bytes_data

def run():
    # 连接RPC服务器
    channel = grpc.insecure_channel('localhost:21001')
    # 调用RPC服务
    stub = va_plugin_proto_pb2_grpc.AlgReportPushStub(channel=channel)
    #组装请求参数
    alg_plugin_obj = va_plugin_proto_pb2.AlgPluginObject(plugin_name='videostream.person',plugin_func='videostream.person.return_callback')
    box = VideoObjectPush_pb2.Box(x=1,y=1,width=1,height=1)
    face = VideoObjectPush_pb2.Object(box=box,objPic=get_bytes(),qualityScore=0.656565,confidenceScore=0.13265,
                                      pts=1546272000000)
    person = VideoObjectPush_pb2.Person(objectId=1,cameraId="xiuxiqu",face=face)
    object = VideoObjectPush_pb2.DetectedObject(person=person)

    resp = stub.Report(va_plugin_proto_pb2.AlgPluginReport(plugin=alg_plugin_obj,camera_id='xiuxiqu',serialize_data=object.SerializeToString()))
    print(resp)

protobuf及grpc的client请求

你可能感兴趣的:(protobuf及grpc的client请求)