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,
 		)
 	}
 }