为了账号安全,请及时绑定邮箱和手机立即绑定

解组 GRPC 对通用类型的响应问题

解组 GRPC 对通用类型的响应问题

Go
红糖糍粑 2022-12-05 16:21:46
我正在尝试使用泛型将 JSON 对象转换为具有枚举的 GRPC 响应,即:type GRPCResponse {    str string    enu EnumType}type EnumType int32const (    Type1 EnumType = 0    Type2 EnumType = 1)解组函数如下所示:func assertHTTPResponseOK[T any](t *testing.T, endpoint string) T {    body, err := GetResponse(endpoint)    var v T    err := json.Unmarshal(body, &v)    require.Nil(t, err)    return v}调用它的代码如下所示:assertHTTPResponseOK[*GRPCResponse](t, "some-endpoint")有问题的 JSON 对象如下所示:{"str":"hello", "enu": "Type2"}我收到以下错误:json: cannot unmarshal string into Go struct field GRPCResponse.enu of type EnumType从类似的问题中,我看到通常的建议是使用jsonpb.Unmarshalorprotojson.Unmarshal而不是典型的json.Unmarshal.在更改 Unmarshal 函数时,我还必须将 T 更改为protoreflect.ProtoMessage. 但是,这会阻止我将指针传递v给Unmarshal,因为它是指向接口的指针,而不是接口。当然我也不能传入一个nil指针(不取v的地址)。所以我的问题是:有没有办法让这个通用对象的指针满足接口protoreflect.ProtoMessage?是否有更好的 Unmarshalling 函数更适合我的问题?
查看完整描述

2 回答

?
慕丝7291255

TA贡献1859条经验 获得超6个赞

我最终传递了我正在解组的对象。


obj := new(GRPCResponse)

assertHTTPResponseOK[*GRPCResponse](t, ctx, "some-endpoint", obj)

func assertHTTPResponseOK[T protoreflect.ProtoMessage](t *testing.T, ctx context.Context, endpoint string, object T) {

    body, err := GetResponse(endpoint)

    require.Nil(t, err)


    err = protojson.Unmarshal(body, object)

    require.Nil(t, err)

}


查看完整回答
反对 回复 2022-12-05
?
ITMISS

TA贡献1871条经验 获得超8个赞

这是一个泛型友好的原型解组器,它避免传递第二个类型,代价是反射调用以查看指针内部的类型并调用它的构造函数。


        var msg T // Constrained to proto.Message


        // Peek the type inside T (as T= *SomeProtoMsgType)

        msgType := reflect.TypeOf(msg).Elem()


        // Make a new one, and throw it back into T

        msg = reflect.New(msgType).Interface().(T)


        errUnmarshal := proto.Unmarshal(body, msg)


查看完整回答
反对 回复 2022-12-05
  • 2 回答
  • 0 关注
  • 142 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信