feat: 结构整理

This commit is contained in:
2021-12-14 14:31:18 +08:00
parent 28fdd5a348
commit e321686d4b
5 changed files with 42 additions and 29 deletions
+6 -3
View File
@@ -1,13 +1,16 @@
package bgm
import (
"bgm/consts"
"fmt"
"gorm.io/driver/mysql"
"gorm.io/gorm"
)
func GormMap(dsnPre string) (map[string]*gorm.DB, error) {
dsn := fmt.Sprintf("%v/test?charset=utf8mb4&parseTime=True&loc=Local", dsnPre)
type MysqlMap map[string]*gorm.DB
func GormMap(dsnPre string) (MysqlMap, error) {
dsn := fmt.Sprintf("%v/%v?charset=utf8mb4&parseTime=True&loc=Local", dsnPre, consts.TestDB)
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
return map[string]*gorm.DB{"local": db}, err
return map[string]*gorm.DB{consts.TestDB: db}, err
}