90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
|
package model
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"mh-server/config"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/codinl/go-logger"
|
||
|
|
"github.com/jinzhu/gorm"
|
||
|
|
_ "github.com/jinzhu/gorm/dialects/mysql"
|
||
|
|
)
|
||
|
|
|
||
|
|
var DB *gorm.DB
|
||
|
|
|
||
|
|
func InitDB() {
|
||
|
|
var err error
|
||
|
|
dialect := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
||
|
|
config.AppConfig.DBUser, config.AppConfig.DBPassword, config.AppConfig.DBHost, config.AppConfig.DBPort,
|
||
|
|
config.AppConfig.DBName)
|
||
|
|
DB, err = gorm.Open(config.AppConfig.DBDriver, dialect)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Println("Connect Database Error, error = ", err)
|
||
|
|
logger.Info("Connect Database Error, error = ", err)
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
DB.DB().SetMaxOpenConns(1000)
|
||
|
|
DB.DB().SetMaxIdleConns(500)
|
||
|
|
DB.DB().SetConnMaxLifetime(time.Second * 10)
|
||
|
|
DB.LogMode(true)
|
||
|
|
DB.SetLogger(DBLoggerWriter{})
|
||
|
|
DB.SingularTable(true)
|
||
|
|
if err := DB.DB().Ping(); err != nil {
|
||
|
|
fmt.Println("Ping Database Fail, error = ", err)
|
||
|
|
logger.Info("Ping Database Fail, error = ", err)
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
fmt.Println("DB init success")
|
||
|
|
logger.Info("DB init success")
|
||
|
|
|
||
|
|
//if config.AppConfig.Env == "dev" {
|
||
|
|
// migrateTable(DB)
|
||
|
|
//}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func CloseDB() {
|
||
|
|
if DB != nil {
|
||
|
|
if err := DB.Close(); err != nil {
|
||
|
|
fmt.Println("DB Close err", err)
|
||
|
|
logger.Error("DB Close err", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type DBLoggerWriter struct {
|
||
|
|
gorm.LogWriter
|
||
|
|
}
|
||
|
|
|
||
|
|
func (DBLoggerWriter) Print(values ...interface{}) {
|
||
|
|
//logger.Info(values...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TransactionBegin() *gorm.DB {
|
||
|
|
return DB.Begin()
|
||
|
|
}
|
||
|
|
|
||
|
|
// Transaction 事务自动处理，如果有错误将回滚，否则将提交事务
|
||
|
|
func Transaction(tx *gorm.DB, err error) error {
|
||
|
|
if err == nil {
|
||
|
|
ret := tx.Commit()
|
||
|
|
if ret.Error != nil {
|
||
|
|
return ret.Error
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
ret := tx.Rollback()
|
||
|
|
if ret.Error != nil {
|
||
|
|
return ret.Error
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func initCreateTable() {
|
||
|
|
if err := DB.Set("gorm:table_options", "ENGINE=InnoDB DEFAULT CHARSET=utf8").AutoMigrate().Error; err != nil {
|
||
|
|
logger.Info("AutoMigrate createTable Error, error = ", err)
|
||
|
|
fmt.Println("AutoMigrate createTable err:", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|