Merge pull request #193 from klauspost/use-pure-go-zstd

Use pure Go Zstandard
xitongsys · Dec 16, 2019 · 706116f · 706116f
2 parents adca8ee + 6b151ce
commit 706116f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/compress/gzip.go b/compress/gzip.go
@@ -4,7 +4,7 @@ package compress
 
 import (
 	"bytes"
-	"compress/gzip"
+	"github.com/klauspost/compress/gzip"
 	"github.com/xitongsys/parquet-go/parquet"
 	"io/ioutil"
 )

diff --git a/compress/zstd.go b/compress/zstd.go
@@ -3,18 +3,20 @@
 package compress
 
 import (
-	"github.com/DataDog/zstd"
+	"github.com/klauspost/compress/zstd"
 	"github.com/xitongsys/parquet-go/parquet"
 )
 
 func init() {
+	// Create encoder/decoder with default parameters.
+	enc, _ := zstd.NewWriter(nil, zstd.WithZeroFrames(true))
+	dec, _ := zstd.NewReader(nil)
 	compressors[parquet.CompressionCodec_ZSTD] = &Compressor{
 		Compress: func(buf []byte) []byte {
-			res, _ := zstd.Compress(nil, buf)
-			return res
+			return enc.EncodeAll(buf, nil)
 		},
 		Uncompress: func(buf []byte) (bytes []byte, err error) {
-			return zstd.Decompress(nil, buf)
+			return dec.DecodeAll(buf, nil)
 		},
 	}
 }