diff --git a/internal/storage/db/postgres.go b/internal/storage/db/postgres.go new file mode 100644 index 0000000..707b586 --- /dev/null +++ b/internal/storage/db/postgres.go @@ -0,0 +1,117 @@ +// Package db 提供基于PostgreSQL的数据存储功能 +// 使用GORM作为ORM库来操作数据库 +// 实现与PostgreSQL数据库的连接和基本操作 +package db + +import ( + "fmt" + "time" + + "git.huangwc.com/pig/pig-farm-controller/internal/logs" + "gorm.io/driver/postgres" + "gorm.io/gorm" +) + +// PostgresStorage 代表基于PostgreSQL的存储实现 +// 使用GORM作为ORM库 +type PostgresStorage struct { + db *gorm.DB + connectionString string + maxOpenConns int + maxIdleConns int + connMaxLifetime int + logger *logs.Logger // 依赖注入的 logger +} + +// NewPostgresStorage 创建并返回一个新的PostgreSQL存储实例 +// 它接收一个 logger 实例,而不是自己创建 +func NewPostgresStorage(connectionString string, maxOpenConns, maxIdleConns, connMaxLifetime int, logger *logs.Logger) *PostgresStorage { + return &PostgresStorage{ + connectionString: connectionString, + maxOpenConns: maxOpenConns, + maxIdleConns: maxIdleConns, + connMaxLifetime: connMaxLifetime, + logger: logger, // 注入 logger + } +} + +// Connect 建立与PostgreSQL数据库的连接 +// 使用GORM建立数据库连接,并使用自定义的 logger 接管 GORM 日志 +func (ps *PostgresStorage) Connect() error { + ps.logger.Info("正在连接PostgreSQL数据库") + + // 创建 GORM 的 logger 适配器 + gormLogger := logs.NewGormLogger(ps.logger) + + var err error + // 在 gorm.Open 时传入我们自定义的 logger + ps.db, err = gorm.Open(postgres.Open(ps.connectionString), &gorm.Config{ + Logger: gormLogger, + }) + if err != nil { + ps.logger.Errorw("数据库连接失败", "error", err) + return fmt.Errorf("数据库连接失败: %w", err) // 使用 %w 进行错误包装 + } + + // 测试连接 + sqlDB, err := ps.db.DB() + if err != nil { + ps.logger.Errorw("获取数据库实例失败", "error", err) + return fmt.Errorf("获取数据库实例失败: %w", err) + } + + if err = sqlDB.Ping(); err != nil { + ps.logger.Errorw("数据库连接测试失败", "error", err) + return fmt.Errorf("数据库连接测试失败: %w", err) + } + + // 设置连接池参数 + sqlDB.SetMaxOpenConns(ps.maxOpenConns) + sqlDB.SetMaxIdleConns(ps.maxIdleConns) + sqlDB.SetConnMaxLifetime(time.Duration(ps.connMaxLifetime) * time.Second) + + ps.logger.Info("PostgreSQL数据库连接成功") + return nil +} + +// Disconnect 断开与PostgreSQL数据库的连接 +// 安全地关闭所有数据库连接 +func (ps *PostgresStorage) Disconnect() error { + if ps.db != nil { + ps.logger.Info("正在断开PostgreSQL数据库连接") + + sqlDB, err := ps.db.DB() + if err != nil { + ps.logger.Errorw("获取数据库实例失败", "error", err) + return fmt.Errorf("获取数据库实例失败: %w", err) + } + + if err := sqlDB.Close(); err != nil { + ps.logger.Errorw("关闭数据库连接失败", "error", err) + return fmt.Errorf("关闭数据库连接失败: %w", err) + } + ps.logger.Info("PostgreSQL数据库连接已断开") + } + return nil +} + +// GetDB 获取GORM数据库实例 +// 用于执行具体的数据库操作 +func (ps *PostgresStorage) GetDB() *gorm.DB { + return ps.db +} + +// Migrate 执行数据库迁移 +func (ps *PostgresStorage) Migrate(models ...interface{}) error { + if len(models) == 0 { + ps.logger.Info("没有需要迁移的数据库模型,跳过迁移步骤") + return nil + } + ps.logger.Info("正在自动迁移数据库表结构") + if err := ps.db.AutoMigrate(models...); err != nil { + ps.logger.Errorw("数据库表结构迁移失败", "error", err) + return fmt.Errorf("数据库表结构迁移失败: %w", err) + } + ps.logger.Info("数据库表结构迁移完成") + return nil +} diff --git a/internal/storage/db/storage.go b/internal/storage/db/storage.go new file mode 100644 index 0000000..f128bbe --- /dev/null +++ b/internal/storage/db/storage.go @@ -0,0 +1,53 @@ +// Package db 提供统一的数据存储接口 +// 定义存储接口规范,支持多种存储后端实现 +// 当前支持PostgreSQL实现 +package db + +import ( + "fmt" + + "git.huangwc.com/pig/pig-farm-controller/internal/config" + "git.huangwc.com/pig/pig-farm-controller/internal/logs" + "gorm.io/gorm" +) + +// Storage 代表统一的存储接口 +// 所有存储实现都需要实现此接口定义的方法 +type Storage interface { + // Connect 建立与存储后端的连接 + Connect() error + + // Disconnect 断开与存储后端的连接 + Disconnect() error + + // GetDB 获取数据库实例 + GetDB() *gorm.DB + + // Migrate 执行数据库迁移 + // 参数为需要迁移的 GORM 模型 + Migrate(models ...interface{}) error +} + +// NewStorage 创建并返回一个存储实例 +// 根据配置返回相应的存储实现 +func NewStorage(cfg config.DatabaseConfig, logger *logs.Logger) Storage { + // 构建数据库连接字符串 + connectionString := fmt.Sprintf( + "user=%s password=%s dbname=%s host=%s port=%d sslmode=%s", + cfg.Username, + cfg.Password, + cfg.DBName, + cfg.Host, + cfg.Port, + cfg.SSLMode, + ) + + // 当前默认返回PostgreSQL存储实现,并将 logger 注入 + return NewPostgresStorage( + connectionString, + cfg.MaxOpenConns, + cfg.MaxIdleConns, + cfg.ConnMaxLifetime, + logger, + ) +}