Grpc-Gateway流式响应解码失败问题及解决方案
本文分析并解决使用Grpc-Gateway处理HTTP请求的流式响应解码失败问题。现象是:流式响应体打印正确,但runtime.jsonpb.Decode
返回nil
。
根本原因在于Grpc-Gateway版本过低和Proto文件定义与实际返回数据结构不符。旧代码直接将响应解码为pb.resp
类型,而Grpc-Gateway实际返回包含result
字段的JSON数据,导致解码失败。
解决方案:三步走
升级Grpc-Gateway: 将github.com/grpc-ecosystem/grpc-gateway
升级到最新版本(例如v2.4.0或更高)。低版本可能存在兼容性问题或bug。
调整Proto文件定义: 由于响应包含result
字段,需要在Proto文件中添加新的message来匹配。修改后的Proto文件如下:
message resp { int32 code = 1; string msg = 2; } message httpresp { resp result = 1; }
httpresp
message精确匹配Grpc-Gateway的JSON响应结构。
pb.httpresp
类型接收结果,并正确使用地址符&
。修改后的解码代码:jsonb := new(runtime.JSONPb) dencoder := jsonb.NewDecoder(resp.Body) for { var result *pb.Httpresp // 使用pb.Httpresp类型 err := dencoder.Decode(&result) // 使用地址符& if err == nil { t.Logf("resp: %+v", result) } else { t.Logf("%+v", err) break } }
通过以上步骤,即可成功解析流式响应中的每个httpresp
消息。修改后的单元测试代码将正确打印每个消息内容,验证问题已解决。