bfb4fb0ccb
- 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
83 lines
2.7 KiB
Go
83 lines
2.7 KiB
Go
// Package provider 提供全局依赖注入配置
|
|
// 使用 Wire 编译时依赖注入，集中管理所有模块的 Provider Set
|
|
package provider
|
|
|
|
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"
|
|
)
|
|
|
|
// App 应用根容器，持有基础设施组件和各模块的 Controller/Service
|
|
type App struct {
|
|
Config *config.Config
|
|
DB *gorm.DB
|
|
Redis *redis.Client
|
|
AuthService *service.AuthService // Auth 认证服务
|
|
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 创建应用实例
|
|
func NewApp(
|
|
cfg *config.Config,
|
|
gormDB *gorm.DB,
|
|
redisClient *redis.Client,
|
|
authService *service.AuthService,
|
|
authCtrl *authController.AuthController,
|
|
adminAuthCtrl *authController.AdminAuthController,
|
|
userManageCtrl *adminController.UserManageController,
|
|
wsHandler *wsApp.Handler,
|
|
hub *ws.Hub,
|
|
pubsub *ws.PubSub,
|
|
) *App {
|
|
return &App{
|
|
Config: cfg,
|
|
DB: gormDB,
|
|
Redis: redisClient,
|
|
AuthService: authService,
|
|
AuthController: authCtrl,
|
|
AdminAuthController: adminAuthCtrl,
|
|
UserManageController: userManageCtrl,
|
|
WSHandler: wsHandler,
|
|
Hub: hub,
|
|
PubSub: pubsub,
|
|
}
|
|
}
|
|
|
|
// provideDBConfig 从全局 Config 中提取 DatabaseConfig
|
|
func provideDBConfig(cfg *config.Config) *config.DatabaseConfig {
|
|
return &cfg.Database
|
|
}
|
|
|
|
// provideRedisConfig 从全局 Config 中提取 RedisConfig
|
|
func provideRedisConfig(cfg *config.Config) *config.RedisConfig {
|
|
return &cfg.Redis
|
|
}
|
|
|
|
// provideJWTConfig 从全局 Config 中提取 JWTConfig
|
|
func provideJWTConfig(cfg *config.Config) *config.JWTConfig {
|
|
return &cfg.JWT
|
|
}
|
|
|
|
// InfraSet 基础设施层 Provider Set
|
|
var InfraSet = wire.NewSet(
|
|
provideDBConfig,
|
|
provideRedisConfig,
|
|
provideJWTConfig,
|
|
db.NewPostgres,
|
|
db.NewRedis,
|
|
NewApp,
|
|
)
|