127 lines
3.4 KiB
Go
127 lines
3.4 KiB
Go
package factory
|
||
|
||
import (
|
||
"fmt"
|
||
"time"
|
||
|
||
"git.toowon.com/jimmy/go-common/config"
|
||
"git.toowon.com/jimmy/go-common/email"
|
||
"git.toowon.com/jimmy/go-common/logger"
|
||
"git.toowon.com/jimmy/go-common/sms"
|
||
"gorm.io/driver/mysql"
|
||
"gorm.io/driver/postgres"
|
||
"gorm.io/driver/sqlite"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// Factory 工厂类，用于从配置创建各种客户端对象
|
||
type Factory struct {
|
||
cfg *config.Config
|
||
}
|
||
|
||
// NewFactory 创建工厂实例
|
||
func NewFactory(cfg *config.Config) *Factory {
|
||
return &Factory{
|
||
cfg: cfg,
|
||
}
|
||
}
|
||
|
||
// NewFactoryFromFile 从配置文件创建工厂实例（便捷方法）
|
||
// filePath: 配置文件路径
|
||
func NewFactoryFromFile(filePath string) (*Factory, error) {
|
||
cfg, err := config.LoadFromFile(filePath)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("failed to load config: %w", err)
|
||
}
|
||
return NewFactory(cfg), nil
|
||
}
|
||
|
||
// GetEmailClient 获取邮件客户端（已初始化）
|
||
// 返回已初始化的邮件客户端对象，可直接使用
|
||
func (f *Factory) GetEmailClient() (*email.Email, error) {
|
||
if f.cfg.Email == nil {
|
||
return nil, fmt.Errorf("email config is nil")
|
||
}
|
||
return email.NewEmail(f.cfg.Email)
|
||
}
|
||
|
||
// GetSMSClient 获取短信客户端（已初始化）
|
||
// 返回已初始化的短信客户端对象，可直接使用
|
||
func (f *Factory) GetSMSClient() (*sms.SMS, error) {
|
||
if f.cfg.SMS == nil {
|
||
return nil, fmt.Errorf("SMS config is nil")
|
||
}
|
||
return sms.NewSMS(f.cfg.SMS)
|
||
}
|
||
|
||
// GetLogger 获取日志记录器（已初始化）
|
||
// 返回已初始化的日志记录器对象，可直接使用
|
||
func (f *Factory) GetLogger() (*logger.Logger, error) {
|
||
if f.cfg.Logger == nil {
|
||
// 如果没有配置，使用默认配置创建
|
||
return logger.NewLogger(nil)
|
||
}
|
||
return logger.NewLogger(f.cfg.Logger)
|
||
}
|
||
|
||
// GetDatabase 获取数据库连接对象（已初始化）
|
||
// 返回已初始化的GORM数据库对象，可直接使用
|
||
func (f *Factory) GetDatabase() (*gorm.DB, error) {
|
||
if f.cfg.Database == nil {
|
||
return nil, fmt.Errorf("database config is nil")
|
||
}
|
||
|
||
// 获取DSN
|
||
dsn, err := f.cfg.GetDatabaseDSN()
|
||
if err != nil {
|
||
return nil, fmt.Errorf("failed to get DSN: %w", err)
|
||
}
|
||
|
||
// 根据数据库类型创建连接
|
||
var db *gorm.DB
|
||
switch f.cfg.Database.Type {
|
||
case "mysql":
|
||
db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
|
||
case "postgres":
|
||
db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{})
|
||
case "sqlite":
|
||
db, err = gorm.Open(sqlite.Open(dsn), &gorm.Config{})
|
||
default:
|
||
return nil, fmt.Errorf("unsupported database type: %s", f.cfg.Database.Type)
|
||
}
|
||
|
||
if err != nil {
|
||
return nil, fmt.Errorf("failed to connect to database: %w", err)
|
||
}
|
||
|
||
// 配置连接池
|
||
sqlDB, err := db.DB()
|
||
if err != nil {
|
||
return nil, fmt.Errorf("failed to get sql.DB: %w", err)
|
||
}
|
||
|
||
if f.cfg.Database.MaxOpenConns > 0 {
|
||
sqlDB.SetMaxOpenConns(f.cfg.Database.MaxOpenConns)
|
||
}
|
||
if f.cfg.Database.MaxIdleConns > 0 {
|
||
sqlDB.SetMaxIdleConns(f.cfg.Database.MaxIdleConns)
|
||
}
|
||
if f.cfg.Database.ConnMaxLifetime > 0 {
|
||
sqlDB.SetConnMaxLifetime(time.Duration(f.cfg.Database.ConnMaxLifetime) * time.Second)
|
||
}
|
||
|
||
return db, nil
|
||
}
|
||
|
||
// GetRedisConfig 获取Redis配置（用于创建Redis客户端）
|
||
// 返回Redis配置对象，调用方可以使用此配置创建Redis客户端
|
||
// 注意：Go标准库没有Redis客户端，需要调用方使用第三方库（如go-redis/redis）创建
|
||
func (f *Factory) GetRedisConfig() *config.RedisConfig {
|
||
return f.cfg.Redis
|
||
}
|
||
|
||
// GetConfig 获取配置对象
|
||
func (f *Factory) GetConfig() *config.Config {
|
||
return f.cfg
|
||
}
|