From faaa1d54d17b355cb342012c6e2c0cce5776711c Mon Sep 17 00:00:00 2001
From: kercylan98 <kercylan@gmail.com>
Date: Sat, 6 May 2023 16:47:53 +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/gnet.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/server/gnet.go b/server/gnet.go
index 42bda19..04b1f8b 100644
--- a/server/gnet.go
+++ b/server/gnet.go
@@ -19,12 +19,13 @@ func (slf *gNet) OnShutdown(server gnet.Server) {
 
 func (slf *gNet) OnOpened(c gnet.Conn) (out []byte, action gnet.Action) {
 	conn := newGNetConn(c)
+	c.SetContext(conn)
 	slf.OnConnectionOpenedEvent(conn)
 	return
 }
 
 func (slf *gNet) OnClosed(c gnet.Conn, err error) (action gnet.Action) {
-	slf.OnConnectionClosedEvent(newGNetConn(c))
+	slf.OnConnectionClosedEvent(c.Context().(*Conn))
 	return
 }
 
@@ -37,7 +38,7 @@ func (slf *gNet) AfterWrite(c gnet.Conn, b []byte) {
 }
 
 func (slf *gNet) React(packet []byte, c gnet.Conn) (out []byte, action gnet.Action) {
-	slf.Server.PushMessage(MessageTypePacket, newGNetConn(c), packet)
+	slf.Server.PushMessage(MessageTypePacket, c.Context().(*Conn), packet)
 	return nil, gnet.None
 }