From 48254fae6d59af51ffbb720ac8afff830e7ef46c Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Tue, 12 Aug 2025 10:46:21 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E4=BC=98=E5=8C=96ws=E6=8E=A7?=
 =?UTF-8?q?=E5=88=B6=E5=99=A8=E5=AE=9E=E4=BE=8B=E5=8C=96=E6=96=B9=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/ws/ws.go | 19 ++++++++-----------
 1 file changed, 8 insertions(+), 11 deletions(-)

diff --git a/src/modules/ws/ws.go b/src/modules/ws/ws.go
index af8147f4..8407c478 100644
--- a/src/modules/ws/ws.go
+++ b/src/modules/ws/ws.go
@@ -14,36 +14,33 @@ func Setup(router *gin.Engine) {
 	logger.Infof("开始加载 ====> ws 模块路由")
 
 	// WebSocket 协议
+	ws := controller.NewWSController
 	wsGroup := router.Group("/ws")
 	{
 		wsGroup.GET("",
 			middleware.AuthorizeUser(nil),
-			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.ws", collectlogs.BUSINESS_TYPE_OTHER)),
-			controller.NewWSController.WS,
+			ws.WS,
 		)
 		wsGroup.GET("/test",
 			middleware.AuthorizeUser(nil),
-			controller.NewWSController.Test,
+			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.ws", collectlogs.BUSINESS_TYPE_OTHER)),
+			ws.Test,
 		)
 		wsGroup.GET("/ssh",
 			middleware.AuthorizeUser(nil),
-			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.ws", collectlogs.BUSINESS_TYPE_OTHER)),
-			controller.NewWSController.SSH,
+			ws.SSH,
 		)
 		wsGroup.GET("/telnet",
 			middleware.AuthorizeUser(nil),
-			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.ws", collectlogs.BUSINESS_TYPE_OTHER)),
-			controller.NewWSController.Telnet,
+			ws.Telnet,
 		)
 		wsGroup.GET("/redis",
 			middleware.AuthorizeUser(nil),
-			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.ws", collectlogs.BUSINESS_TYPE_OTHER)),
-			controller.NewWSController.Redis,
+			ws.Redis,
 		)
 		wsGroup.GET("/view",
 			middleware.AuthorizeUser(nil),
-			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.ws", collectlogs.BUSINESS_TYPE_OTHER)),
-			controller.NewWSController.ShellView,
+			ws.ShellView,
 		)
 	}
 }