✨ use websocket

router/modules/websocket_router.go

@@ -0,0 +1,17 @@
+package modules
+
+import (
+	"github.com/gin-gonic/gin"
+	"schisandra-cloud-album/api"
+)
+
+var websocketAPI = api.Api.WebsocketApi
+
+func WebsocketRouter(router *gin.RouterGroup) {
+	group := router.Group("/ws")
+	{
+		group.GET("/socket", websocketAPI.NewSocketClient)
+		group.GET("/delete", websocketAPI.DeleteClient)
+	}
+
+}

router/router.go

@@ -22,10 +22,11 @@ func InitRouter() *gin.Engine {
 	// 国际化设置
 	publicGroup.Use(middleware.I18n())
 
-	modules.SwaggerRouter(router)      // 注册swagger路由
-	modules.UserRouter(publicGroup)    // 注册鉴权路由
-	modules.CaptchaRouter(publicGroup) // 注册验证码路由
-	modules.SmsRouter(publicGroup)     // 注册短信验证码路由
-	modules.OauthRouter(publicGroup)   // 注册oauth路由
+	modules.SwaggerRouter(router)        // 注册swagger路由
+	modules.UserRouter(publicGroup)      // 注册鉴权路由
+	modules.CaptchaRouter(publicGroup)   // 注册验证码路由
+	modules.SmsRouter(publicGroup)       // 注册短信验证码路由
+	modules.OauthRouter(publicGroup)     // 注册oauth路由
+	modules.WebsocketRouter(publicGroup) // 注册websocket路由
 	return router
 }