golang:如何在proto3中用上golang对应的interface{}类型

首先,我希望所有golang中用于http请求响应的结构,都使用proto3来定义。
麻烦的是,有的情况下某个字段的类型可能是动态的,对应的JSON类型可能是number/string/boolean/null中的其中一种。

一开始我尝试用proto.Any类型,就像这样:

import "google/protobuf/any.proto";

message MyRequest{
    google.protobuf.Any user_input = 1;  // 用户可能输入 number / string / boolean / null 中的其中一种
}

使用protoc生成代码后,发现这玩意儿完全没办法做json的encode/decode。
理想的办法是让生成golang代码中的 user_input 成为 interface{} 类型。但如何才能让proto3生成golang的interface类型呢?

尝试后发现可以用下面的办法解决:

使用gogo proto的扩展语法

import "google/protobuf/descriptor.proto";
import "github.com/gogo/protobuf/gogoproto/gogo.proto";

message MyRequest{
    bytes user_input = 1[(gogoproto.customtype) = "InterfaceType"];  // 使用一个叫做 InterfaceType 的自定义类型
}

注意:InterfaceType 直接写成 interface{} 是不行的。因为 interface{} 类型没有实现序列化的接口。

执行protoc后生成了如下代码:

type MyRequest struct {
	UserInput []InterfaceType `protobuf:"bytes,4,rep,name=user_input,proto3,customtype=InterfaceType" json:"user_input,omitempty"`
}

编写 InterfaceType 类型对应的序列化代码

// interface_type.go
// 放在xxx.pb.go的同一目录下
package proto

import (
	"encoding/json"
	"errors"
)

type InterfaceType struct {
	Value interface{}
}

func (t InterfaceType) Marshal() ([]byte, error) {
	return nil, errors.New("not implement")
}
func (t *InterfaceType) MarshalTo(data []byte) (n int, err error) {
	return 0, errors.New("not implement")
}
func (t *InterfaceType) Unmarshal(data []byte) error {
	return errors.New("not implement")
}
func (t *InterfaceType) Size() int {
	return -1
}

// 因为只做JSON的序列化,所以只实现这两个方法就行了
func (t InterfaceType) MarshalJSON() ([]byte, error) {
	return json.Marshal(t.Value)
}
func (t *InterfaceType) UnmarshalJSON(data []byte) error {
	return json.Unmarshal(data, &t.Value)
}

测试一下

// my_request.pb_test.go
package proto

import (
	"encoding/json"
	"testing"
)

func Test_MyRequest(t *testing.T) {
	j := `{"user_input":123}`
	inst := &MyRequest{}
	err := json.Unmarshal([]byte(j), inst)
	if err != nil {
		t.Errorf("json decode error, err=%+v", err)
		return
	}
	t.Logf("%+v", MyRequest)
	str, err := json.Marshal(inst)
	if err != nil {
		t.Errorf("json encode error, err=%+v", err)
		return
	}
	t.Logf("json=%s", string(str))
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值