Added BodyInflate and WriteInflate to match BodyGunzip and WriteGunzip

1 files changed, 17 insertions, 0 deletions

diff --git a/compress.go b/compress.go
index 5b9e58d..c3b81cf 100644
--- a/compress.go
+++ b/compress.go
@@ -171,6 +171,23 @@ func WriteGunzip(w io.Writer, p []byte) (int, error) {
 	return nn, err
 }
 
+// WriteInflate writes inflated p to w and returns the number of uncompressed
+// bytes written to w.
+func WriteInflate(w io.Writer, p []byte) (int, error) {
+	r := &byteSliceReader{p}
+	zr, err := acquireFlateReader(r)
+	if err != nil {
+		return 0, err
+	}
+	n, err := copyZeroAlloc(w, zr)
+	releaseFlateReader(zr)
+	nn := int(n)
+	if int64(nn) != n {
+		return 0, fmt.Errorf("too much data inflated: %d", n)
+	}
+	return nn, err
+}
+
 // AppendGunzipBytes append gunzipped src to dst and returns the resulting dst.
 func AppendGunzipBytes(dst, src []byte) ([]byte, error) {
 	w := &byteSliceWriter{dst}