实现日志

internal/storage/db/postgres.go

@@ -5,9 +5,9 @@ package db
 
 import (
 	"fmt"
-	"log"
 	"time"
 
+	"git.huangwc.com/pig/pig-farm-controller/internal/logs"
 	"gorm.io/driver/postgres"
 	"gorm.io/gorm"
 )
@@ -29,6 +29,9 @@ type PostgresStorage struct {
 
 	// connMaxLifetime 连接最大生命周期(秒)
 	connMaxLifetime int
+
+	// logger 日志记录器
+	logger *logs.Logger
 }
 
 // NewPostgresStorage 创建并返回一个新的PostgreSQL存储实例
@@ -39,25 +42,31 @@ func NewPostgresStorage(connectionString string, maxOpenConns, maxIdleConns, con
 		maxOpenConns:     maxOpenConns,
 		maxIdleConns:     maxIdleConns,
 		connMaxLifetime:  connMaxLifetime,
+		logger:           logs.NewLogger(),
 	}
 }
 
 // Connect 建立与PostgreSQL数据库的连接
 // 使用GORM建立数据库连接
 func (ps *PostgresStorage) Connect() error {
+	ps.logger.Info("Connecting to PostgreSQL database")
+
 	var err error
 	ps.db, err = gorm.Open(postgres.Open(ps.connectionString), &gorm.Config{})
 	if err != nil {
+		ps.logger.Error(fmt.Sprintf("Failed to connect to database: %v", err))
 		return fmt.Errorf("failed to connect to database: %v", err)
 	}
 
 	// 测试连接
 	sqlDB, err := ps.db.DB()
 	if err != nil {
+		ps.logger.Error(fmt.Sprintf("Failed to get database instance: %v", err))
 		return fmt.Errorf("failed to get database instance: %v", err)
 	}
 
 	if err = sqlDB.Ping(); err != nil {
+		ps.logger.Error(fmt.Sprintf("Failed to ping database: %v", err))
 		return fmt.Errorf("failed to ping database: %v", err)
 	}
 
@@ -66,7 +75,7 @@ func (ps *PostgresStorage) Connect() error {
 	sqlDB.SetMaxIdleConns(ps.maxIdleConns)
 	sqlDB.SetConnMaxLifetime(time.Duration(ps.connMaxLifetime) * time.Second)
 
-	log.Println("Successfully connected to PostgreSQL database")
+	ps.logger.Info("Successfully connected to PostgreSQL database")
 	return nil
 }
 
@@ -74,15 +83,19 @@ func (ps *PostgresStorage) Connect() error {
 // 安全地关闭所有数据库连接
 func (ps *PostgresStorage) Disconnect() error {
 	if ps.db != nil {
+		ps.logger.Info("Disconnecting from PostgreSQL database")
+
 		sqlDB, err := ps.db.DB()
 		if err != nil {
+			ps.logger.Error(fmt.Sprintf("Failed to get database instance: %v", err))
 			return fmt.Errorf("failed to get database instance: %v", err)
 		}
 
 		if err := sqlDB.Close(); err != nil {
+			ps.logger.Error(fmt.Sprintf("Failed to close database connection: %v", err))
 			return fmt.Errorf("failed to close database connection: %v", err)
 		}
-		log.Println("Successfully disconnected from PostgreSQL database")
+		ps.logger.Info("Successfully disconnected from PostgreSQL database")
 	}
 	return nil
 }