From 0dc7b6fcdcaa283dcfd48e3b4a155b0bef80d14a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=A0=E5=BE=B7=E6=BB=A1?= Date: Thu, 22 Apr 2021 22:49:38 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E9=85=8D=E7=BD=AE=E8=BF=9E?= =?UTF-8?q?=E6=8E=A5=E7=AE=A1=E7=90=86=E7=9A=84=E5=AE=9E=E4=BE=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config/server.go | 14 ++++++++++++++ 1 file changed, 14 insertions(+) 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<张德满>