mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-15 00:13:30 +01:00
73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
|
package prompb
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"io"
|
||
|
|
||
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/bytesutil"
|
||
|
"github.com/golang/snappy"
|
||
|
)
|
||
|
|
||
|
// ReadSnappy reads r, unpacks it using snappy, appends it to dst
|
||
|
// and returns the result.
|
||
|
func ReadSnappy(dst []byte, r io.Reader, maxSize int64) ([]byte, error) {
|
||
|
bb := bodyBufferPool.Get()
|
||
|
bb.B = bb.B[:0]
|
||
|
cb := copyBufferPool.Get()
|
||
|
cb.B = bytesutil.Resize(cb.B, 16*1024)
|
||
|
lr := io.LimitReader(r, maxSize+1)
|
||
|
reqLen, err := io.CopyBuffer(bb, lr, cb.B)
|
||
|
copyBufferPool.Put(cb)
|
||
|
if err != nil {
|
||
|
bodyBufferPool.Put(bb)
|
||
|
return dst, fmt.Errorf("cannot read compressed request: %s", err)
|
||
|
}
|
||
|
if reqLen > maxSize {
|
||
|
return dst, fmt.Errorf("too big packed request; mustn't exceed %d bytes", maxSize)
|
||
|
}
|
||
|
|
||
|
buf := dst[len(dst):cap(dst)]
|
||
|
buf, err = snappy.Decode(buf, bb.B)
|
||
|
bodyBufferPool.Put(bb)
|
||
|
if err != nil {
|
||
|
err = fmt.Errorf("cannot decompress request with length %d: %s", reqLen, err)
|
||
|
return dst, err
|
||
|
}
|
||
|
if int64(len(buf)) > maxSize {
|
||
|
return dst, fmt.Errorf("too big unpacked request; musn't exceed %d bytes", maxSize)
|
||
|
}
|
||
|
if len(buf) > 0 && len(dst) < cap(dst) && &buf[0] == &dst[len(dst):cap(dst)][0] {
|
||
|
dst = dst[:len(dst)+len(buf)]
|
||
|
} else {
|
||
|
dst = append(dst, buf...)
|
||
|
}
|
||
|
return dst, nil
|
||
|
}
|
||
|
|
||
|
var bodyBufferPool bytesutil.ByteBufferPool
|
||
|
var copyBufferPool bytesutil.ByteBufferPool
|
||
|
|
||
|
// Reset resets wr.
|
||
|
func (wr *WriteRequest) Reset() {
|
||
|
for i := range wr.Timeseries {
|
||
|
ts := &wr.Timeseries[i]
|
||
|
ts.Labels = nil
|
||
|
ts.Samples = nil
|
||
|
}
|
||
|
wr.Timeseries = wr.Timeseries[:0]
|
||
|
|
||
|
for i := range wr.labelsPool {
|
||
|
lb := &wr.labelsPool[i]
|
||
|
lb.Name = nil
|
||
|
lb.Value = nil
|
||
|
}
|
||
|
wr.labelsPool = wr.labelsPool[:0]
|
||
|
|
||
|
for i := range wr.samplesPool {
|
||
|
s := &wr.samplesPool[i]
|
||
|
s.Value = 0
|
||
|
s.Timestamp = 0
|
||
|
}
|
||
|
wr.samplesPool = wr.samplesPool[:0]
|
||
|
}
|