43 lines
990 B
Go
43 lines
990 B
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"gorm.io/driver/mysql"
|
||
"gorm.io/gorm"
|
||
"log"
|
||
"sync"
|
||
)
|
||
|
||
var _db *gorm.DB
|
||
var once sync.Once
|
||
|
||
func InitDb() (*gorm.DB, error) {
|
||
config := GetConfig()
|
||
mysqlConf := config.MySql
|
||
dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local&timeout=%s",
|
||
mysqlConf.User, mysqlConf.Password, mysqlConf.Host, mysqlConf.Port, mysqlConf.Db, mysqlConf.Timeout)
|
||
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
sqlDb, err := db.DB()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
sqlDb.SetMaxOpenConns(100) // 设置数据库连接池最大连接数
|
||
sqlDb.SetMaxIdleConns(20) // 连接池最大允许的空闲连接数，如果没有sql任务需要执行的连接数大于20，超过的连接会被连接池关闭。
|
||
return db, nil
|
||
}
|
||
|
||
func GetDb() *gorm.DB {
|
||
once.Do(func() {
|
||
db, err := InitDb()
|
||
if err != nil {
|
||
log.Fatal("连接数据库失败, error: ", err)
|
||
}
|
||
_db = db
|
||
})
|
||
return _db
|
||
}
|