stu_uuos/lib/db/db.go

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)
}