177 lines
4.1 KiB
Go
177 lines
4.1 KiB
Go
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
|
|
}
|