wechat

server/initialize/router.go

@@ -8,6 +8,7 @@ import (
 	"github.com/flipped-aurora/gin-vue-admin/server/global"
 	"github.com/flipped-aurora/gin-vue-admin/server/middleware"
 	"github.com/flipped-aurora/gin-vue-admin/server/router"
+	"github.com/flipped-aurora/gin-vue-admin/server/websocket"
 	"github.com/gin-gonic/gin"
 	swaggerFiles "github.com/swaggo/files"
 	ginSwagger "github.com/swaggo/gin-swagger"
@@ -42,6 +43,9 @@ func Routers() *gin.Engine {
 
 	sseServer := McpRun()
 
+	// 初始化WebSocket
+	InitWebSocket()
+
 	// 注册mcp服务
 	Router.GET(global.GVA_CONFIG.MCP.SSEPath, func(c *gin.Context) {
 		sseServer.SSEHandler().ServeHTTP(c.Writer, c.Request)
@@ -78,6 +82,9 @@ func Routers() *gin.Engine {
 	PrivateGroup.Use(middleware.JWTAuth()).Use(middleware.CasbinHandler())
 	UserGroup.Use(middleware.UserJWTAuth())
 
+	// WebSocket路由（不使用UserJWTAuth中间件，使用自定义认证）
+	Router.GET("/user/ws", websocket.WebSocketAuthMiddleware(), websocket.HandleConnection)
+
 	{
 		// 健康监测
 		PublicGroup.GET("/health", func(c *gin.Context) {

server/initialize/websocket.go

@@ -0,0 +1,10 @@
+package initialize
+
+import (
+	"github.com/flipped-aurora/gin-vue-admin/server/websocket"
+)
+
+// InitWebSocket 初始化WebSocket
+func InitWebSocket() {
+	websocket.InitWebSocketServer()
+}