You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
706 B
Go
43 lines
706 B
Go
package iox
|
|
|
|
import (
|
|
"bytes"
|
|
"sync"
|
|
)
|
|
|
|
// A BufferPool is a pool to buffer bytes.Buffer objects.
|
|
type BufferPool struct {
|
|
capability int
|
|
pool *sync.Pool
|
|
}
|
|
|
|
// NewBufferPool returns a BufferPool.
|
|
func NewBufferPool(capability int) *BufferPool {
|
|
return &BufferPool{
|
|
capability: capability,
|
|
pool: &sync.Pool{
|
|
New: func() any {
|
|
return new(bytes.Buffer)
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
// Get returns a bytes.Buffer object from bp.
|
|
func (bp *BufferPool) Get() *bytes.Buffer {
|
|
buf := bp.pool.Get().(*bytes.Buffer)
|
|
buf.Reset()
|
|
return buf
|
|
}
|
|
|
|
// Put returns buf into bp.
|
|
func (bp *BufferPool) Put(buf *bytes.Buffer) {
|
|
if buf == nil {
|
|
return
|
|
}
|
|
|
|
if buf.Cap() < bp.capability {
|
|
bp.pool.Put(buf)
|
|
}
|
|
}
|