make sure zstd decomp closes (#25)

* make sure zstd decomp closes * remove debugging tools
axiomhq · Nov 11, 2021 · 4b2bd96 · 4b2bd96
1 parent c47dfe4
commit 4b2bd96
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/proxy/proxy.go b/proxy/proxy.go
@@ -37,6 +37,15 @@ func init() {
 	}
 }
 
+type zstdCloser struct {
+	*zstd.Decoder
+}
+
+func (zc *zstdCloser) Close() error {
+	zc.Decoder.Close()
+	return nil
+}
+
 func Decompress(rdr io.ReadCloser, encoding string) (io.ReadCloser, error) {
 	switch encoding {
 	case "gzip":
@@ -51,7 +60,8 @@ func Decompress(rdr io.ReadCloser, encoding string) (io.ReadCloser, error) {
 		if err != nil {
 			return nil, err
 		}
-		return io.NopCloser(decomp), nil
+		// else we leak
+		return &zstdCloser{Decoder: decomp}, nil
 	default:
 		return rdr, nil
 	}