This commit is contained in:
kzkzzzz
2024-06-01 18:28:09 +08:00
parent 0b36406c6f
commit 7fe58f3725
4 changed files with 6 additions and 6 deletions

View File

@@ -138,12 +138,12 @@ func NewGormLogger(writer gormLogger.Writer, gormLoggerConfig gormLogger.Config)
return gormLogger.New(writer, gormLoggerConfig)
}
func CloseInstance(key string) {
func CloseDB(key string) {
db, _ := DB(key).DB()
db.Close()
}
func CloseAllInstance() {
func CloseAllDB() {
for _, v := range instanceMap {
db, _ := v.DB()
db.Close()

View File

@@ -20,7 +20,7 @@ func TestMysql(t *testing.T) {
return
}
defer CloseAllInstance()
defer CloseAllDB()
var res = make(map[string]interface{})
err = DB().Table("image").Limit(1).Take(&res).Error