数据库连接初始化优化

权限部分修改

common/middleware/db.go

@@ -2,16 +2,11 @@ package middleware
 
 import (
 	"github.com/gin-gonic/gin"
-	"gorm.io/gorm"
+
+	"go-admin/common/global"
 )
 
-func WithContextDb(dbMap map[string]*gorm.DB) gin.HandlerFunc {
-	return func(c *gin.Context) {
-		if db, ok := dbMap["*"]; ok {
-			c.Set("db", db)
-		} else {
-			c.Set("db", dbMap[c.Request.Host])
-		}
-		c.Next()
-	}
+func WithContextDb(c *gin.Context) {
+	c.Set("db", global.Cfg.GetDbByKey(c.Request.Host))
+	c.Next()
 }