使GRPC消息JSON可序列化
问题描述
我有一个包含重复标量值的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_fields
和json_format.MessageToJson()
。设置为False
时,不会将所有默认字段打印为JSON,尽管它们是通过网络物理发送的。
遗憾的是,默认情况下该选项处于关闭状态。
将其设置为True
后,问题消失。
相关文章