34 lines
741 B
Go
34 lines
741 B
Go
package bytepool
|
|
|
|
import "sync"
|
|
|
|
// BytePool is a cached pool of reusable byte slices.
|
|
type BytePool struct {
|
|
sync.Pool
|
|
}
|
|
|
|
// New allocates a new BytePool with slices of the provided capacity.
|
|
func New(length, capacity int) *BytePool {
|
|
var bp BytePool
|
|
bp.Pool.New = func() interface{} {
|
|
return make([]byte, length, capacity)
|
|
}
|
|
return &bp
|
|
}
|
|
|
|
// Get returns a byte slice from the pool.
|
|
func (bp *BytePool) Get() []byte {
|
|
return bp.Pool.Get().([]byte)
|
|
}
|
|
|
|
// Put returns a byte slice to the pool.
|
|
func (bp *BytePool) Put(b []byte) {
|
|
b = b[:cap(b)]
|
|
// Zero out the bytes.
|
|
// Apparently this specific expression is optimized by the compiler, see
|
|
// github.com/golang/go/issues/5373.
|
|
for i := range b {
|
|
b[i] = 0
|
|
}
|
|
bp.Pool.Put(b)
|
|
}
|