device-manage-server/device-manage/app/admin/router/initrouter.go

package router

import (
	"device-manage/app/admin/middleware"
	"device-manage/app/admin/middleware/handler"
	"device-manage/common/global"
	"device-manage/common/log"
	_ "device-manage/pkg/jwtauth"
	"device-manage/tools"
	"device-manage/tools/config"
	"github.com/gin-gonic/gin"
	"os"
)

func InitRouter() {
	var r *gin.Engine
	h := global.Cfg.GetEngine()
	if h == nil {
		h = gin.New()
		global.Cfg.SetEngine(h)
	}
	switch h.(type) {
	case *gin.Engine:
		r = h.(*gin.Engine)
	default:
		log.Fatal("not support other engine")
		os.Exit(-1)
	}
	if config.SslConfig.Enable {
		r.Use(handler.TlsHandler())
	}
	r.Use(middleware.WithContextDb(middleware.GetGormFromConfig(global.Cfg)))
	middleware.InitMiddleware(r)
	// the jwt middleware
	var err error
	authMiddleware, err := middleware.AuthInit()
	tools.HasError(err, "JWT Init Error", 500)

	// 注册系统路由
	InitSysRouter(r, authMiddleware)

	// 注册业务路由
	// TODO: 这里可存放业务路由，里边并无实际路由只有演示代码
	InitBusinessRouter(r, authMiddleware)

	//return r
}