refactor: add io.EOF check

DarthPestilane · Jan 30, 2024 · af8cc9e · af8cc9e
1 parent 7ec026f
commit af8cc9e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/packer.go b/packer.go
@@ -83,6 +83,9 @@ func (d *DefaultPacker) Unpack(reader io.Reader) (*Message, error) {
 	id := d.bytesOrder().Uint32(headerBuffer[4:8])
 	data := make([]byte, dataSize)
 	if _, err := io.ReadFull(reader, data); err != nil {
+		if err == io.EOF {
+			return nil, err
+		}
 		return nil, fmt.Errorf("read data err: %s", err)
 	}
 	return NewMessage(int(id), data), nil