zhangdeman/websocket
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

升级默认的连接管理的实现

Browse Source
This commit is contained in:
白茶清欢 2021-04-17 23:58:26 +08:00
parent 99f17f86ec
commit 854c9b0ad7
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
storage/IStorage.go
View File

@ -18,7 +18,7 @@ type IConnection interface {
// Store 存储 // Store 存储
Store(ctx *context.WSContext) Store(ctx *context.WSContext)
// Del 移除 // Del 移除
Del(ctx *context.WSContext) Del(ctx *context.WSContext, message string)
// GetCtxList 获取连接列表 // GetCtxList 获取连接列表
GetCtxList(cidList ...string) []*context.WSContext GetCtxList(cidList ...string) []*context.WSContext
// Clear 清空连接 // Clear 清空连接

6
storage/connection.go
View File

@ -31,7 +31,11 @@ func (c *connection) Store(ctx *context.WSContext) {
c.table.Set(ctx.ConnectionID, ctx) c.table.Set(ctx.ConnectionID, ctx)
} }
func (c *connection) Del(ctx *context.WSContext) { func (c *connection) Del(ctx *context.WSContext, message string) {
if len(message) > 0 {
_ = ctx.Session.Write([]byte(message))
_ = ctx.Session.Close()
}
c.table.Del(ctx.ConnectionID) c.table.Del(ctx.ConnectionID)
} }