1
0
Fork 0
mirror of https://github.com/vbatts/go-mtree.git synced 2024-12-25 06:56:30 +00:00
go-mtree/vendor/github.com/sirupsen/logrus/buffer_pool.go
Vincent Batts fa29fcb58d
go*: go get -u ./...
Signed-off-by: Vincent Batts <vbatts@hashbangbash.com>
2023-03-22 10:57:01 -04:00

43 lines
663 B
Go

package logrus
import (
"bytes"
"sync"
)
var (
bufferPool BufferPool
)
type BufferPool interface {
Put(*bytes.Buffer)
Get() *bytes.Buffer
}
type defaultPool struct {
pool *sync.Pool
}
func (p *defaultPool) Put(buf *bytes.Buffer) {
p.pool.Put(buf)
}
func (p *defaultPool) Get() *bytes.Buffer {
return p.pool.Get().(*bytes.Buffer)
}
// SetBufferPool allows to replace the default logrus buffer pool
// to better meets the specific needs of an application.
func SetBufferPool(bp BufferPool) {
bufferPool = bp
}
func init() {
SetBufferPool(&defaultPool{
pool: &sync.Pool{
New: func() interface{} {
return new(bytes.Buffer)
},
},
})
}