Messing with stuff

internal/decompress/zstd.go

@@ -1,6 +1,7 @@
 package decompress
 
 import (
+	"bytes"
 	"io"
 
 	"github.com/klauspost/compress/zstd"
@@ -12,3 +13,19 @@ func (z Zstd) Reader(src io.Reader) (io.ReadCloser, error) {
 	r, err := zstd.NewReader(src)
 	return r.IOReadCloser(), err
 }
+
+type ZstdDecodeAll struct {
+	rdr *zstd.Decoder
+}
+
+func (z *ZstdDecodeAll) Reader(src io.Reader) (io.ReadCloser, error) {
+	if z.rdr == nil {
+		z.rdr, _ = zstd.NewReader(nil)
+	}
+	data, err := io.ReadAll(src)
+	if err != nil {
+		return nil, err
+	}
+	out, err := z.rdr.DecodeAll(data, nil)
+	return io.NopCloser(bytes.NewReader(out)), err
+}