优化db包

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
+}

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,
+	)
+}