add compatibility with flate.Reader

go-stomp · Mar 9, 2021 · 2796331 · 2796331
1 parent 0036ca2
commit 2796331
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/message.go b/message.go
@@ -57,3 +57,12 @@ func (msg *Message) Read(p []byte) (int, error) {
 	msg.Body = msg.Body[n:]
 	return n, nil
 }
+
+func (msg *Message) ReadByte() (byte, error) {
+	if len(msg.Body) == 0 {
+		return 0, io.EOF
+	}
+	n := msg.Body[0]
+	msg.Body = msg.Body[1:]
+	return n, nil
+}