mudhorn/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2025-03-05 20:59:13 -08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

optimize more allocs

Browse source
This commit is contained in:
Nicolás Pazos 2023-11-01 21:35:52 -03:00 committed by Callum Styan
parent 08c3fc6ca9
commit 1ef02aad26
1 changed files with 21 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
storage/remote/compression.go
View file

@ -163,15 +163,22 @@ func (s *s2Compression) Decompress(data []byte) ([]byte, error) {
type zstdCompression struct {
level zstd.EncoderLevel
buf []byte
r *reZstd.Decoder
w *reZstd.Encoder
}
func (z *zstdCompression) Compress(data []byte) ([]byte, error) {
w, err := reZstd.NewWriter(nil, reZstd.WithEncoderLevel(z.level))
if err != nil {
return nil, err
var err error
if z.w == nil {
z.w, err = reZstd.NewWriter(nil, reZstd.WithEncoderLevel(z.level))
if err != nil {
return nil, err
}
}
z.w.Reset(nil)
z.buf = z.buf[:0]
res := w.EncodeAll(data, z.buf)
res := z.w.EncodeAll(data, z.buf)
if len(res) > cap(z.buf) {
z.buf = res
}
@ -179,12 +186,20 @@ func (z *zstdCompression) Compress(data []byte) ([]byte, error) {
}
func (z *zstdCompression) Decompress(data []byte) ([]byte, error) {
decoder, err := reZstd.NewReader(nil)
var err error
if z.r == nil {
z.r, err = reZstd.NewReader(nil)
if err != nil {
return nil, err
}
}
err = z.r.Reset(nil)
if err != nil {
return nil, err
}
z.buf = z.buf[:0]
buf, err := decoder.DecodeAll(data, z.buf)
buf, err := z.r.DecodeAll(data, z.buf)
if err != nil {
return nil, err
}