重构架构

2026-02-20 18:44:43 +08:00
parent aba7b68439
commit cc71248ef4
52 changed files with 1404 additions and 2360 deletions
--- a/pkg/database/database.go
+++ b/pkg/database/database.go
@@ -0,0 +1,176 @@
+package database
+
+import (
+	"bufio"
+	"database/sql"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+	"sync"
+
+	"sunhpc/pkg/config"
+	"sunhpc/pkg/logger"
+
+	_ "github.com/go-sql-driver/mysql"
+	_ "github.com/mattn/go-sqlite3"
+)
+
+type DB struct {
+	db     *sql.DB
+	logger logger.Logger
+}
+
+var (
+	dbInstance *DB
+	dbOnce     sync.Once
+	dbErr      error
+)
+
+func GetInstance(dbConfig *config.DatabaseConfig, log logger.Logger) (*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
+			return
+		}
+
+		fullPath := filepath.Join(dbConfig.Path, dbConfig.Name)
+		log.Debugf("数据库路径: %s", fullPath)
+
+		// 构建DSN
+		dsn := fmt.Sprintf("%s?_foreign_keys=on&_journal_mode=WAL&_timeout=5000&cache=shared",
+			fullPath)
+		log.Debugf("DSN: %s", dsn)
+
+		// 打开SQLite 连接
+		sqlDB, err := sql.Open("sqlite3", dsn)
+		if err != nil {
+			log.Errorf("数据库打开失败: %v", err)
+			dbErr = err
+			return
+		}
+
+		// 设置连接池参数
+		sqlDB.SetMaxOpenConns(1)    // SQLite 只支持单连接
+		sqlDB.SetMaxIdleConns(1)    // 保持一个空闲连接
+		sqlDB.SetConnMaxLifetime(0) // 禁用连接生命周期超时
+		sqlDB.SetConnMaxIdleTime(0) // 禁用空闲连接超时
+
+		// 测试数据库连接
+		if err := sqlDB.Ping(); err != nil {
+			sqlDB.Close()
+			log.Errorf("数据库连接失败: %v", err)
+			dbErr = err
+			return
+		}
+
+		// 赋值 *DB 类型的单例(而非直接复制 *sql.DB)
+		log.Info("数据库连接成功")
+		dbInstance = &DB{sqlDB, log}
+	})
+
+	if dbErr != nil {
+		return nil, dbErr
+	}
+
+	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)
+
+	logger.Warnf("%s [Y/Yes]: ", prompt)
+	response, err := reader.ReadString('\n')
+	if err != nil {
+		return false
+	}
+
+	response = strings.ToLower(strings.TrimSpace(response))
+	return response == "y" || response == "yes"
+}
+
+func (d *DB) InitTables(force bool) error {
+	d.logger.Info("开始初始化数据库表...")
+
+	if force {
+		// 确认是否强制删除
+		if !confirmAction("确认强制删除所有表和触发器？") {
+			d.logger.Info("操作已取消")
+			os.Exit(0)
+			return nil
+		}
+
+		// 强制删除所有表和触发器
+		d.logger.Debug("强制删除所有表和触发器...")
+		if err := dropTables(d.db); err != nil {
+			return fmt.Errorf("删除表失败: %w", err)
+		}
+		d.logger.Debug("删除所有表和触发器成功")
+
+		if err := dropTriggers(d.db); err != nil {
+			return fmt.Errorf("删除触发器失败: %w", err)
+		}
+		d.logger.Debug("删除所有触发器成功")
+	}
+
+	// ✅ 调用 schema.go 中的函数
+	for _, ddl := range CreateTableStatements() {
+		d.logger.Debugf("执行: %s", ddl)
+		if _, err := d.db.Exec(ddl); err != nil {
+			return fmt.Errorf("数据表创建失败: %w", err)
+		}
+	}
+	d.logger.Info("数据库表创建成功")
+	/*
+		使用sqlite3命令 测试数据库是否存在表
+		✅ 查询所有表
+		sqlite3 /var/lib/sunhpc/sunhpc.db
+		.tables # 查看所有表
+		select * from sqlite_master where type='table'; # 查看表定义
+		PRAGMA integrity_check; # 检查数据库完整性
+	*/
+	return nil
+}
+
+func dropTables(db *sql.DB) error {
+	// ✅ 调用 schema.go 中的函数
+	for _, table := range DropTableOrder() {
+		if _, err := db.Exec(fmt.Sprintf("DROP TABLE IF EXISTS `%s`", table)); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func dropTriggers(db *sql.DB) error {
+	// ✅ 调用 schema.go 中的函数
+	for _, trigger := range DropTriggerStatements() {
+		if _, err := db.Exec(fmt.Sprintf("DROP TRIGGER IF EXISTS `%s`", trigger)); err != nil {
+			return err
+		}
+	}
+	return nil
+}