diff --git a/codec/proto/proto.go b/codec/proto/proto.go index b70b4c84..d79d5166 100644 --- a/codec/proto/proto.go +++ b/codec/proto/proto.go @@ -18,13 +18,13 @@ func (c *Codec) ReadHeader(m *codec.Message, t codec.MessageType) error { } func (c *Codec) ReadBody(b interface{}) error { + if b == nil { + return nil + } buf, err := ioutil.ReadAll(c.Conn) if err != nil { return err } - if b == nil { - return nil - } return proto.Unmarshal(buf, b.(proto.Message)) }