mirror of
https://github.com/GoogleContainerTools/kaniko
synced 2026-09-30 18:18:45 +02:00
* Bump github.com/docker/docker to latest release * Pin deps to docker/docker's requirements fix build breakages fix unit test Trying to see whether am-i-in-a-container checks are broken for integration tests...
53 lines
797 B
Go
53 lines
797 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)
|
|
}
|
|
|
|
func getBuffer() *bytes.Buffer {
|
|
return bufferPool.Get()
|
|
}
|
|
|
|
func putBuffer(buf *bytes.Buffer) {
|
|
buf.Reset()
|
|
bufferPool.Put(buf)
|
|
}
|
|
|
|
// 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)
|
|
},
|
|
},
|
|
})
|
|
}
|