Messing with stuff

This commit is contained in:
Caleb Gardner
2022-06-19 16:25:50 -05:00
parent 981f1697ab
commit 1b934de04d
2 changed files with 17 additions and 3 deletions
+17
View File
@@ -1,6 +1,7 @@
package decompress package decompress
import ( import (
"bytes"
"io" "io"
"github.com/klauspost/compress/zstd" "github.com/klauspost/compress/zstd"
@@ -12,3 +13,19 @@ func (z Zstd) Reader(src io.Reader) (io.ReadCloser, error) {
r, err := zstd.NewReader(src) r, err := zstd.NewReader(src)
return r.IOReadCloser(), err 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
}
-3
View File
@@ -61,9 +61,6 @@ func (r *Reader) Read(p []byte) (n int, err error) {
p[n+i] = tmp[i] p[n+i] = tmp[i]
} }
n += tmpN n += tmpN
}
if err != nil {
} }
return return
} }