重构架构

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
+}
--- a/pkg/database/schema.go
+++ b/pkg/database/schema.go
@@ -0,0 +1,294 @@
+// Package db defines the database schema.
+package database
+
+// CurrentSchemaVersion returns the current schema version (for migrations)
+func CurrentSchemaVersion() int {
+	return 1
+}
+
+// CreateTableStatements returns a list of CREATE TABLE statements.
+func CreateTableStatements() []string {
+	return []string{
+		createAliasesTable(),
+		createAttributesTable(),
+		createBootactionTable(),
+		createDistributionsTable(),
+		createFirewallsTable(),
+		createNetworksTable(),
+		createPartitionsTable(),
+		createPublicKeysTable(),
+		createSoftwareTable(),
+		createNodesTable(),
+		createSubnetsTable(),
+		createTrg_nodes_before_delete(),
+	}
+}
+
+// DropTableOrder returns table names in reverse dependency order for safe DROP.
+func DropTableOrder() []string {
+	return []string{
+		"aliases",
+		"attributes",
+		"bootactions",
+		"distributions",
+		"firewalls",
+		"networks",
+		"partitions",
+		"publickeys",
+		"software",
+		"nodes",
+		"subnets",
+	}
+}
+
+func DropTriggerStatements() []string {
+	return []string{
+		"trg_nodes_before_delete",
+	}
+}
+
+// --- Private DDL Functions ---
+func createAliasesTable() string {
+	return `
+CREATE TABLE IF NOT EXISTS aliases (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    node_id INTEGER NOT NULL,
+    alias TEXT NOT NULL,
+    CONSTRAINT fk_aliases_node FOREIGN KEY(node_id) REFERENCES nodes(id),
+    UNIQUE(node_id, alias)
+);
+create index if not exists idx_aliases_node on aliases (node_id);
+`
+}
+
+func createAttributesTable() string {
+	return `
+CREATE TABLE IF NOT EXISTS attributes (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    node_id INTEGER NOT NULL,
+    attr TEXT NOT NULL,
+    value TEXT,
+    shadow TEXT,
+    CONSTRAINT fk_attributes_node FOREIGN KEY(node_id) REFERENCES nodes(id)
+);
+create index if not exists idx_attributes_node on attributes (node_id);
+`
+}
+
+func createBootactionTable() string {
+	return `
+CREATE TABLE IF NOT EXISTS bootactions (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    node_id INTEGER NOT NULL,
+    action TEXT,
+    kernel TEXT,
+    initrd TEXT,
+    cmdline TEXT,
+    CONSTRAINT fk_bootactions_node FOREIGN KEY(node_id) REFERENCES nodes(id),
+    UNIQUE(node_id)
+);
+create index if not exists idx_bootactions_node on bootactions (node_id);
+`
+}
+
+func createDistributionsTable() string {
+	return `
+CREATE TABLE IF NOT EXISTS distributions (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    node_id INTEGER,
+    name TEXT NOT NULL,
+    version TEXT,
+    lang TEXT,
+    os_release TEXT,
+    constraint distributions_nodes_fk FOREIGN KEY(node_id) REFERENCES nodes(id)
+);
+create index if not exists idx_distributions_node on distributions (node_id);
+`
+}
+
+func createFirewallsTable() string {
+	return `
+CREATE TABLE IF NOT EXISTS firewalls (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    node_id INTEGER,
+    rulename TEXT NOT NULL,
+    rulesrc TEXT NOT NULL,
+    insubnet INTEGER,
+    outsubnet INTEGER,
+    service TEXT,
+    protocol TEXT,
+    action TEXT,
+    chain TEXT,
+    flags TEXT,
+    comment TEXT,
+    constraint firewalls_nodes_fk FOREIGN KEY(node_id) REFERENCES nodes(id)
+);
+create index if not exists idx_firewalls_node on firewalls (node_id);
+`
+}
+
+func createNetworksTable() string {
+	return `
+CREATE TABLE IF NOT EXISTS networks (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    node_id INTEGER,
+    subnet_id INTEGER,
+    mac TEXT,
+    ip TEXT,
+    name TEXT,
+    device TEXT,
+    module TEXT,
+    vlanid INTEGER,
+    options TEXT,
+    channel TEXT,
+    disable_kvm INTEGER NOT NULL DEFAULT 0 CHECK (disable_kvm IN (0, 1)),
+    constraint networks_nodes_fk FOREIGN KEY(node_id) REFERENCES nodes(id),
+    constraint networks_subnets_fk FOREIGN KEY(subnet_id) REFERENCES subnets(id)
+);
+create index if not exists idx_networks_node on networks (node_id);
+`
+}
+
+func createNodesTable() string {
+	return `
+CREATE TABLE IF NOT EXISTS nodes (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    name TEXT NOT NULL,
+    cpus INTEGER NOT NULL,
+    rack INTEGER NOT NULL,
+    rank INTEGER NOT NULL,
+    arch TEXT,
+    os TEXT,
+    runaction TEXT,
+    installaction TEXT
+);
+create index if not exists idx_nodes_name on nodes (name);
+`
+}
+
+func createPartitionsTable() string {
+	return `
+CREATE TABLE IF NOT EXISTS partitions (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    node_id INTEGER,
+    device TEXT NOT NULL,
+    formatflags TEXT NOT NULL,
+    fstype TEXT NOT NULL,
+    mountpoint TEXT NOT NULL,
+    partitionflags TEXT NOT NULL,
+    partitionid TEXT NOT NULL,
+    partitionsize TEXT NOT NULL,
+    sectorstart TEXT NOT NULL,
+    constraint partitions_nodes_fk FOREIGN KEY(node_id) REFERENCES nodes(id)
+);
+create index if not exists idx_partitions_node on partitions (node_id);
+`
+}
+
+func createPublicKeysTable() string {
+	return `
+CREATE TABLE IF NOT EXISTS publickeys (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    node_id INTEGER,
+    publickey TEXT NOT NULL,
+    description TEXT,
+    constraint publickeys_nodes_fk FOREIGN KEY(node_id) REFERENCES nodes(id)
+);
+create index if not exists idx_publickeys_node on publickeys (node_id);
+`
+}
+
+func createSubnetsTable() string {
+	return `
+CREATE TABLE IF NOT EXISTS subnets (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    name TEXT NOT NULL,
+    dnszone TEXT NOT NULL,
+    subnet TEXT NOT NULL,
+    netmask TEXT NOT NULL,
+    mtu INTEGER NOT NULL DEFAULT 1500,
+    servedns INTEGER NOT NULL DEFAULT 0 CHECK (servedns IN (0, 1)),
+    UNIQUE(name, dnszone)
+);`
+}
+
+func createSoftwareTable() string {
+	return `
+CREATE TABLE IF NOT EXISTS software (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    name TEXT NOT NULL,
+    pathversion TEXT,
+    fullversion TEXT,
+    description TEXT,
+    website TEXT,
+    license TEXT,
+    install_method TEXT NOT NULL CHECK (install_method IN (
+        'source', 
+        'binary',
+        'rpm',
+        'docker',
+        'apptainer',
+        'conda',
+        'mamba',
+        'spack',
+        'tarball',
+        'zipball',
+        'pip',
+        'npm',
+        'custom'
+        )),
+    -- 源码编译相关参数
+    source_url TEXT, -- 源码下载地址
+    source_checksum TEXT, -- 源码校验和
+    source_checksum_type TEXT NOT NULL CHECK (source_checksum_type IN (
+        'md5',
+        'sha1',
+        'sha256',
+        'sha512'
+        )),
+    build_dependencies TEXT,    -- 编译依赖(JSON格式)
+    configure_params TEXT,      -- 配置参数(JSON格式)
+    make_params TEXT,           -- make参数(JSON格式)
+    make_install_params TEXT,   -- make install参数(JSON格式)
+
+    -- 安装路径参数
+    install_path TEXT NOT NULL, -- 安装路径
+    env_vars TEXT,     -- 环境变量(JSON格式)
+
+    -- 状态信息
+    is_installed INTEGER NOT NULL DEFAULT 0 CHECK (is_installed IN (0, 1)), -- 是否安装
+    install_date TEXT, -- 安装日期
+    updated_date TEXT, -- 更新日期
+    install_user TEXT, -- 安装用户
+    notes TEXT,        -- 安装备注
+
+    -- 元数据
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP, -- 创建时间
+    updated_at DATETIME DEFAULT CURRENT_TIMESTAMP, -- 更新时间
+
+    UNIQUE(name)
+);
+create index if not exists idx_software_name on software (name);
+create index if not exists idx_software_install_method on software (install_method);
+create index if not exists idx_software_is_installed on software (is_installed);
+`
+}
+
+func createTrg_nodes_before_delete() string {
+	return `
+CREATE TRIGGER IF NOT EXISTS trg_nodes_before_delete
+    BEFORE DELETE ON nodes
+    FOR EACH ROW
+    BEGIN
+        -- 先删除子表的关联记录
+        DELETE FROM aliases WHERE node_id = OLD.id;
+        DELETE FROM attributes WHERE node_id = OLD.id;
+        DELETE FROM bootactions WHERE node_id = OLD.id;
+        DELETE FROM distributions WHERE node_id = OLD.id;
+        DELETE FROM firewalls WHERE node_id = OLD.id;
+        DELETE FROM networks WHERE node_id = OLD.id;
+        DELETE FROM partitions WHERE node_id = OLD.id;
+        DELETE FROM publickeys WHERE node_id = OLD.id;
+    END;
+`
+}