diff --git a/config/server.go b/config/server.go index 1096854..3271d9c 100644 --- a/config/server.go +++ b/config/server.go @@ -135,6 +135,20 @@ func DisableStoreConnection() SetWSServerConfig { } } +// SetConnectionManager 连接管理实例 +// +// Author : go_developer@163.com<张德满> +// +// Date : 10:48 下午 2021/4/22 +func SetConnectionManager(manager storage.IConnection) SetWSServerConfig { + return func(wsc *WSServerConfig) { + if nil == manager { + manager = storage.NewDefaultConnectionManager() + } + wsc.ConnectionManager = manager + } +} + // SetMaxMessageSize 限制消息大小 // // Author : go_developer@163.com<张德满>