Merge pull request #92 from moredure/patch-1

add compatibility with flate.Reader
go-stomp · Mar 13, 2021 · 1509067 · 1509067
2 parents 0036ca2 + 2796331
commit 1509067
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
+}