You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
730 B
44 lines
730 B
package bin |
|
|
|
import "sync" |
|
|
|
// Pool is a bin.Buffer pool. |
|
type Pool struct { |
|
pool sync.Pool |
|
} |
|
|
|
// NewPool creates new Pool. |
|
// Length is initial buffer length. |
|
func NewPool(length int) *Pool { |
|
return &Pool{ |
|
pool: sync.Pool{ |
|
New: func() interface{} { |
|
var r []byte |
|
if length > 0 { |
|
r = make([]byte, 0, length) |
|
} |
|
return &Buffer{Buf: r} |
|
}, |
|
}, |
|
} |
|
} |
|
|
|
// Put returns buffer to pool. |
|
func (b *Pool) Put(buf *Buffer) { |
|
b.pool.Put(buf) |
|
} |
|
|
|
// Get takes buffer from pool. |
|
func (b *Pool) Get() *Buffer { |
|
buf := b.pool.Get().(*Buffer) |
|
buf.Reset() |
|
return buf |
|
} |
|
|
|
// GetSize takes buffer with given size from pool. |
|
func (b *Pool) GetSize(length int) *Buffer { |
|
buf := b.Get() |
|
buf.ResetN(length) |
|
|
|
return buf |
|
}
|
|
|