utils: binary reader, add ReadUntilFromBufioReader()

Signed-off-by: Arran Walker <arran.walker@fiveturns.org>

1 files changed, 12 insertions, 6 deletions

diff --git a/utils/binary/read.go b/utils/binary/read.go
index ac55609..12e57c3 100644
--- a/utils/binary/read.go
+++ b/utils/binary/read.go
@@ -26,12 +26,7 @@ func Read(r io.Reader, data ...interface{}) error {
 // ReadUntil reads from r untin delim is found
 func ReadUntil(r io.Reader, delim byte) ([]byte, error) {
 	if bufr, ok := r.(*bufio.Reader); ok {
-		value, err := bufr.ReadBytes(delim)
-		if err != nil || len(value) == 0 {
-			return nil, err
-		}
-
-		return value[:len(value)-1], nil
+		return ReadUntilFromBufioReader(bufr, delim)
 	}
 
 	var buf [1]byte
@@ -53,6 +48,17 @@ func ReadUntil(r io.Reader, delim byte) ([]byte, error) {
 	}
 }
 
+// ReadUntilFromBufioReader is like bufio.ReadBytes but drops the delimiter
+// from the result.
+func ReadUntilFromBufioReader(r *bufio.Reader, delim byte) ([]byte, error) {
+	value, err := r.ReadBytes(delim)
+	if err != nil || len(value) == 0 {
+		return nil, err
+	}
+
+	return value[:len(value)-1], nil
+}
+
 // ReadVariableWidthInt reads and returns an int in Git VLQ special format:
 //
 // Ordinary VLQ has some redundancies, example:  the number 358 can be