diff --git a/codec/json/json.go b/codec/json/json.go index 42de4a4d..5af47a07 100644 --- a/codec/json/json.go +++ b/codec/json/json.go @@ -5,6 +5,8 @@ import ( "encoding/json" "io" + "github.com/golang/protobuf/jsonpb" + "github.com/golang/protobuf/proto" "github.com/micro/go-micro/codec" ) @@ -22,6 +24,9 @@ func (c *Codec) ReadBody(b interface{}) error { if b == nil { return nil } + if pb, ok := b.(proto.Message); ok { + return jsonpb.UnmarshalNext(c.Decoder, pb) + } return c.Decoder.Decode(b) } diff --git a/codec/json/marshaler.go b/codec/json/marshaler.go index b9d0be28..a13e5f39 100644 --- a/codec/json/marshaler.go +++ b/codec/json/marshaler.go @@ -2,6 +2,9 @@ package json import ( "encoding/json" + + "github.com/golang/protobuf/jsonpb" + "github.com/golang/protobuf/proto" ) type Marshaler struct{} @@ -11,6 +14,9 @@ func (j Marshaler) Marshal(v interface{}) ([]byte, error) { } func (j Marshaler) Unmarshal(d []byte, v interface{}) error { + if pb, ok := v.(proto.Message); ok { + return jsonpb.UnmarshalString(string(d), pb) + } return json.Unmarshal(d, v) }