136 lines
5.6 KiB
Go
136 lines
5.6 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"
|
|
contactController "github.com/echochat/backend/app/contact/controller"
|
|
fileController "github.com/echochat/backend/app/file/controller"
|
|
groupController "github.com/echochat/backend/app/group/controller"
|
|
imController "github.com/echochat/backend/app/im/controller"
|
|
imHandler "github.com/echochat/backend/app/im/handler"
|
|
wsApp "github.com/echochat/backend/app/ws"
|
|
"github.com/echochat/backend/config"
|
|
"github.com/echochat/backend/pkg/db"
|
|
"github.com/echochat/backend/pkg/storage"
|
|
"github.com/echochat/backend/pkg/ws"
|
|
"github.com/google/wire"
|
|
"github.com/minio/minio-go/v7"
|
|
"github.com/redis/go-redis/v9"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// App 应用根容器，持有基础设施组件和各模块的 Controller/Service
|
|
type App struct {
|
|
Config *config.Config
|
|
DB *gorm.DB
|
|
Redis *redis.Client
|
|
MinioClient *minio.Client // MinIO 对象存储客户端
|
|
AuthService *service.AuthService // Auth 认证服务
|
|
AuthController *authController.AuthController // 前台认证控制器
|
|
AdminAuthController *authController.AdminAuthController // 后台认证控制器
|
|
UserManageController *adminController.UserManageController // 管理端用户管理控制器
|
|
OnlineController *adminController.OnlineController // 管理端在线监控控制器
|
|
ContactManageController *adminController.ContactManageController // 管理端好友关系管理控制器
|
|
GroupManageController *adminController.GroupManageController // 管理端群组管理控制器
|
|
MessageManageController *adminController.MessageManageController // 管理端消息管理控制器
|
|
WSHandler *wsApp.Handler // WebSocket 连接处理器
|
|
Hub *ws.Hub // WebSocket Hub 连接管理
|
|
PubSub *ws.PubSub // Redis Pub/Sub 消息路由
|
|
OnlineService *wsApp.OnlineService // 在线状态管理服务
|
|
ContactController *contactController.ContactController // 联系人控制器
|
|
IMController *imController.IMController // IM 即时通讯控制器
|
|
IMEventHandler *imHandler.EventHandler // IM WS 事件处理器
|
|
OfflinePusher *imHandler.OfflinePusher // 离线消息推送器
|
|
FileController *fileController.FileController // 文件上传控制器
|
|
GroupController *groupController.GroupController // 群聊管理控制器
|
|
}
|
|
|
|
// NewApp 创建应用实例
|
|
func NewApp(
|
|
cfg *config.Config,
|
|
gormDB *gorm.DB,
|
|
redisClient *redis.Client,
|
|
minioClient *minio.Client,
|
|
authService *service.AuthService,
|
|
authCtrl *authController.AuthController,
|
|
adminAuthCtrl *authController.AdminAuthController,
|
|
userManageCtrl *adminController.UserManageController,
|
|
onlineCtrl *adminController.OnlineController,
|
|
contactManageCtrl *adminController.ContactManageController,
|
|
groupManageCtrl *adminController.GroupManageController,
|
|
msgManageCtrl *adminController.MessageManageController,
|
|
wsHandler *wsApp.Handler,
|
|
hub *ws.Hub,
|
|
pubsub *ws.PubSub,
|
|
onlineService *wsApp.OnlineService,
|
|
contactCtrl *contactController.ContactController,
|
|
imCtrl *imController.IMController,
|
|
imEventHandler *imHandler.EventHandler,
|
|
offlinePusher *imHandler.OfflinePusher,
|
|
fileCtrl *fileController.FileController,
|
|
groupCtrl *groupController.GroupController,
|
|
) *App {
|
|
// 注入离线消息推送器到 WS Handler
|
|
wsHandler.SetOfflinePusher(offlinePusher)
|
|
|
|
return &App{
|
|
Config: cfg,
|
|
DB: gormDB,
|
|
Redis: redisClient,
|
|
MinioClient: minioClient,
|
|
AuthService: authService,
|
|
AuthController: authCtrl,
|
|
AdminAuthController: adminAuthCtrl,
|
|
UserManageController: userManageCtrl,
|
|
OnlineController: onlineCtrl,
|
|
ContactManageController: contactManageCtrl,
|
|
GroupManageController: groupManageCtrl,
|
|
MessageManageController: msgManageCtrl,
|
|
WSHandler: wsHandler,
|
|
Hub: hub,
|
|
PubSub: pubsub,
|
|
OnlineService: onlineService,
|
|
ContactController: contactCtrl,
|
|
IMController: imCtrl,
|
|
IMEventHandler: imEventHandler,
|
|
OfflinePusher: offlinePusher,
|
|
FileController: fileCtrl,
|
|
GroupController: groupCtrl,
|
|
}
|
|
}
|
|
|
|
// 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
|
|
}
|
|
|
|
// provideMinioConfig 从全局 Config 中提取 MinioConfig
|
|
func provideMinioConfig(cfg *config.Config) *config.MinioConfig {
|
|
return &cfg.Minio
|
|
}
|
|
|
|
// InfraSet 基础设施层 Provider Set
|
|
var InfraSet = wire.NewSet(
|
|
provideDBConfig,
|
|
provideRedisConfig,
|
|
provideJWTConfig,
|
|
provideMinioConfig,
|
|
db.NewPostgres,
|
|
db.NewRedis,
|
|
storage.NewMinioClient,
|
|
NewApp,
|
|
)
|