- pkg/ws/message.go: 统一消息协议(Message/Response/PushMessage) - pkg/ws/hub.go: Hub 连接管理(注册/注销/按 userID 查找/在线计数) - pkg/ws/client.go: 客户端连接封装(readPump/writePump/心跳 30s) - pkg/ws/pubsub.go: Redis Pub/Sub 消息路由(按用户频道发布/订阅) - app/ws/handler.go: WebSocket 升级处理(JWT 认证 + 消息分发) - app/ws/router.go: GET /ws 路由注册 - app/ws/provider.go: Wire Provider Set - 更新 provider/router 集成 WebSocket 模块 Made-with: Cursor
9 lines
193 B
Go
9 lines
193 B
Go
package ws
|
|
|
|
import "github.com/gin-gonic/gin"
|
|
|
|
// RegisterRoutes 注册 WebSocket 相关路由
|
|
func RegisterRoutes(engine *gin.Engine, handler *Handler) {
|
|
engine.GET("/ws", handler.Upgrade)
|
|
}
|