refactor🎨: engine 初始化调整

This commit is contained in:
zhangwenjian
2021-06-11 09:25:10 +08:00
parent 898b1ea1e6
commit 860226ab41
4 changed files with 45 additions and 29 deletions
+3 -8
View File
@@ -1,7 +1,7 @@
package router
import (
"github.com/go-admin-team/go-admin-core/sdk/pkg"
_ "github.com/go-admin-team/go-admin-core/sdk/pkg"
"os"
"github.com/gin-gonic/gin"
@@ -15,8 +15,8 @@ func InitRouter() {
var r *gin.Engine
h := sdk.Runtime.GetEngine()
if h == nil {
h = gin.New()
sdk.Runtime.SetEngine(h)
log.Fatal("not found engine...")
os.Exit(-1)
}
switch h.(type) {
case *gin.Engine:
@@ -25,10 +25,6 @@ func InitRouter() {
log.Fatal("not support other engine")
os.Exit(-1)
}
r.Use(common.Sentinel()).
Use(common.RequestId(pkg.TrafficKey))
common.InitMiddleware(r)
// the jwt middleware
authMiddleware, err := common.AuthInit()
if err != nil {
@@ -36,6 +32,5 @@ func InitRouter() {
}
// 注册业务路由
// TODO: 这里可存放业务路由,里边并无实际路由只有演示代码
initRouter(r, authMiddleware)
}