mirror of https://github.com/ehang-io/nps
60 lines
1023 B
Go
60 lines
1023 B
Go
package pool
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
const PoolSize = 64 * 1024
|
|
const PoolSizeSmall = 100
|
|
const PoolSizeUdp = 1472
|
|
const PoolSizeCopy = 32 << 10
|
|
|
|
var BufPool = sync.Pool{
|
|
New: func() interface{} {
|
|
return make([]byte, PoolSize)
|
|
},
|
|
}
|
|
|
|
var BufPoolUdp = sync.Pool{
|
|
New: func() interface{} {
|
|
return make([]byte, PoolSizeUdp)
|
|
},
|
|
}
|
|
var BufPoolMax = sync.Pool{
|
|
New: func() interface{} {
|
|
return make([]byte, PoolSize)
|
|
},
|
|
}
|
|
var BufPoolSmall = sync.Pool{
|
|
New: func() interface{} {
|
|
return make([]byte, PoolSizeSmall)
|
|
},
|
|
}
|
|
var BufPoolCopy = sync.Pool{
|
|
New: func() interface{} {
|
|
buf := make([]byte, PoolSizeCopy)
|
|
return &buf
|
|
},
|
|
}
|
|
func PutBufPoolUdp(buf []byte) {
|
|
if cap(buf) == PoolSizeUdp {
|
|
BufPoolUdp.Put(buf[:PoolSizeUdp])
|
|
}
|
|
}
|
|
|
|
func PutBufPoolCopy(buf []byte) {
|
|
if cap(buf) == PoolSizeCopy {
|
|
BufPoolCopy.Put(&buf)
|
|
}
|
|
}
|
|
|
|
func GetBufPoolCopy() ([]byte) {
|
|
return (*BufPoolCopy.Get().(*[]byte))[:PoolSizeCopy]
|
|
}
|
|
|
|
func PutBufPoolMax(buf []byte) {
|
|
if cap(buf) == PoolSize {
|
|
BufPoolMax.Put(buf[:PoolSize])
|
|
}
|
|
}
|