mirror of
https://github.com/skyle1995/NetworkAuth.git
synced 2026-05-25 02:24:05 +08:00
126 lines
3.5 KiB
Go
126 lines
3.5 KiB
Go
package database
|
||
|
||
import (
|
||
"fmt"
|
||
"sync"
|
||
|
||
"networkDev/utils"
|
||
|
||
"github.com/glebarez/sqlite"
|
||
"github.com/sirupsen/logrus"
|
||
"github.com/spf13/viper"
|
||
"gorm.io/driver/mysql"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
var (
|
||
// dbInstance 全局 *gorm.DB 实例，使用单例确保全局复用
|
||
dbInstance *gorm.DB
|
||
// once 确保初始化只执行一次
|
||
once sync.Once
|
||
)
|
||
|
||
// Init 初始化数据库连接（根据配置自动选择驱动）
|
||
// - 默认使用 SQLite（github.com/glebarez/sqlite）
|
||
// - 生产环境支持 MySQL（gorm.io/driver/mysql）
|
||
func Init() (*gorm.DB, error) {
|
||
var initErr error
|
||
once.Do(func() {
|
||
dbType := viper.GetString("database.type")
|
||
switch dbType {
|
||
case "mysql":
|
||
initErr = initMySQL()
|
||
default:
|
||
initErr = initSQLite()
|
||
}
|
||
|
||
// 如果数据库初始化成功，配置连接池和启动健康检查
|
||
if initErr == nil && dbInstance != nil {
|
||
// 加载数据库配置
|
||
var configPrefix string
|
||
if dbType == "mysql" {
|
||
configPrefix = "database.mysql"
|
||
} else {
|
||
configPrefix = "database.sqlite"
|
||
}
|
||
|
||
dbConfig := utils.LoadDatabaseConfig(configPrefix)
|
||
|
||
// 验证配置
|
||
if err := utils.ValidateDatabaseConfig(dbConfig); err != nil {
|
||
logrus.WithError(err).Warn("数据库配置验证失败，使用默认配置")
|
||
dbConfig = utils.GetDefaultDatabaseConfig()
|
||
}
|
||
|
||
// 配置连接池
|
||
if err := utils.ConfigureConnectionPool(dbInstance, dbConfig); err != nil {
|
||
logrus.WithError(err).Error("配置数据库连接池失败")
|
||
}
|
||
|
||
// 启动健康检查
|
||
utils.StartHealthCheck(dbInstance, dbConfig)
|
||
}
|
||
})
|
||
return dbInstance, initErr
|
||
}
|
||
|
||
// GetDB 获取全局 *gorm.DB 实例
|
||
// 如果未初始化，会尝试初始化一次
|
||
func GetDB() (*gorm.DB, error) {
|
||
if dbInstance != nil {
|
||
return dbInstance, nil
|
||
}
|
||
return Init()
|
||
}
|
||
|
||
// initSQLite 初始化 SQLite 数据库
|
||
// 使用 viper 中的 database.sqlite.path 作为数据库文件路径
|
||
func initSQLite() error {
|
||
path := viper.GetString("database.sqlite.path")
|
||
if path == "" {
|
||
path = "./database.db"
|
||
}
|
||
dsn := fmt.Sprintf("file:%s?cache=shared&_busy_timeout=5000&_fk=1", path)
|
||
db, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{})
|
||
if err != nil {
|
||
logrus.WithError(err).Error("SQLite 初始化失败")
|
||
return err
|
||
}
|
||
|
||
// SQLite 连接池配置（SQLite 对连接池支持有限，但仍可设置基本参数）
|
||
if sqlDB, err := db.DB(); err == nil {
|
||
// SQLite 通常使用单连接，但可以设置一些基本参数
|
||
sqlDB.SetMaxOpenConns(1) // SQLite 建议使用单连接
|
||
sqlDB.SetMaxIdleConns(1)
|
||
}
|
||
|
||
dbInstance = db
|
||
logrus.WithField("path", path).Info("SQLite 连接已建立")
|
||
return nil
|
||
}
|
||
|
||
// initMySQL 初始化 MySQL 数据库
|
||
// 从 viper 读取 database.mysql.* 配置构建 DSN
|
||
func initMySQL() error {
|
||
host := viper.GetString("database.mysql.host")
|
||
port := viper.GetInt("database.mysql.port")
|
||
user := viper.GetString("database.mysql.username")
|
||
pass := viper.GetString("database.mysql.password")
|
||
dbname := viper.GetString("database.mysql.database")
|
||
charset := viper.GetString("database.mysql.charset")
|
||
if charset == "" {
|
||
charset = "utf8mb4"
|
||
}
|
||
|
||
dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=True&loc=Local", user, pass, host, port, dbname, charset)
|
||
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
|
||
if err != nil {
|
||
logrus.WithError(err).Error("MySQL 初始化失败")
|
||
return err
|
||
}
|
||
|
||
dbInstance = db
|
||
logrus.WithField("host", host).WithField("database", dbname).Info("MySQL 连接已建立")
|
||
return nil
|
||
}
|