feat(ws): WebSocket 核心模块(Hub + Client + PubSub + Handler)
- 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
This commit is contained in:
@@ -6,8 +6,10 @@ import (
|
||||
adminController "github.com/echochat/backend/app/admin/controller"
|
||||
authController "github.com/echochat/backend/app/auth/controller"
|
||||
"github.com/echochat/backend/app/auth/service"
|
||||
wsApp "github.com/echochat/backend/app/ws"
|
||||
"github.com/echochat/backend/config"
|
||||
"github.com/echochat/backend/pkg/db"
|
||||
"github.com/echochat/backend/pkg/ws"
|
||||
"github.com/google/wire"
|
||||
"github.com/redis/go-redis/v9"
|
||||
"gorm.io/gorm"
|
||||
@@ -22,6 +24,9 @@ type App struct {
|
||||
AuthController *authController.AuthController // 前台认证控制器
|
||||
AdminAuthController *authController.AdminAuthController // 后台认证控制器
|
||||
UserManageController *adminController.UserManageController // 管理端用户管理控制器
|
||||
WSHandler *wsApp.Handler // WebSocket 连接处理器
|
||||
Hub *ws.Hub // WebSocket Hub 连接管理
|
||||
PubSub *ws.PubSub // Redis Pub/Sub 消息路由
|
||||
}
|
||||
|
||||
// NewApp 创建应用实例
|
||||
@@ -33,6 +38,9 @@ func NewApp(
|
||||
authCtrl *authController.AuthController,
|
||||
adminAuthCtrl *authController.AdminAuthController,
|
||||
userManageCtrl *adminController.UserManageController,
|
||||
wsHandler *wsApp.Handler,
|
||||
hub *ws.Hub,
|
||||
pubsub *ws.PubSub,
|
||||
) *App {
|
||||
return &App{
|
||||
Config: cfg,
|
||||
@@ -42,6 +50,9 @@ func NewApp(
|
||||
AuthController: authCtrl,
|
||||
AdminAuthController: adminAuthCtrl,
|
||||
UserManageController: userManageCtrl,
|
||||
WSHandler: wsHandler,
|
||||
Hub: hub,
|
||||
PubSub: pubsub,
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -6,6 +6,7 @@ package provider
|
||||
import (
|
||||
"github.com/echochat/backend/app/admin"
|
||||
"github.com/echochat/backend/app/auth"
|
||||
wsApp "github.com/echochat/backend/app/ws"
|
||||
"github.com/echochat/backend/config"
|
||||
"github.com/google/wire"
|
||||
)
|
||||
@@ -16,6 +17,7 @@ func InitializeApp(cfg *config.Config) (*App, error) {
|
||||
InfraSet,
|
||||
auth.AuthSet,
|
||||
admin.AdminSet,
|
||||
wsApp.WSSet,
|
||||
)
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
@@ -7,12 +7,13 @@
|
||||
package provider
|
||||
|
||||
import (
|
||||
adminController "github.com/echochat/backend/app/admin/controller"
|
||||
adminDAO "github.com/echochat/backend/app/admin/dao"
|
||||
adminService "github.com/echochat/backend/app/admin/service"
|
||||
controller2 "github.com/echochat/backend/app/admin/controller"
|
||||
dao2 "github.com/echochat/backend/app/admin/dao"
|
||||
service2 "github.com/echochat/backend/app/admin/service"
|
||||
"github.com/echochat/backend/app/auth/controller"
|
||||
"github.com/echochat/backend/app/auth/dao"
|
||||
"github.com/echochat/backend/app/auth/service"
|
||||
"github.com/echochat/backend/app/ws"
|
||||
"github.com/echochat/backend/config"
|
||||
"github.com/echochat/backend/pkg/db"
|
||||
)
|
||||
@@ -38,9 +39,12 @@ func InitializeApp(cfg *config.Config) (*App, error) {
|
||||
authService := service.NewAuthService(userDAO, roleDAO, jwtConfig, tokenStore)
|
||||
authController := controller.NewAuthController(authService)
|
||||
adminAuthController := controller.NewAdminAuthController(authService)
|
||||
userManageDAO := adminDAO.NewUserManageDAO(gormDB)
|
||||
userManageService := adminService.NewUserManageService(userManageDAO, userDAO, roleDAO)
|
||||
userManageController := adminController.NewUserManageController(userManageService)
|
||||
app := NewApp(cfg, gormDB, client, authService, authController, adminAuthController, userManageController)
|
||||
userManageDAO := dao2.NewUserManageDAO(gormDB)
|
||||
userManageService := service2.NewUserManageService(userManageDAO, userDAO, roleDAO)
|
||||
userManageController := controller2.NewUserManageController(userManageService)
|
||||
hub := ws.ProvideHub()
|
||||
pubSub := ws.ProvidePubSub(client, hub)
|
||||
handler := ws.ProvideWSHandler(hub, pubSub, jwtConfig)
|
||||
app := NewApp(cfg, gormDB, client, authService, authController, adminAuthController, userManageController, handler, hub, pubSub)
|
||||
return app, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user