@@ -8,7 +8,6 @@ import (
|
||||
// nolint: staticcheck
|
||||
oldproto "github.com/golang/protobuf/proto"
|
||||
"github.com/segmentio/encoding/proto"
|
||||
mproto "github.com/unistack-org/micro-proto/proto"
|
||||
"github.com/unistack-org/micro/v3/codec"
|
||||
newproto "google.golang.org/protobuf/proto"
|
||||
)
|
||||
@@ -21,8 +20,6 @@ func (c *protoCodec) Marshal(v interface{}) ([]byte, error) {
|
||||
return nil, nil
|
||||
case *codec.Frame:
|
||||
return m.Data, nil
|
||||
case *mproto.Frame:
|
||||
return m.Data, nil
|
||||
case newproto.Message, oldproto.Message, proto.Message:
|
||||
return proto.Marshal(m)
|
||||
}
|
||||
@@ -39,8 +36,6 @@ func (c *protoCodec) Unmarshal(d []byte, v interface{}) error {
|
||||
return nil
|
||||
case *codec.Frame:
|
||||
m.Data = d
|
||||
case *mproto.Frame:
|
||||
m.Data = d
|
||||
case newproto.Message, oldproto.Message, proto.Message:
|
||||
err = proto.Unmarshal(d, m)
|
||||
default:
|
||||
@@ -57,13 +52,6 @@ func (c *protoCodec) ReadBody(conn io.Reader, b interface{}) error {
|
||||
switch m := b.(type) {
|
||||
case nil:
|
||||
return nil
|
||||
case *mproto.Frame:
|
||||
buf, err := ioutil.ReadAll(conn)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
m.Data = buf
|
||||
return nil
|
||||
case *codec.Frame:
|
||||
buf, err := ioutil.ReadAll(conn)
|
||||
if err != nil {
|
||||
@@ -85,9 +73,6 @@ func (c *protoCodec) Write(conn io.Writer, m *codec.Message, b interface{}) erro
|
||||
switch m := b.(type) {
|
||||
case nil:
|
||||
return nil
|
||||
case *mproto.Frame:
|
||||
_, err := conn.Write(m.Data)
|
||||
return err
|
||||
case *codec.Frame:
|
||||
_, err := conn.Write(m.Data)
|
||||
return err
|
||||
|
Reference in New Issue
Block a user