mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-03 15:00:26 +08:00
005e27a63c
* 取消用户设置的压缩级别，现在压缩级别通过系统自动设置 * Pool中的对象命中100万次时自动销毁，避免内存泄漏 * 降低Pool中的对象数量，避免占用太多内存 * 根据系统CPU线程数自动计算压缩级别，避免消耗太多CPU * zstd限制解码的最大Window * zstd使用低内存模式
59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
|
|
|
|
package compressions
|
|
|
|
import (
|
|
"github.com/klauspost/compress/zstd"
|
|
"io"
|
|
)
|
|
|
|
type ZSTDWriter struct {
|
|
BaseWriter
|
|
|
|
writer *zstd.Encoder
|
|
level int
|
|
}
|
|
|
|
func NewZSTDWriter(writer io.Writer, level int) (Writer, error) {
|
|
return sharedZSTDWriterPool.Get(writer, level)
|
|
}
|
|
|
|
func newZSTDWriter(writer io.Writer) (Writer, error) {
|
|
var level = 1
|
|
var zstdLevel = zstd.SpeedFastest
|
|
|
|
zstdWriter, err := zstd.NewWriter(writer, zstd.WithEncoderLevel(zstdLevel), zstd.WithWindowSize(16<<10), zstd.WithLowerEncoderMem(true))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &ZSTDWriter{
|
|
writer: zstdWriter,
|
|
level: level,
|
|
}, nil
|
|
}
|
|
|
|
func (this *ZSTDWriter) Write(p []byte) (int, error) {
|
|
return this.writer.Write(p)
|
|
}
|
|
|
|
func (this *ZSTDWriter) Flush() error {
|
|
return this.writer.Flush()
|
|
}
|
|
|
|
func (this *ZSTDWriter) Reset(writer io.Writer) {
|
|
this.writer.Reset(writer)
|
|
}
|
|
|
|
func (this *ZSTDWriter) RawClose() error {
|
|
return this.writer.Close()
|
|
}
|
|
|
|
func (this *ZSTDWriter) Close() error {
|
|
return this.Finish(this)
|
|
}
|
|
|
|
func (this *ZSTDWriter) Level() int {
|
|
return this.level
|
|
}
|