80 lines
1.7 KiB
Go
80 lines
1.7 KiB
Go
package model
|
|
|
|
import (
|
|
"fmt"
|
|
"gitee.com/codinl/erp_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:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
|
config.ConfigValue.MysqlUser, config.ConfigValue.MysqlPassword, config.ConfigValue.MysqlHost, config.ConfigValue.MysqlPort,
|
|
config.ConfigValue.MysqlDBName)
|
|
DB, err = gorm.Open(config.ConfigValue.MysqlDriver, dialect)
|
|
if err != nil {
|
|
logger.Info("Connect Database Error:", err)
|
|
panic(err)
|
|
}
|
|
DB.DB().SetMaxOpenConns(1000)
|
|
DB.DB().SetMaxIdleConns(500)
|
|
DB.DB().SetConnMaxLifetime(time.Second * 10)
|
|
DB.LogMode(true)
|
|
|
|
DB.SingularTable(true)
|
|
if err := DB.DB().Ping(); err != nil {
|
|
logger.Info("Ping Database Fail:", err)
|
|
panic(err)
|
|
}
|
|
fmt.Println("DB init success")
|
|
logger.Info("DB init success")
|
|
}
|
|
|
|
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{}) {
|
|
}
|
|
|
|
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:", err)
|
|
}
|
|
}
|