使GRPC消息JSON可序列化

2022-04-03 00:00:00 python json grpc grpc-python grpc-node

问题描述

我有一个包含重复标量值的GRPC定义。

例如


message Request {
  bool dryrun = 1;   
  bool verbose = 2;
  bool enabled = 3;
  float ratio = 4;
  int32 count = 5;
  string somevar = 6;
  bool withcondition = 7;
  bool nocondition = 8;
  float portion = 9;
  int32 population = 10;
  string xvars = 11;
  string singleoption = 12;
  repeated int32 multioptions = 13;
}

我注意到,当这个请求被发送到一个Python服务器时,mylist被转换为RepeatedScalarContainer,这是不可JSON序列化的。Some SO questions建议我使用

from google.protobuf.json_format import MessageToJson
serialized = MessageToJson(original)

将协议错误消息转换为有效的JSON对象。 但在我的例子中,对收到的Python对象执行上述操作会得到一个截断的对象

{
  "enabled": true,
  "ratio": 0.5,
  "count": 100,
  "somevar": "/input",
  "withcondition": true,
  "singleoption": "option1",
  "multioptions": [
    1,
    3
  ]
}

您可以看到缺少许多消息属性。

作为一种解决办法,我仍然可以提取接收到的对象的属性来恢复数据,并通过暴力将它们转换为可序列化的JSON,但这看起来既难看又不可靠。这是我的办法

    def handle_request(request):
        all_attrs = dir(request)
        i_last_private_attr = next((a for a, attr in enumerate(reversed(all_attrs)) if attr.startswith('_')), None)
        if not i_last_private_attr:
            _logger.error('Failed to find any request')
            return
        i_first_arg_attr = len(all_attrs) - i_last_private_attr
        req_dict = {attr: getattr(request, attr) for attr in all_attrs[i_first_arg_attr:]}
        for k, v in req_dict.items():
            if isinstance(v, Iterable) and not isinstance(v, (dict, list, tuple, str)):
                req_dict[k] = list(req_dict[k])

我应该如何使收到的原始GRPC请求像只包含内置基元值类型的Python对象一样工作?


解决方案

我自己解决了。

有一个选项including_default_value_fieldsjson_format.MessageToJson()。设置为False时,不会将所有默认字段打印为JSON,尽管它们是通过网络物理发送的。

遗憾的是,默认情况下该选项处于关闭状态。

将其设置为True后,问题消失。

相关文章