5d6c982c9c
- Migrate all log.Printf/log.Println calls to structured zap logging - Use cmp.Or for cleaner default value handling - Replace manual loops with slices.ContainsFunc/IndexFunc - Add batch query methods (IsLikedBatch, IsFavoritedBatch) to solve N+1 problem - Update JSON tags from omitempty to omitzero for numeric fields - Use errgroup-style wg.Go for worker goroutines - Remove duplicate casbin/v2 dependency (keeping v3) - Add plans/ to gitignore
138 lines
3.6 KiB
Go
138 lines
3.6 KiB
Go
package wire
|
||
|
||
import (
|
||
"time"
|
||
|
||
"carrot_bbs/internal/cache"
|
||
"carrot_bbs/internal/config"
|
||
"carrot_bbs/internal/model"
|
||
"carrot_bbs/internal/pkg/email"
|
||
"carrot_bbs/internal/pkg/gorse"
|
||
"carrot_bbs/internal/pkg/openai"
|
||
"carrot_bbs/internal/pkg/redis"
|
||
"carrot_bbs/internal/pkg/s3"
|
||
"carrot_bbs/internal/pkg/sse"
|
||
"carrot_bbs/internal/repository"
|
||
|
||
"github.com/google/wire"
|
||
"go.uber.org/zap"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// InfrastructureSet 基础设施 Provider Set
|
||
var InfrastructureSet = wire.NewSet(
|
||
// 配置
|
||
ProvideConfig,
|
||
|
||
// 日志
|
||
ProvideLogger,
|
||
|
||
// 数据库
|
||
ProvideDB,
|
||
|
||
// 事务管理器
|
||
ProvideTransactionManager,
|
||
|
||
// Redis 和缓存
|
||
ProvideRedisClient,
|
||
ProvideCache,
|
||
|
||
// SSE Hub
|
||
ProvideSSEHub,
|
||
|
||
// 外部服务客户端
|
||
ProvideGorseClient,
|
||
ProvideOpenAIClient,
|
||
ProvideEmailClient,
|
||
ProvideS3Client,
|
||
)
|
||
|
||
// ProvideConfig 加载配置
|
||
func ProvideConfig() (*config.Config, error) {
|
||
return config.Load("configs/config.yaml")
|
||
}
|
||
|
||
// ProvideDB 提供数据库连接
|
||
func ProvideDB(cfg *config.Config) (*gorm.DB, error) {
|
||
return model.NewDB(&cfg.Database)
|
||
}
|
||
|
||
// ProvideRedisClient 提供 Redis 客户端
|
||
func ProvideRedisClient(cfg *config.Config) *redis.Client {
|
||
client, err := redis.New(&cfg.Redis)
|
||
if err != nil {
|
||
zap.L().Warn("Failed to connect to Redis, falling back to memory cache",
|
||
zap.Error(err),
|
||
)
|
||
return nil
|
||
}
|
||
return client
|
||
}
|
||
|
||
// ProvideCache 提供缓存实例
|
||
func ProvideCache(cfg *config.Config, redisClient *redis.Client) cache.Cache {
|
||
cache.Configure(cache.Settings{
|
||
Enabled: cfg.Cache.Enabled,
|
||
KeyPrefix: cfg.Cache.KeyPrefix,
|
||
DefaultTTL: time.Duration(cfg.Cache.DefaultTTL) * time.Second,
|
||
NullTTL: time.Duration(cfg.Cache.NullTTL) * time.Second,
|
||
JitterRatio: cfg.Cache.JitterRatio,
|
||
PostListTTL: time.Duration(cfg.Cache.Modules.PostList) * time.Second,
|
||
ConversationTTL: time.Duration(cfg.Cache.Modules.Conversation) * time.Second,
|
||
UnreadCountTTL: time.Duration(cfg.Cache.Modules.UnreadCount) * time.Second,
|
||
GroupMembersTTL: time.Duration(cfg.Cache.Modules.GroupMembers) * time.Second,
|
||
DisableFlushDB: cfg.Cache.DisableFlushDB,
|
||
})
|
||
|
||
// 直接创建 RedisCache 实例，不依赖全局变量
|
||
if redisClient == nil {
|
||
zap.L().Warn("Redis client is nil, cache will not be available")
|
||
return nil
|
||
}
|
||
|
||
cacheInstance := cache.NewRedisCache(redisClient)
|
||
zap.L().Info("Initialized Redis cache via dependency injection")
|
||
return cacheInstance
|
||
}
|
||
|
||
// ProvideSSEHub 提供 SSE Hub
|
||
func ProvideSSEHub() *sse.Hub {
|
||
return sse.NewHub()
|
||
}
|
||
|
||
// ProvideGorseClient 提供 Gorse 客户端
|
||
func ProvideGorseClient(cfg *config.Config) gorse.Client {
|
||
return gorse.NewClient(gorse.ConfigFromAppConfig(&cfg.Gorse))
|
||
}
|
||
|
||
// ProvideOpenAIClient 提供 OpenAI 客户端
|
||
func ProvideOpenAIClient(cfg *config.Config) openai.Client {
|
||
return openai.NewClient(openai.ConfigFromAppConfig(&cfg.OpenAI))
|
||
}
|
||
|
||
// ProvideEmailClient 提供邮件客户端
|
||
func ProvideEmailClient(cfg *config.Config) email.Client {
|
||
return email.NewClient(email.ConfigFromAppConfig(&cfg.Email))
|
||
}
|
||
|
||
// ProvideS3Client 提供 S3 客户端
|
||
func ProvideS3Client(cfg *config.Config) (*s3.Client, error) {
|
||
return s3.New(&cfg.S3)
|
||
}
|
||
|
||
// ProvideTransactionManager 提供事务管理器
|
||
func ProvideTransactionManager(db *gorm.DB) repository.TransactionManager {
|
||
return repository.NewTransactionManager(db)
|
||
}
|
||
|
||
// ProvideLogger 提供 Zap Logger
|
||
func ProvideLogger() *zap.Logger {
|
||
logger, err := zap.NewProduction()
|
||
if err != nil {
|
||
panic("Failed to create logger: " + err.Error())
|
||
}
|
||
// 设置为全局 logger，以便在其他地方使用 zap.L()
|
||
zap.ReplaceGlobals(logger)
|
||
return logger
|
||
}
|