From 61ba4cfad62ac36dc4948ecd7401399651f7a0f3 Mon Sep 17 00:00:00 2001
From: kercylan98 <kercylan@gmail.com>
Date: Thu, 11 May 2023 10:44:38 +0800
Subject: [PATCH] =?UTF-8?q?=E8=BF=9E=E6=8E=A5=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/conn.go | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/server/conn.go b/server/conn.go
index d1625e5..e7b0ff6 100644
--- a/server/conn.go
+++ b/server/conn.go
@@ -15,6 +15,7 @@ func newKcpConn(session *kcp.UDPSession) *Conn {
 			_, err := session.Write(data)
 			return err
 		},
+		data: map[any]any{},
 	}
 }
 
@@ -26,6 +27,7 @@ func newGNetConn(conn gnet.Conn) *Conn {
 		write: func(data []byte) error {
 			return conn.AsyncWrite(data)
 		},
+		data: map[any]any{},
 	}
 }
 
@@ -37,6 +39,7 @@ func newWebsocketConn(ws *websocket.Conn, ip string) *Conn {
 		write: func(data []byte) error {
 			return ws.WriteMessage(websocket.BinaryMessage, data)
 		},
+		data: map[any]any{},
 	}
 }
 
@@ -62,31 +65,30 @@ func (slf *Conn) Write(data []byte) error {
 // Close 关闭连接
 func (slf *Conn) Close() {
 	if slf.ws != nil {
-		slf.ws.Close()
+		_ = slf.ws.Close()
 	} else if slf.gn != nil {
-		slf.gn.Close()
+		_ = slf.gn.Close()
 	} else if slf.kcp != nil {
-		slf.kcp.Close()
+		_ = slf.kcp.Close()
 	}
 	slf.write = nil
 }
 
+// SetData 设置连接数据
 func (slf *Conn) SetData(key, value any) *Conn {
-	if slf.data == nil {
-		slf.data = map[any]any{}
-	}
 	slf.data[key] = value
 	return slf
 }
 
+// GetData 获取连接数据
 func (slf *Conn) GetData(key any) any {
 	return slf.data[key]
 }
 
+// ReleaseData 释放数据
 func (slf *Conn) ReleaseData() *Conn {
 	for k := range slf.data {
 		delete(slf.data, k)
 	}
-	slf.data = nil
 	return slf
 }