From 848fda9edb867323eae8fde0e87785b381f5b53c Mon Sep 17 00:00:00 2001 From: gufeijun Date: Sun, 21 Feb 2021 16:31:28 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8A=9F=E8=83=BD=E5=A2=9E=E5=BC=BA:=E7=BB=99C?= =?UTF-8?q?onnction=E6=96=B0=E5=A2=9EContext()=E6=96=B9=E6=B3=95,=E8=BF=94?= =?UTF-8?q?=E5=9B=9Econtext?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ziface/iconnection.go | 8 ++++++-- znet/connection.go | 5 +++++ 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/ziface/iconnection.go b/ziface/iconnection.go index 7fe9063..4203ab9 100644 --- a/ziface/iconnection.go +++ b/ziface/iconnection.go @@ -1,6 +1,9 @@ package ziface -import "net" +import ( + "context" + "net" +) //定义连接接口 type IConnection interface { @@ -8,7 +11,8 @@ type IConnection interface { Start() //停止连接,结束当前连接状态M Stop() - + //返回ctx,用于用户自定义的go程获取连接退出状态 + Context() context.Context //从当前连接获取原始的socket TCPConn GetTCPConnection() *net.TCPConn //获取当前连接ID diff --git a/znet/connection.go b/znet/connection.go index 9f11c89..d90e285 100644 --- a/znet/connection.go +++ b/znet/connection.go @@ -276,3 +276,8 @@ func (c *Connection) RemoveProperty(key string) { delete(c.property, key) } + +//返回ctx,用于用户自定义的go程获取连接退出状态 +func (c *Connection) Context() context.Context { + return c.ctx +}