chenyh
/
vRp.CD2g_test
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

连接关闭逻辑优化

This commit is contained in:
kercylan98 2023-05-19 16:48:32 +08:00
parent 11fd51ba43
commit 83b61b9254
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server/conn.go
View File

@ -105,7 +105,9 @@ func (slf *Conn) Close() {
_ = slf.kcp.Close() _ = slf.kcp.Close()
} }
slf.write = nil slf.write = nil
slf.packetPool.Close() if slf.packetPool != nil {
slf.packetPool.Close()
}
slf.packetPool = nil slf.packetPool = nil
slf.packets = nil slf.packets = nil
} }
@ -137,6 +139,9 @@ func (slf *Conn) IsWebsocket() bool {
// Write 向连接中写入数据 // Write 向连接中写入数据
// - messageType: websocket模式中指定消息类型 // - messageType: websocket模式中指定消息类型
func (slf *Conn) Write(data []byte, messageType ...int) { func (slf *Conn) Write(data []byte, messageType ...int) {
if slf.packetPool == nil {
return
}
cp := slf.packetPool.Get() cp := slf.packetPool.Get()
if len(messageType) > 0 { if len(messageType) > 0 {
cp.websocketMessageType = messageType[0] cp.websocketMessageType = messageType[0]
@ -149,7 +154,7 @@ func (slf *Conn) Write(data []byte, messageType ...int) {
// writeLoop 写循环 // writeLoop 写循环
func (slf *Conn) writeLoop() { func (slf *Conn) writeLoop() {
slf.packetPool = synchronization.NewPool[*connPacket](64, slf.packetPool = synchronization.NewPool[*connPacket](10*1024,
func() *connPacket { func() *connPacket {
return &connPacket{} return &connPacket{}
}, func(data *connPacket) { }, func(data *connPacket) {