优化数据库,日志,命令模块

pkg/database/database.go

@@ -5,7 +5,6 @@ import (
 	"database/sql"
 	"fmt"
 	"os"
-	"path/filepath"
 	"strings"
 	"sync"
 
@@ -16,65 +15,57 @@ import (
 	_ "github.com/mattn/go-sqlite3"
 )
 
-type DB struct {
-	db     *sql.DB
-	logger logger.Logger
-}
-
+// =========================================================
+// 全局变量
+// =========================================================
 var (
-	dbInstance *DB
+	dbInstance *sql.DB
 	dbOnce     sync.Once
+	dbMutex    sync.RWMutex
 	dbErr      error
 )
 
-func GetInstance(dbConfig *config.DatabaseConfig, log logger.Logger) (*DB, error) {
+// =========================================================
+// GetDB - 获取数据库连接(单例模式)
+// =========================================================
+func GetDB() (*sql.DB, error) {
 	dbOnce.Do(func() {
-		// 兜底: 未注入则使用全局默认日志实例
-		if log == nil {
-			log = logger.DefaultLogger
-		}
-		log.Debugf("开始初始化数据库,路径: %s", dbConfig.Path)
-
-		// 确认数据库目录存在
-		if err := os.MkdirAll(dbConfig.Path, 0755); err != nil {
-			log.Errorf("创建数据库目录失败: %v", err)
-			dbErr = err
+		if dbInstance != nil {
 			return
 		}
 
-		fullPath := filepath.Join(dbConfig.Path, dbConfig.Name)
-		log.Debugf("数据库路径: %s", fullPath)
+		// 确保配置已加载
+		cfg, err := config.LoadConfig()
+		if err != nil {
+			dbErr = fmt.Errorf("加载配置失败: %w", err)
+			return
+		}
 
 		// 构建DSN
-		dsn := fmt.Sprintf("%s?_foreign_keys=on&_journal_mode=WAL&_timeout=5000&cache=shared",
-			fullPath)
-		log.Debugf("DSN: %s", dsn)
+		logger.Debugf("DSN: %s", cfg.Database.DSN)
 
 		// 打开SQLite 连接
-		sqlDB, err := sql.Open("sqlite3", dsn)
+		sqlDB, err := sql.Open("sqlite3", cfg.Database.DSN)
 		if err != nil {
-			log.Errorf("数据库打开失败: %v", err)
-			dbErr = err
+			dbErr = fmt.Errorf("数据库打开失败: %w", err)
 			return
 		}
 
 		// 设置连接池参数
-		sqlDB.SetMaxOpenConns(1)    // SQLite 只支持单连接
-		sqlDB.SetMaxIdleConns(1)    // 保持一个空闲连接
+		sqlDB.SetMaxOpenConns(10)   // 最大打开连接数
+		sqlDB.SetMaxIdleConns(5)    // 保持空闲连接
 		sqlDB.SetConnMaxLifetime(0) // 禁用连接生命周期超时
 		sqlDB.SetConnMaxIdleTime(0) // 禁用空闲连接超时
 
 		// 测试数据库连接
 		if err := sqlDB.Ping(); err != nil {
 			sqlDB.Close()
-			log.Errorf("数据库连接失败: %v", err)
-			dbErr = err
+			dbErr = fmt.Errorf("数据库连接失败: %w", err)
 			return
 		}
 
-		// 赋值 *DB 类型的单例(而非直接复制 *sql.DB)
-		log.Info("数据库连接成功")
-		dbInstance = &DB{sqlDB, log}
+		logger.Debug("数据库连接成功")
+		dbInstance = sqlDB
 	})
 
 	if dbErr != nil {
@@ -84,21 +75,6 @@ func GetInstance(dbConfig *config.DatabaseConfig, log logger.Logger) (*DB, error
 	return dbInstance, nil
 }
 
-// Close 关闭数据库连接
-func (d *DB) Close() error {
-	if d.db != nil {
-		d.logger.Debug("关闭数据库连接")
-		err := d.db.Close()
-		if err != nil {
-			d.logger.Errorf("数据库连接关闭失败: %v", err)
-			return err
-		}
-		d.logger.Info("数据库连接关闭成功")
-		return nil
-	}
-	return nil
-}
-
 func confirmAction(prompt string) bool {
 	reader := bufio.NewReader(os.Stdin)
 
@@ -112,38 +88,38 @@ func confirmAction(prompt string) bool {
 	return response == "y" || response == "yes"
 }
 
-func (d *DB) InitTables(force bool) error {
-	d.logger.Info("开始初始化数据库表...")
+func InitTables(db *sql.DB, force bool) error {
 
 	if force {
 		// 确认是否强制删除
 		if !confirmAction("确认强制删除所有表和触发器？") {
-			d.logger.Info("操作已取消")
+			logger.Info("操作已取消")
+			db.Close()
 			os.Exit(0)
 			return nil
 		}
 
 		// 强制删除所有表和触发器
-		d.logger.Debug("强制删除所有表和触发器...")
-		if err := dropTables(d.db); err != nil {
+		logger.Debug("强制删除所有表和触发器...")
+		if err := dropTables(db); err != nil {
 			return fmt.Errorf("删除表失败: %w", err)
 		}
-		d.logger.Debug("删除所有表和触发器成功")
+		logger.Debug("删除所有表和触发器成功")
 
-		if err := dropTriggers(d.db); err != nil {
+		if err := dropTriggers(db); err != nil {
 			return fmt.Errorf("删除触发器失败: %w", err)
 		}
-		d.logger.Debug("删除所有触发器成功")
+		logger.Debug("删除所有触发器成功")
 	}
 
 	// ✅ 调用 schema.go 中的函数
 	for _, ddl := range CreateTableStatements() {
-		d.logger.Debugf("执行: %s", ddl)
-		if _, err := d.db.Exec(ddl); err != nil {
+		logger.Debugf("执行: %s", ddl)
+		if _, err := db.Exec(ddl); err != nil {
 			return fmt.Errorf("数据表创建失败: %w", err)
 		}
 	}
-	d.logger.Info("数据库表创建成功")
+	logger.Info("数据库表创建成功")
 	/*
 		使用sqlite3命令 测试数据库是否存在表
 		✅ 查询所有表
@@ -174,3 +150,65 @@ func dropTriggers(db *sql.DB) error {
 	}
 	return nil
 }
+
+func CloseDB() error {
+	dbMutex.Lock()
+	defer dbMutex.Unlock()
+
+	if dbInstance == nil {
+		if err := dbInstance.Close(); err != nil {
+			return err
+		}
+		dbInstance = nil
+	}
+	return nil
+}
+
+// 使用事务回滚测试
+func RunTestWithRollback(db *sql.DB, testFunc func(*sql.Tx) error) error {
+	tx, err := db.Begin()
+	if err != nil {
+		return err
+	}
+
+	// 执行测试
+	if err := testFunc(tx); err != nil {
+		tx.Rollback()
+		return err
+	}
+
+	// 回滚事务，所有更改（包括 ID 递增）都会撤销
+	return tx.Rollback()
+}
+
+// 使用示例
+func TestNodeInsert(db *sql.DB) error {
+	logger.Debug("测试数据插入...")
+	return RunTestWithRollback(db, func(tx *sql.Tx) error {
+		// 插入测试数据
+		logger.Debug("执行插入测试数据...")
+
+		_, err := tx.Exec(`
+			INSERT INTO nodes (name, cpus, rack, rank) 
+			VALUES (?, ?, ?, ?)
+			`, "test-node", 64, 1, 1)
+		if err != nil {
+			return err
+		}
+
+		// 验证插入
+		var count int
+		logger.Debug("执行查询测试数据...")
+		err = tx.QueryRow(`
+			SELECT COUNT(*) FROM nodes WHERE name = ?
+			`, "test-node").Scan(&count)
+
+		if err != nil {
+			return err
+		}
+		logger.Infof("测试数据插入成功，共 %d 条", count)
+
+		// 不需要手动删除，回滚会自动撤销
+		return nil
+	})
+}