40 lines
1.0 KiB
Go
40 lines
1.0 KiB
Go
|
// Package database 负责数据库连接初始化与 GORM 实例管理
|
|||
|
|
package database
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"gorm.io/driver/mysql"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
"gorm.io/gorm/logger"
|
|||
|
|
|
|||
|
|
"nl-game-api-gin/internal/config"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// DB 全局数据库实例，Init 成功后可直接使用
|
|||
|
|
var DB *gorm.DB
|
|||
|
|
|
|||
|
|
// Init 连接 MySQL 并配置连接池
|
|||
|
|
func Init() error {
|
|||
|
|
db, err := gorm.Open(mysql.Open(config.C.DSN()), &gorm.Config{
|
|||
|
|
// 只打印慢查询与错误 SQL，避免日志刷屏
|
|||
|
|
Logger: logger.Default.LogMode(logger.Warn),
|
|||
|
|
// dev 自动建表时不生成物理外键约束，与 sql/init.sql 的手写表结构保持一致
|
|||
|
|
DisableForeignKeyConstraintWhenMigrating: true,
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
return fmt.Errorf("连接 MySQL 失败: %w", err)
|
|||
|
|
}
|
|||
|
|
sqlDB, err := db.DB()
|
|||
|
|
if err != nil {
|
|||
|
|
return fmt.Errorf("获取底层连接池失败: %w", err)
|
|||
|
|
}
|
|||
|
|
// 连接池参数：空闲 10、最大 100、连接最长存活 1 小时
|
|||
|
|
sqlDB.SetMaxIdleConns(10)
|
|||
|
|
sqlDB.SetMaxOpenConns(100)
|
|||
|
|
sqlDB.SetConnMaxLifetime(time.Hour)
|
|||
|
|
DB = db
|
|||
|
|
return nil
|
|||
|
|
}
|