mirror of https://github.com/jlelse/GoBlog
30 lines
426 B
Go
30 lines
426 B
Go
package bufferpool
|
|
|
|
import (
|
|
"bytes"
|
|
"sync"
|
|
)
|
|
|
|
var bufferPool = sync.Pool{
|
|
New: func() interface{} {
|
|
return new(bytes.Buffer)
|
|
},
|
|
}
|
|
|
|
var poolMutex sync.Mutex
|
|
|
|
func Get() *bytes.Buffer {
|
|
poolMutex.Lock()
|
|
defer poolMutex.Unlock()
|
|
return bufferPool.Get().(*bytes.Buffer)
|
|
}
|
|
|
|
func Put(bufs ...*bytes.Buffer) {
|
|
poolMutex.Lock()
|
|
defer poolMutex.Unlock()
|
|
for _, buf := range bufs {
|
|
buf.Reset()
|
|
bufferPool.Put(buf)
|
|
}
|
|
}
|