Blizzard
62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
package initialize
|
|
|
|
import (
|
|
"os"
|
|
"sundynix-go/global"
|
|
"sundynix-go/model/radio"
|
|
"sundynix-go/model/system"
|
|
|
|
"go.uber.org/zap"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// Gorm 根据全局配置中的数据库类型返回对应的 *gorm.DB 实例。
|
|
// 该函数通过检查 global.Config.System.DbType 的值来决定使用哪种数据库连接。
|
|
//
|
|
// 返回值:
|
|
// - *gorm.DB: 返回对应数据库类型的 *gorm.DB 实例
|
|
func Gorm() *gorm.DB {
|
|
switch global.Config.System.DbType {
|
|
case "mysql":
|
|
return GormMysql() // 返回 MySQL 数据库的 *gorm.DB 实例
|
|
case "pgsql":
|
|
return GromPgsql() // 返回 PostgreSQL 数据库的 *gorm.DB 实例
|
|
case "sqlite":
|
|
return GormSqlite() // 返回 SQLite 数据库的 *gorm.DB 实例
|
|
default:
|
|
return GormMysql() // 默认返回 MySQL 数据库 of *gorm.DB 实例
|
|
}
|
|
}
|
|
|
|
// MigrateTable 创建数据库表结构。
|
|
func MigrateTable() {
|
|
db := global.DB
|
|
err := db.AutoMigrate(
|
|
system.User{},
|
|
system.Client{},
|
|
system.Role{},
|
|
system.Menu{},
|
|
system.SysOperationRecord{},
|
|
system.Oss{},
|
|
|
|
radio.Vip{},
|
|
radio.RadioCategory{},
|
|
radio.RadioChannel{},
|
|
radio.RadioProgram{},
|
|
radio.RadioSubscription{},
|
|
radio.Order{},
|
|
radio.PayNotify{},
|
|
|
|
radio.RadioComment{},
|
|
radio.RadioHistory{},
|
|
radio.RadioFavorite{},
|
|
radio.RadioLike{},
|
|
radio.RadioListenLog{},
|
|
)
|
|
if err != nil {
|
|
global.Logger.Error("Migrate table failed,err:", zap.Error(err))
|
|
os.Exit(0)
|
|
}
|
|
global.Logger.Info("Migrate table success")
|
|
}
|