ffdc3e3e6b
完成所有Handler的依赖注入改造： - AuthHandler: 认证相关功能 - UserHandler: 用户管理功能 - TextureHandler: 材质管理功能 - ProfileHandler: 档案管理功能 - CaptchaHandler: 验证码功能 - YggdrasilHandler: Yggdrasil API功能 新增错误类型定义： - internal/errors/errors.go: 统一的错误类型和工厂函数 更新main.go： - 使用container.NewContainer创建依赖容器 - 使用handler.RegisterRoutesWithDI注册路由 代码遵循Go最佳实践： - 依赖通过构造函数注入 - Handler通过结构体方法实现 - 统一的错误处理模式 - 清晰的分层架构
138 lines
2.9 KiB
Go
138 lines
2.9 KiB
Go
package container
|
||
|
||
import (
|
||
"carrotskin/internal/repository"
|
||
"carrotskin/pkg/auth"
|
||
"carrotskin/pkg/redis"
|
||
"carrotskin/pkg/storage"
|
||
|
||
"go.uber.org/zap"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// Container 依赖注入容器
|
||
// 集中管理所有依赖，便于测试和维护
|
||
type Container struct {
|
||
// 基础设施依赖
|
||
DB *gorm.DB
|
||
Redis *redis.Client
|
||
Logger *zap.Logger
|
||
JWT *auth.JWTService
|
||
Storage *storage.StorageClient
|
||
|
||
// Repository层
|
||
UserRepo repository.UserRepository
|
||
ProfileRepo repository.ProfileRepository
|
||
TextureRepo repository.TextureRepository
|
||
TokenRepo repository.TokenRepository
|
||
ConfigRepo repository.SystemConfigRepository
|
||
}
|
||
|
||
// NewContainer 创建依赖容器
|
||
func NewContainer(
|
||
db *gorm.DB,
|
||
redisClient *redis.Client,
|
||
logger *zap.Logger,
|
||
jwtService *auth.JWTService,
|
||
storageClient *storage.StorageClient,
|
||
) *Container {
|
||
c := &Container{
|
||
DB: db,
|
||
Redis: redisClient,
|
||
Logger: logger,
|
||
JWT: jwtService,
|
||
Storage: storageClient,
|
||
}
|
||
|
||
// 初始化Repository
|
||
c.UserRepo = repository.NewUserRepository(db)
|
||
c.ProfileRepo = repository.NewProfileRepository(db)
|
||
c.TextureRepo = repository.NewTextureRepository(db)
|
||
c.TokenRepo = repository.NewTokenRepository(db)
|
||
c.ConfigRepo = repository.NewSystemConfigRepository(db)
|
||
|
||
return c
|
||
}
|
||
|
||
// NewTestContainer 创建测试用容器（可注入mock依赖）
|
||
func NewTestContainer(opts ...Option) *Container {
|
||
c := &Container{}
|
||
for _, opt := range opts {
|
||
opt(c)
|
||
}
|
||
return c
|
||
}
|
||
|
||
// Option 容器配置选项
|
||
type Option func(*Container)
|
||
|
||
// WithDB 设置数据库连接
|
||
func WithDB(db *gorm.DB) Option {
|
||
return func(c *Container) {
|
||
c.DB = db
|
||
}
|
||
}
|
||
|
||
// WithRedis 设置Redis客户端
|
||
func WithRedis(redis *redis.Client) Option {
|
||
return func(c *Container) {
|
||
c.Redis = redis
|
||
}
|
||
}
|
||
|
||
// WithLogger 设置日志
|
||
func WithLogger(logger *zap.Logger) Option {
|
||
return func(c *Container) {
|
||
c.Logger = logger
|
||
}
|
||
}
|
||
|
||
// WithJWT 设置JWT服务
|
||
func WithJWT(jwt *auth.JWTService) Option {
|
||
return func(c *Container) {
|
||
c.JWT = jwt
|
||
}
|
||
}
|
||
|
||
// WithStorage 设置存储客户端
|
||
func WithStorage(storage *storage.StorageClient) Option {
|
||
return func(c *Container) {
|
||
c.Storage = storage
|
||
}
|
||
}
|
||
|
||
// WithUserRepo 设置用户仓储
|
||
func WithUserRepo(repo repository.UserRepository) Option {
|
||
return func(c *Container) {
|
||
c.UserRepo = repo
|
||
}
|
||
}
|
||
|
||
// WithProfileRepo 设置档案仓储
|
||
func WithProfileRepo(repo repository.ProfileRepository) Option {
|
||
return func(c *Container) {
|
||
c.ProfileRepo = repo
|
||
}
|
||
}
|
||
|
||
// WithTextureRepo 设置材质仓储
|
||
func WithTextureRepo(repo repository.TextureRepository) Option {
|
||
return func(c *Container) {
|
||
c.TextureRepo = repo
|
||
}
|
||
}
|
||
|
||
// WithTokenRepo 设置令牌仓储
|
||
func WithTokenRepo(repo repository.TokenRepository) Option {
|
||
return func(c *Container) {
|
||
c.TokenRepo = repo
|
||
}
|
||
}
|
||
|
||
// WithConfigRepo 设置系统配置仓储
|
||
func WithConfigRepo(repo repository.SystemConfigRepository) Option {
|
||
return func(c *Container) {
|
||
c.ConfigRepo = repo
|
||
}
|
||
}
|