unistack-org/micro
3
0
Fork 3
Code Issues 14 Pull Requests 2 Actions Packages Projects Releases 242 Wiki Activity

Don't read unless we have b

Browse Source
This commit is contained in:
Asim Aslam 2019-01-17 12:09:04 +00:00
parent 617db003d4
commit 9ce9977d21
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
codec/proto/proto.go
View File

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