feat: issues #809

2021-12-07 18:46:44 +08:00
parent 8b94b8b043
commit fc003dbc16
8 changed files with 132 additions and 2 deletions
--- a/server/initialize/db_list.go
+++ b/server/initialize/db_list.go
@@ -0,0 +1,24 @@
+package initialize
+
+import (
+	"github.com/flipped-aurora/gin-vue-admin/server/global"
+	"gorm.io/gorm"
+)
+
+func DBList() {
+	dbMap := make(map[string]*gorm.DB)
+	for _, info := range global.GVA_CONFIG.DBList {
+		if info.Disable {
+			continue
+		}
+		switch info.Type {
+		case "mysql":
+			dbMap[info.Dbname] = GormMysqlByConfig(info)
+		case "pgsql":
+			dbMap[info.Dbname] = GormPgSqlByConfig(info)
+		default:
+			continue
+		}
+	}
+	global.GVA_DBList = dbMap
+}