9012069bd7
* chore: execute goimports to format the code Signed-off-by: promalert <promalert@outlook.com> * goimports -w . --------- Signed-off-by: promalert <promalert@outlook.com> Co-authored-by: Chris Lu <chris.lu@gmail.com>
26 lines
446 B
Go
26 lines
446 B
Go
package operation
|
|
|
|
import (
|
|
"sync/atomic"
|
|
|
|
"github.com/valyala/bytebufferpool"
|
|
)
|
|
|
|
var bufferCounter int64
|
|
|
|
func GetBuffer() *bytebufferpool.ByteBuffer {
|
|
defer func() {
|
|
atomic.AddInt64(&bufferCounter, 1)
|
|
// println("+", bufferCounter)
|
|
}()
|
|
return bytebufferpool.Get()
|
|
}
|
|
|
|
func PutBuffer(buf *bytebufferpool.ByteBuffer) {
|
|
defer func() {
|
|
atomic.AddInt64(&bufferCounter, -1)
|
|
// println("-", bufferCounter)
|
|
}()
|
|
bytebufferpool.Put(buf)
|
|
}
|