enterprise/common/dao/dao.go

package dao

import "gorm.io/gorm"

var (
	db      *gorm.DB
	unifyDB *gorm.DB
	corpDB  *gorm.DB
)

func SetDB(d *gorm.DB) {
	db = d
}

func SetUnifyDB(d *gorm.DB) {
	unifyDB = d
}
func SetCorpDB(d *gorm.DB) {
	corpDB = d
}

func GetDB() *gorm.DB {
	return db
}