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.
43 lines
825 B
43 lines
825 B
1 year ago
|
package mysql
|
||
|
|
|
||
|
|
import (
|
||
|
|
"gorm.io/driver/mysql"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
"gorm.io/gorm/logger"
|
||
|
|
)
|
||
|
|
|
||
|
|
// MysqlClient sql连接对象
|
||
|
|
type MysqlClient struct {
|
||
|
|
client *gorm.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func InitMysqlClient(dsn string, debug bool) (*MysqlClient, error) {
|
||
|
|
dsn += "?charset=utf8mb4&parseTime=True&loc=Local"
|
||
|
|
// "user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
|
||
|
|
|
||
|
|
var ormLogger logger.Interface
|
||
|
|
if debug {
|
||
|
|
ormLogger = logger.Default.LogMode(logger.Info)
|
||
|
|
} else {
|
||
|
|
ormLogger = logger.Default
|
||
|
|
}
|
||
|
|
|
||
|
|
user_db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
|
||
|
|
Logger: ormLogger,
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return &MysqlClient{client: user_db}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *MysqlClient) C() *gorm.DB {
|
||
|
|
return m.client
|
||
|
|
}
|
||
|
|
|
||
|
|
// Begin 开启事务
|
||
|
|
func (u *MysqlClient) Begin() *gorm.DB {
|
||
|
|
return u.client.Begin()
|
||
|
|
}
|