mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-07 02:20:25 +08:00
005e27a63c
* 取消用户设置的压缩级别，现在压缩级别通过系统自动设置 * Pool中的对象命中100万次时自动销毁，避免内存泄漏 * 降低Pool中的对象数量，避免占用太多内存 * 根据系统CPU线程数自动计算压缩级别，避免消耗太多CPU * zstd限制解码的最大Window * zstd使用低内存模式
46 lines
853 B
Go
46 lines
853 B
Go
// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
|
|
|
|
package compressions
|
|
|
|
import (
|
|
"github.com/klauspost/compress/zstd"
|
|
"io"
|
|
)
|
|
|
|
type ZSTDReader struct {
|
|
BaseReader
|
|
|
|
reader *zstd.Decoder
|
|
}
|
|
|
|
func NewZSTDReader(reader io.Reader) (Reader, error) {
|
|
return sharedZSTDReaderPool.Get(reader)
|
|
}
|
|
|
|
func newZSTDReader(reader io.Reader) (Reader, error) {
|
|
r, err := zstd.NewReader(reader, zstd.WithDecoderMaxWindow(256<<20))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &ZSTDReader{
|
|
reader: r,
|
|
}, nil
|
|
}
|
|
|
|
func (this *ZSTDReader) Read(p []byte) (n int, err error) {
|
|
return this.reader.Read(p)
|
|
}
|
|
|
|
func (this *ZSTDReader) Reset(reader io.Reader) error {
|
|
return this.reader.Reset(reader)
|
|
}
|
|
|
|
func (this *ZSTDReader) RawClose() error {
|
|
this.reader.Close()
|
|
return nil
|
|
}
|
|
|
|
func (this *ZSTDReader) Close() error {
|
|
return this.Finish(this)
|
|
}
|