mirrors/seaweedfs
1
0
Fork 0
mirror of https://github.com/chrislusf/seaweedfs synced 2024-07-03 23:56:41 +02:00
Code Issues Projects Releases Wiki Activity

zstd fix

Browse source
This commit is contained in:
Chris Lu 2020-06-24 11:39:09 -07:00
parent 2ff37ccdbd
commit 5be12eea37
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
weed/util/compression.go
View file

@ -25,6 +25,13 @@ func GzipData(input []byte) ([]byte, error) {
} }
return buf.Bytes(), nil return buf.Bytes(), nil
} }
var zstdEncoder, _ = zstd.NewWriter(nil)
func ZstdData(input []byte) ([]byte, error) {
return zstdEncoder.EncodeAll(input, nil), nil
}
func DecompressData(input []byte) ([]byte, error) { func DecompressData(input []byte) ([]byte, error) {
if IsGzippedContent(input) { if IsGzippedContent(input) {
return ungzipData(input) return ungzipData(input)
@ -46,10 +53,9 @@ func ungzipData(input []byte) ([]byte, error) {
return output, err return output, err
} }
var zstdEncoder, _ = zstd.NewWriter(nil) var decoder, _ = zstd.NewReader(nil)
func unzstdData(input []byte) ([]byte, error) { func unzstdData(input []byte) ([]byte, error) {
return zstdEncoder.EncodeAll(input, nil), nil return decoder.DecodeAll(input, nil)
} }
func IsGzippedContent(data []byte) bool { func IsGzippedContent(data []byte) bool {
@ -63,7 +69,7 @@ func IsZstdContent(data []byte) bool {
if len(data) < 4 { if len(data) < 4 {
return false return false
} }
return data[0] == 0xFD && data[1] == 0x2F && data[2] == 0xB5 && data[3] == 0x28 return data[3] == 0xFD && data[2] == 0x2F && data[1] == 0xB5 && data[0] == 0x28
} }
/* /*