|
|
package server
|
|
|
|
|
|
import (
|
|
|
"sync"
|
|
|
)
|
|
|
|
|
|
// 对象池（减少GC压力）
|
|
|
var (
|
|
|
// messagePool 消息缓冲区池
|
|
|
messagePool = sync.Pool{
|
|
|
New: func() interface{} {
|
|
|
return make([]byte, 0, 4096)
|
|
|
},
|
|
|
}
|
|
|
|
|
|
// messagesPool 消息列表池
|
|
|
messagesPool = sync.Pool{
|
|
|
New: func() interface{} {
|
|
|
return make([][]byte, 0, 4)
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
|
|
|
// getMessageBuffer 从池中获取消息缓冲区
|
|
|
func getMessageBuffer() []byte {
|
|
|
return messagePool.Get().([]byte)
|
|
|
}
|
|
|
|
|
|
// putMessageBuffer 将消息缓冲区放回池中
|
|
|
func putMessageBuffer(buf []byte) {
|
|
|
if buf == nil {
|
|
|
return
|
|
|
}
|
|
|
// 重置长度但保留容量
|
|
|
buf = buf[:0]
|
|
|
// 如果容量太大，不回收（避免池中积累大对象）
|
|
|
if cap(buf) > 64*1024 {
|
|
|
return
|
|
|
}
|
|
|
messagePool.Put(buf)
|
|
|
}
|
|
|
|
|
|
// getMessagesSlice 从池中获取消息列表
|
|
|
func getMessagesSlice() [][]byte {
|
|
|
return messagesPool.Get().([][]byte)
|
|
|
}
|
|
|
|
|
|
// putMessagesSlice 将消息列表放回池中
|
|
|
func putMessagesSlice(messages [][]byte) {
|
|
|
if messages == nil {
|
|
|
return
|
|
|
}
|
|
|
// 清空但保留容量
|
|
|
for i := range messages {
|
|
|
messages[i] = nil
|
|
|
}
|
|
|
messages = messages[:0]
|
|
|
// 如果容量太大，不回收
|
|
|
if cap(messages) > 64 {
|
|
|
return
|
|
|
}
|
|
|
messagesPool.Put(messages)
|
|
|
}
|
|
|
|
|
|
// requestPool Request对象池
|
|
|
var requestPool = sync.Pool{
|
|
|
New: func() interface{} {
|
|
|
// 注意：这里不能直接创建requestImpl，因为它是未导出的
|
|
|
// 对象池暂时不用于Request和Response，因为它们需要特定的初始化参数
|
|
|
// 如果需要使用对象池，需要修改request和response的创建方式
|
|
|
return nil
|
|
|
},
|
|
|
}
|
|
|
|
|
|
// responsePool Response对象池
|
|
|
var responsePool = sync.Pool{
|
|
|
New: func() interface{} {
|
|
|
// 注意：这里不能直接创建responseImpl，因为它是未导出的
|
|
|
// 对象池暂时不用于Request和Response，因为它们需要特定的初始化参数
|
|
|
return nil
|
|
|
},
|
|
|
}
|
|
|
|