lan
a111872b32
- Upgrade casbin from v2 to v3 across go.mod and pkg/auth/casbin.go - Add slide captcha verification to registration flow (CheckVerified, ConsumeVerified) - Add DB wrapper with connection pool statistics and health checks - Add Redis connection pool optimizations with stats and health monitoring - Add new config options: ConnMaxLifetime, HealthCheckInterval, EnableRetryOnError - Optimize slow query threshold from 200ms to 100ms - Add ping with retry mechanism for database and Redis connections
118 lines
2.6 KiB
Go
118 lines
2.6 KiB
Go
package database
|
||
|
||
import (
|
||
"carrotskin/internal/model"
|
||
"carrotskin/pkg/config"
|
||
"fmt"
|
||
"sync"
|
||
|
||
"go.uber.org/zap"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
var (
|
||
// dbInstance 全局数据库实例（使用 *DB 封装）
|
||
dbInstance *DB
|
||
// once 确保只初始化一次
|
||
once sync.Once
|
||
// initError 初始化错误
|
||
initError error
|
||
)
|
||
|
||
// Init 初始化数据库连接（线程安全，只会执行一次）
|
||
func Init(cfg config.DatabaseConfig, logger *zap.Logger) error {
|
||
once.Do(func() {
|
||
dbInstance, initError = New(cfg)
|
||
if initError != nil {
|
||
logger.Error("数据库初始化失败", zap.Error(initError))
|
||
return
|
||
}
|
||
logger.Info("数据库连接成功")
|
||
})
|
||
return initError
|
||
}
|
||
|
||
// GetDB 获取数据库实例（线程安全）
|
||
// 返回 *gorm.DB 以保持向后兼容
|
||
func GetDB() (*gorm.DB, error) {
|
||
if dbInstance == nil {
|
||
return nil, fmt.Errorf("数据库未初始化，请先调用 database.Init()")
|
||
}
|
||
return dbInstance.DB, nil
|
||
}
|
||
|
||
// GetDBWrapper 获取数据库封装实例（包含连接池统计功能）
|
||
func GetDBWrapper() (*DB, error) {
|
||
if dbInstance == nil {
|
||
return nil, fmt.Errorf("数据库未初始化，请先调用 database.Init()")
|
||
}
|
||
return dbInstance, nil
|
||
}
|
||
|
||
// MustGetDB 获取数据库实例，如果未初始化则panic
|
||
// 返回 *gorm.DB 以保持向后兼容
|
||
func MustGetDB() *gorm.DB {
|
||
db, err := GetDB()
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
return db
|
||
}
|
||
|
||
// AutoMigrate 自动迁移数据库表结构
|
||
func AutoMigrate(logger *zap.Logger) error {
|
||
db, err := GetDB()
|
||
if err != nil {
|
||
return fmt.Errorf("获取数据库实例失败: %w", err)
|
||
}
|
||
|
||
logger.Info("开始执行数据库迁移...")
|
||
|
||
// 迁移所有表 - 注意顺序：先创建被引用的表，再创建引用表
|
||
// 使用分批迁移，避免某些表的问题影响其他表
|
||
tables := []interface{}{
|
||
// 用户相关表（先创建，因为其他表可能引用它）
|
||
&model.User{},
|
||
&model.UserPointLog{},
|
||
&model.UserLoginLog{},
|
||
|
||
// 档案相关表
|
||
&model.Profile{},
|
||
|
||
// 材质相关表
|
||
&model.Texture{},
|
||
&model.UserTextureFavorite{},
|
||
&model.TextureDownloadLog{},
|
||
|
||
// 认证相关表
|
||
&model.Client{}, // Client表用于管理Token版本
|
||
|
||
// Yggdrasil相关表（在User之后创建，因为它引用User）
|
||
&model.Yggdrasil{},
|
||
|
||
// 审计日志表
|
||
&model.AuditLog{},
|
||
|
||
// Casbin权限规则表
|
||
&model.CasbinRule{},
|
||
}
|
||
|
||
// 批量迁移表
|
||
if err := db.AutoMigrate(tables...); err != nil {
|
||
logger.Error("数据库迁移失败", zap.Error(err))
|
||
return fmt.Errorf("数据库迁移失败: %w", err)
|
||
}
|
||
|
||
logger.Info("数据库迁移完成")
|
||
return nil
|
||
}
|
||
|
||
// Close 关闭数据库连接
|
||
func Close() error {
|
||
if dbInstance == nil {
|
||
return nil
|
||
}
|
||
|
||
return dbInstance.Close()
|
||
}
|