You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
137 lines
3.6 KiB
137 lines
3.6 KiB
6 months ago
|
package initialize
|
||
|
|
||
|
import (
|
||
|
"bkb-seller/global"
|
||
|
"bkb-seller/initialize/internal"
|
||
|
"bkb-seller/model"
|
||
|
"os"
|
||
|
|
||
|
"go.uber.org/zap"
|
||
|
"gorm.io/driver/mysql"
|
||
|
"gorm.io/gorm"
|
||
|
"gorm.io/gorm/logger"
|
||
|
)
|
||
|
|
||
|
//@function: Gorm
|
||
|
//@description: 初始化数据库并产生数据库全局变量
|
||
|
//@return: *gorm.DB
|
||
|
|
||
|
func Gorm() *gorm.DB {
|
||
|
switch global.MG_CONFIG.System.DbType {
|
||
|
case "mysql":
|
||
|
return GormMysql()
|
||
|
default:
|
||
|
return GormMysql()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// MysqlTables
|
||
|
//@function: MysqlTables
|
||
|
//@description: 注册数据库表专用
|
||
|
//@param: db *gorm.DB
|
||
|
|
||
|
func MysqlTables(db *gorm.DB) {
|
||
|
err := db.AutoMigrate(
|
||
|
//model.SysUser{},
|
||
|
model.JwtBlacklist{},
|
||
|
model.SysOperationRecord{},
|
||
|
|
||
|
// Code generated by bkb-seller Begin; DO NOT EDIT.
|
||
|
model.SellerStore{},
|
||
|
model.TbAttribute{},
|
||
|
model.TbCategory{},
|
||
|
model.TbGoods{},
|
||
|
model.TbGoodsSpecs{},
|
||
|
model.Mission{},
|
||
|
model.MissionClaim{},
|
||
|
model.MissionClaimVideo{},
|
||
|
model.SysBrochure{},
|
||
|
model.BatchDeliver{},
|
||
|
model.OrderDeliver{},
|
||
|
model.Account{},
|
||
|
model.Wallet{},
|
||
|
model.MissionClaimAddress{},
|
||
|
model.MissionClaimWorks{},
|
||
|
model.MissionClaimOrder{},
|
||
|
model.MissionClaimOrderGoods{},
|
||
|
model.User{},
|
||
|
model.SysDictType{},
|
||
|
model.SysDictData{},
|
||
|
model.MissionStop{},
|
||
|
model.Bill{},
|
||
|
model.BillFund{},
|
||
|
model.Order{},
|
||
|
model.OrderAddress{},
|
||
|
model.OrderDeliver{},
|
||
|
model.OrderGoods{},
|
||
|
model.OrderGoodsSpecs{},
|
||
|
model.OrderPostSale{},
|
||
|
model.SourceFile{},
|
||
|
model.Withdrawal{},
|
||
|
)
|
||
|
if err != nil {
|
||
|
global.MG_LOG.Error("register table failed", zap.Any("err", err))
|
||
|
os.Exit(0)
|
||
|
}
|
||
|
|
||
|
global.MG_LOG.Info("register table success")
|
||
|
}
|
||
|
|
||
|
// @function: GormMysql
|
||
|
// @description: 初始化Mysql数据库
|
||
|
// @return: *gorm.DB
|
||
|
func GormMysql() *gorm.DB {
|
||
|
m := global.MG_CONFIG.Mysql
|
||
|
if m.Dbname == "" {
|
||
|
return nil
|
||
|
}
|
||
|
dsn := m.Username + ":" + m.Password + "@tcp(" + m.Path + ")/" + m.Dbname + "?" + m.Config
|
||
|
mysqlConfig := mysql.Config{
|
||
|
DSN: dsn, // DSN data source name
|
||
|
DefaultStringSize: 191, // string 类型字段的默认长度
|
||
|
DisableDatetimePrecision: true, // 禁用 datetime 精度,MySQL 5.6 之前的数据库不支持
|
||
|
DontSupportRenameIndex: true, // 重命名索引时采用删除并新建的方式,MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
|
||
|
DontSupportRenameColumn: true, // 用 `change` 重命名列,MySQL 8 之前的数据库和 MariaDB 不支持重命名列
|
||
|
SkipInitializeWithVersion: false, // 根据版本自动配置
|
||
|
}
|
||
|
if db, err := gorm.Open(mysql.New(mysqlConfig), &gorm.Config{Logger: logger.Default.LogMode(logger.Info)}); err != nil {
|
||
|
//global.GVA_LOG.Error("MySQL启动异常", zap.Any("err", err))
|
||
|
//os.Exit(0)
|
||
|
//return nil
|
||
|
return nil
|
||
|
} else {
|
||
|
sqlDB, _ := db.DB()
|
||
|
sqlDB.SetMaxIdleConns(m.MaxIdleConns)
|
||
|
sqlDB.SetMaxOpenConns(m.MaxOpenConns)
|
||
|
return db
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//@function: gormConfig
|
||
|
//@description: 根据配置决定是否开启日志
|
||
|
//@param: mod bool
|
||
|
//@return: *gorm.Config
|
||
|
|
||
|
func gormConfig(mod bool) *gorm.Config {
|
||
|
var config = &gorm.Config{DisableForeignKeyConstraintWhenMigrating: true}
|
||
|
switch global.MG_CONFIG.Mysql.LogMode {
|
||
|
case "silent", "Silent":
|
||
|
config.Logger = internal.Default.LogMode(logger.Silent)
|
||
|
case "error", "Error":
|
||
|
config.Logger = internal.Default.LogMode(logger.Error)
|
||
|
case "warn", "Warn":
|
||
|
config.Logger = internal.Default.LogMode(logger.Warn)
|
||
|
case "info", "Info":
|
||
|
config.Logger = internal.Default.LogMode(logger.Info)
|
||
|
case "zap", "Zap":
|
||
|
config.Logger = internal.Default.LogMode(logger.Info)
|
||
|
default:
|
||
|
if mod {
|
||
|
config.Logger = internal.Default.LogMode(logger.Info)
|
||
|
break
|
||
|
}
|
||
|
config.Logger = internal.Default.LogMode(logger.Silent)
|
||
|
}
|
||
|
return config
|
||
|
}
|