From bb34744411a04c779914ca1ec826aa56ddecb0a3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=99=BD=E8=8C=B6=E6=B8=85=E6=AC=A2?=
 <go_developer@163.com>
Date: Mon, 22 Jul 2024 18:38:48 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E8=8E=B7=E5=8F=96=E9=95=BF?=
 =?UTF-8?q?=E8=BF=9E=E6=8E=A5=E5=BC=95=E6=93=8E=20+=20=E8=8E=B7=E5=8F=96se?=
 =?UTF-8?q?rver=E9=85=8D=E7=BD=AE=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 construct.go | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/construct.go b/construct.go
index 125b8bd..bd1b624 100644
--- a/construct.go
+++ b/construct.go
@@ -42,6 +42,24 @@ type Server struct {
 	conf      *config.WSServerConfig // 配置
 }
 
+// GetConfig 获取service实例配置
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 18:36 2024/7/22
+func (s *Server) GetConfig() *config.WSServerConfig {
+	return s.conf
+}
+
+// GetMelody 获取websocket实例
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 18:37 2024/7/22
+func (s *Server) GetMelody() *melody.Melody {
+	return s.wsServer
+}
+
 var (
 	// ginRouterTable 表
 	ginRouterTable map[int]*gin.Engine
@@ -94,6 +112,21 @@ func NewWebsocketServe(finishHook func(), wsInstanceList ...abstract.IWebsocket)
 	return nil
 }
 
+// GetWsServer 获取ws Server
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 18:36 2024/7/22
+func GetWsServer(port int, serverFlag string) (*Server, error) {
+	if _, exist := wsServerTable[port]; !exist {
+		return nil, errors.New(fmt.Sprintf("%v : port is not listen", port))
+	}
+	if _, exist := wsServerTable[port][serverFlag]; !exist {
+		return nil, errors.New(fmt.Sprintf("server flag %v on port %v is not found", serverFlag, port))
+	}
+	return wsServerTable[port][serverFlag], nil
+}
+
 // 初始化server
 func initServer(wsInstance abstract.IWebsocket) {
 	// 初始化ws server