Update compress.go (valyala#978)

Add compatibility with flate.Reader to reduce allocations of bufio.NewReader structs backed by default size slices.
shihuo-cn · Feb 24, 2021 · 0880335 · 0880335
1 parent e7294d2
commit 0880335
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/compress.go b/compress.go
@@ -342,6 +342,15 @@ func (r *byteSliceReader) Read(p []byte) (int, error) {
 	return n, nil
 }
 
+func (r *byteSliceReader) ReadByte() (byte, error) {
+	if len(r.b) == 0 {
+		return 0, io.EOF
+	}
+	n := r.b[0]
+	r.b = r.b[1:]
+	return n, nil
+}
+
 func acquireStacklessDeflateWriter(w io.Writer, level int) stackless.Writer {
 	nLevel := normalizeCompressLevel(level)
 	p := stacklessDeflateWriterPoolMap[nLevel]