Add gzipz package.

gzipz/gzipz.go

@@ -0,0 +1,38 @@
+package gzipz
+
+import (
+ "bytes"
+ "compress/gzip"
+ "io/ioutil"
+
+ "github.com/ibrt/golang-bites/internal"
+)
+
+// MustCompress implements in-memory GZIP compression.
+func MustCompress(buf []byte) []byte {
+ cBuf := &bytes.Buffer{}
+ w, err := gzip.NewWriterLevel(cBuf, gzip.BestCompression)
+ internal.MaybePanic(err)
+ defer func() {
+ internal.MaybePanic(w.Close())
+ }()
+
+ _, err = w.Write(buf)
+ internal.MaybePanic(err)
+ internal.MaybePanic(w.Close())
+
+ return cBuf.Bytes()
+}
+
+// MustDecompress implements in-memory GZIP decompression.
+func MustDecompress(buf []byte) []byte {
+ r, err := gzip.NewReader(bytes.NewReader(buf))
+ internal.MaybePanic(err)
+ defer func() {
+ internal.MaybePanic(r.Close())
+ }()
+
+ dBuf, err := ioutil.ReadAll(r)
+ internal.MaybePanic(err)
+ return dBuf
+}

gzipz/gzipz_test.go

@@ -0,0 +1,18 @@
+package gzipz_test
+
+import (
+ "strings"
+ "testing"
+
+ "github.com/stretchr/testify/require"
+
+ "github.com/ibrt/golang-bites/gzipz"
+)
+
+func TestCompress(t *testing.T) {
+ buf := []byte(strings.Repeat("x", 1000))
+ gBuf := gzipz.MustCompress(buf)
+ require.Less(t, len(gBuf), len(buf))
+ uBuf := gzipz.MustDecompress(gBuf)
+ require.Equal(t, buf, uBuf)
+}