forked from TrueCloudLab/frostfs-api-go
Dmitrii Stepanov
f50872f1bc
Now it is possible set marshaled data to reduce memory allocations. Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
40 lines
693 B
Go
40 lines
693 B
Go
package signature
|
|
|
|
import "sync"
|
|
|
|
const poolSliceMaxSize = 128 * 1024
|
|
|
|
type buffer struct {
|
|
data []byte
|
|
}
|
|
|
|
var buffersPool = sync.Pool{
|
|
New: func() any {
|
|
return new(buffer)
|
|
},
|
|
}
|
|
|
|
func tryGetNewBufferFromPool(size int) (*buffer, bool) {
|
|
if size > poolSliceMaxSize {
|
|
return &buffer{}, false
|
|
}
|
|
return newBufferFromPool(size), true
|
|
}
|
|
|
|
func newBufferFromPool(size int) *buffer {
|
|
result := buffersPool.Get().(*buffer)
|
|
if cap(result.data) < size {
|
|
result.data = make([]byte, size)
|
|
} else {
|
|
result.data = result.data[:size]
|
|
}
|
|
return result
|
|
}
|
|
|
|
func returnBufferToPool(buf *buffer) {
|
|
if cap(buf.data) > poolSliceMaxSize {
|
|
return
|
|
}
|
|
buf.data = buf.data[:0]
|
|
buffersPool.Put(buf)
|
|
}
|