fix nil buf check

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>

codec_test.go

@@ -0,0 +1,19 @@
+package proto
+
+import (
+	"bytes"
+	"testing"
+)
+
+func TestReadBody(t *testing.T) {
+	t.Skip("skip as no proto")
+	s := &struct {
+		Name string
+	}{}
+	c := NewCodec()
+	b := bytes.NewReader(nil)
+	err := c.ReadBody(b, s)
+	if err != nil {
+		t.Fatal(err)
+	}
+}

proto.go

@@ -24,7 +24,7 @@ func (c *protoCodec) Marshal(v interface{}) ([]byte, error) {
 }
 
 func (c *protoCodec) Unmarshal(d []byte, v interface{}) error {
-	if d == nil {
+	if len(d) == 0 {
 		return nil
 	}
 	switch m := v.(type) {
@@ -50,6 +50,8 @@ func (c *protoCodec) ReadBody(conn io.Reader, b interface{}) error {
 		buf, err := ioutil.ReadAll(conn)
 		if err != nil {
 			return err
+		} else if len(buf) == 0 {
+			return nil
 		}
 		m.Data = buf
 		return nil
@@ -57,7 +59,7 @@ func (c *protoCodec) ReadBody(conn io.Reader, b interface{}) error {
 		buf, err := ioutil.ReadAll(conn)
 		if err != nil {
 			return err
-		} else if buf == nil {
+		} else if len(buf) == 0 {
 			return nil
 		}
 		return proto.Unmarshal(buf, m)