mirror of https://github.com/ehang-io/nps
62 lines
1.0 KiB
Go
62 lines
1.0 KiB
Go
package pool
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
const PoolSize = 64 * 1024
|
|
const PoolSizeSmall = 100
|
|
const PoolSizeUdp = 1472
|
|
const PoolSizeCopy = 32 * 1024
|
|
|
|
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{} {
|
|
return make([]byte, PoolSizeCopy)
|
|
},
|
|
}
|
|
|
|
func PutBufPoolUdp(buf []byte) {
|
|
if cap(buf) == PoolSizeUdp {
|
|
BufPoolUdp.Put(buf[:PoolSizeUdp])
|
|
}
|
|
}
|
|
|
|
func PutBufPoolCopy(buf []byte) {
|
|
if cap(buf) == PoolSizeCopy {
|
|
BufPoolCopy.Put(buf[:PoolSizeCopy])
|
|
}
|
|
}
|
|
|
|
func PutBufPoolSmall(buf []byte) {
|
|
if cap(buf) == PoolSizeSmall {
|
|
BufPoolSmall.Put(buf[:PoolSizeSmall])
|
|
}
|
|
}
|
|
|
|
func PutBufPoolMax(buf []byte) {
|
|
if cap(buf) == PoolSize {
|
|
BufPoolMax.Put(buf[:PoolSize])
|
|
}
|
|
}
|