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.
58 lines
958 B
58 lines
958 B
package db
|
|
|
|
import (
|
|
"gitea.baoapi.com/root/stu_uuos/logger"
|
|
"gitea.baoapi.com/root/stu_uuos/util"
|
|
"github.com/jinzhu/gorm"
|
|
log "github.com/micro/go-micro/v2/logger"
|
|
)
|
|
|
|
var db *gorm.DB
|
|
var db2 *gorm.DB
|
|
var db3 *gorm.DB
|
|
|
|
func Init() {
|
|
openPostgres()
|
|
db = postgresDB
|
|
|
|
openPostgres2()
|
|
db2 = postgresDB2
|
|
|
|
openPostgres3()
|
|
db3 = postgresDB3
|
|
}
|
|
|
|
func DB() *gorm.DB {
|
|
return db
|
|
}
|
|
|
|
func DB2() *gorm.DB {
|
|
return db2
|
|
}
|
|
|
|
func DB3() *gorm.DB {
|
|
return db3
|
|
}
|
|
|
|
func Close() error {
|
|
if err := postgresDB.Close(); err != nil {
|
|
log.Errorf("%v close error", "db")
|
|
}
|
|
|
|
if err := postgresDB2.Close(); err != nil {
|
|
log.Errorf("%v close error", "db2")
|
|
}
|
|
if err := postgresDB3.Close(); err != nil {
|
|
log.Errorf("%v close error", "db3")
|
|
}
|
|
util.PrintInfo("close database connections")
|
|
return nil
|
|
}
|
|
|
|
func dbCfg(db *gorm.DB) {
|
|
db.SingularTable(true)
|
|
db.LogMode(true)
|
|
db.SetLogger(logger.GetLogger())
|
|
db.DB().SetMaxOpenConns(50)
|
|
db.DB().SetMaxIdleConns(20)
|
|
}
|