mirror of
https://github.com/e1732a364fed/v2ray_simple.git
synced 2025-12-24 13:27:56 +08:00
101 lines
2.0 KiB
Go
101 lines
2.0 KiB
Go
package utils
|
||
|
||
import (
|
||
"bytes"
|
||
"sync"
|
||
)
|
||
|
||
var (
|
||
standardBytesPool sync.Pool //1500, 最大MTU的长度
|
||
|
||
// 实际上tcp默认是 16384, 16k，实际上范围是1k～128k之间，我们64k已经够了
|
||
//而 udp则最大还不到 64k。(65535－20－8)
|
||
standardPacketPool sync.Pool //64*1024
|
||
|
||
bufPool sync.Pool
|
||
)
|
||
|
||
//即MTU
|
||
const StandardBytesLength int = 1500
|
||
|
||
//本作设定的最大buf大小，64k
|
||
const maxBufLen int = 64 * 1024
|
||
|
||
func init() {
|
||
standardBytesPool = sync.Pool{
|
||
New: func() interface{} {
|
||
return make([]byte, StandardBytesLength)
|
||
},
|
||
}
|
||
|
||
standardPacketPool = sync.Pool{
|
||
New: func() interface{} {
|
||
return make([]byte, maxBufLen)
|
||
},
|
||
}
|
||
|
||
bufPool = sync.Pool{
|
||
New: func() interface{} {
|
||
return &bytes.Buffer{}
|
||
},
|
||
}
|
||
}
|
||
|
||
func GetBuf() *bytes.Buffer {
|
||
return bufPool.Get().(*bytes.Buffer)
|
||
}
|
||
|
||
func PutBuf(buf *bytes.Buffer) {
|
||
buf.Reset()
|
||
bufPool.Put(buf)
|
||
}
|
||
|
||
//建议在 Read net.Conn 时, 使用 GetPacket函数 获取到足够大的 []byte（64*1024字节）
|
||
func GetPacket() []byte {
|
||
return standardPacketPool.Get().([]byte)
|
||
}
|
||
|
||
// 放回用 GetPacket 获取的 []byte, 或者长度够长的可回收的 []byte
|
||
func PutPacket(bs []byte) {
|
||
c := cap(bs)
|
||
if c < maxBufLen { //如果不够大，考虑放到更小的 pool里
|
||
if c > StandardBytesLength {
|
||
standardBytesPool.Put(bs[:c])
|
||
}
|
||
return
|
||
}
|
||
|
||
standardPacketPool.Put(bs[:c])
|
||
}
|
||
|
||
// 从pool中获取 []byte, 在 size <= 1500时有最佳性能
|
||
func GetBytes(size int) []byte {
|
||
if size < StandardBytesLength {
|
||
bs := standardBytesPool.Get().([]byte)
|
||
return bs[:size]
|
||
}
|
||
|
||
randomBytes1 := standardBytesPool.Get().([]byte)
|
||
|
||
if len(randomBytes1) >= size {
|
||
return randomBytes1[:size]
|
||
} else {
|
||
standardBytesPool.Put(randomBytes1)
|
||
return make([]byte, size)
|
||
}
|
||
|
||
}
|
||
|
||
// 根据bs长度 选择放入各种pool中, 只有cap>=1500 才会被处理
|
||
func PutBytes(bs []byte) {
|
||
c := cap(bs)
|
||
if c < StandardBytesLength {
|
||
|
||
return
|
||
} else if c >= StandardBytesLength && c < maxBufLen {
|
||
standardBytesPool.Put(bs[:c])
|
||
} else if c >= maxBufLen {
|
||
standardPacketPool.Put(bs[:c])
|
||
}
|
||
}
|