mirror of
https://github.com/fhmq/hmq.git
synced 2026-09-01 15:24:53 +00:00
Plugins support (#46)
* modify * update * add acl * add feature * update dockerfile * add deploy * update * update * plugins * plugins * update * update * update * fixed * remove * fixed * add log * update * fixed * update * fix config * add http api * add http api * resp * add config for work chan * update * fixed * update * disable trace * fixed * change acl * fixed * fixed res * dd * dd * ddd * dd * update * fixed * update * add * fixed * update key * add log * update * format * update * update auth * update * update readme * added * update * fixed * fixed * fix * upade * update * update
This commit is contained in:
+63
@@ -0,0 +1,63 @@
|
||||
package sarama
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"compress/gzip"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"sync"
|
||||
|
||||
"github.com/eapache/go-xerial-snappy"
|
||||
"github.com/pierrec/lz4"
|
||||
)
|
||||
|
||||
var (
|
||||
lz4ReaderPool = sync.Pool{
|
||||
New: func() interface{} {
|
||||
return lz4.NewReader(nil)
|
||||
},
|
||||
}
|
||||
|
||||
gzipReaderPool sync.Pool
|
||||
)
|
||||
|
||||
func decompress(cc CompressionCodec, data []byte) ([]byte, error) {
|
||||
switch cc {
|
||||
case CompressionNone:
|
||||
return data, nil
|
||||
case CompressionGZIP:
|
||||
var (
|
||||
err error
|
||||
reader *gzip.Reader
|
||||
readerIntf = gzipReaderPool.Get()
|
||||
)
|
||||
if readerIntf != nil {
|
||||
reader = readerIntf.(*gzip.Reader)
|
||||
} else {
|
||||
reader, err = gzip.NewReader(bytes.NewReader(data))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
defer gzipReaderPool.Put(reader)
|
||||
|
||||
if err := reader.Reset(bytes.NewReader(data)); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return ioutil.ReadAll(reader)
|
||||
case CompressionSnappy:
|
||||
return snappy.Decode(data)
|
||||
case CompressionLZ4:
|
||||
reader := lz4ReaderPool.Get().(*lz4.Reader)
|
||||
defer lz4ReaderPool.Put(reader)
|
||||
|
||||
reader.Reset(bytes.NewReader(data))
|
||||
return ioutil.ReadAll(reader)
|
||||
case CompressionZSTD:
|
||||
return zstdDecompress(nil, data)
|
||||
default:
|
||||
return nil, PacketDecodingError{fmt.Sprintf("invalid compression specified (%d)", cc)}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user