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

fix: zstd to correctly reset buffer

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

8
storage/remote/compression.go
View file

@ -123,7 +123,7 @@ type snappyAltCompression struct {
} }
func (s *snappyAltCompression) Compress(data []byte) ([]byte, error) { func (s *snappyAltCompression) Compress(data []byte) ([]byte, error) {
s.buf = s.buf[0:cap(s.buf)] s.buf = s.buf[:0]
res := reSnappy.Encode(s.buf, data) res := reSnappy.Encode(s.buf, data)
if n := reSnappy.MaxEncodedLen(len(data)); n > len(s.buf) { if n := reSnappy.MaxEncodedLen(len(data)); n > len(s.buf) {
s.buf = make([]byte, n) s.buf = make([]byte, n)
@ -131,7 +131,7 @@ func (s *snappyAltCompression) Compress(data []byte) ([]byte, error) {
return res, nil return res, nil
} }
func (s *snappyAltCompression) Decompress(data []byte) ([]byte, error) { func (s *snappyAltCompression) Decompress(data []byte) ([]byte, error) {
s.buf = s.buf[0:cap(s.buf)] s.buf = s.buf[:0]
uncompressed, err := reSnappy.Decode(s.buf, data) uncompressed, err := reSnappy.Decode(s.buf, data)
if len(uncompressed) > cap(s.buf) { if len(uncompressed) > cap(s.buf) {
s.buf = uncompressed s.buf = uncompressed
@ -152,7 +152,7 @@ func (s *s2Compression) Compress(data []byte) ([]byte, error) {
} }
func (s *s2Compression) Decompress(data []byte) ([]byte, error) { func (s *s2Compression) Decompress(data []byte) ([]byte, error) {
s.buf = s.buf[0:cap(s.buf)] s.buf = s.buf[:0]
uncompressed, err := reS2.Decode(s.buf, data) uncompressed, err := reS2.Decode(s.buf, data)
if len(uncompressed) > cap(s.buf) { if len(uncompressed) > cap(s.buf) {
s.buf = uncompressed s.buf = uncompressed
@ -170,7 +170,7 @@ func (z *zstdCompression) Compress(data []byte) ([]byte, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
z.buf = z.buf[0:cap(z.buf)] z.buf = z.buf[:0]
res := w.EncodeAll(data, z.buf) res := w.EncodeAll(data, z.buf)
if len(res) > cap(z.buf) { if len(res) > cap(z.buf) {
z.buf = res z.buf = res