路由单独中间件

lib/routes/routes.go

@@ -18,6 +18,9 @@ import (
 	"ems.agt/features/pm"
 	"ems.agt/features/security"
 	"ems.agt/features/state"
+	sysmenu "ems.agt/features/sys_menu"
+	sysrole "ems.agt/features/sys_role"
+	sysuser "ems.agt/features/sys_user"
 	"ems.agt/features/trace"
 	"ems.agt/lib/midware"
 	"ems.agt/lib/services"
@@ -249,6 +252,24 @@ func init() {
 	Register("POST", lm.ExtBackupDataUri, lm.ExtDatabaseBackupData, nil)
 	Register("POST", lm.CustomExtBackupDataUri, lm.ExtDatabaseBackupData, nil)
 
+	// 登录用户路由信息
+	Register("GET", security.UriRouters, security.Routers, midware.Authorize(nil))
+	Register("GET", security.CustomUriRouters, security.Routers, midware.Authorize(nil))
+
+	// 菜单接口添加到路由
+	for _, v := range sysmenu.Routers() {
+		Register(v.Method, v.Pattern, v.Handler, v.Middleware)
+	}
+
+	// 角色接口添加到路由
+	for _, v := range sysrole.Routers() {
+		Register(v.Method, v.Pattern, v.Handler, v.Middleware)
+	}
+
+	// 用户接口添加到路由
+	for _, v := range sysuser.Routers() {
+		Register(v.Method, v.Pattern, v.Handler, v.Middleware)
+	}
 }
 
 // To resolv rest POST/PUT/DELETE/PATCH cross domain
@@ -268,22 +289,12 @@ func NewRouter() *mux.Router {
 	r.Use(midware.ArrowIPAddr)
 
 	for _, router := range routers {
-		// r.Methods(router.Method).
-		// 	Path(router.Pattern).
-		// 	Handler(router.Handler)
-		// if router.Middleware != nil {
-		// 	r.Use(router.Middleware)
-		// }
-		// if router.Middleware != nil {
-		// 	r.HandleFunc(router.Pattern, router.Handler).Methods(router.Method).Subrouter().Use(router.Middleware)
-		// } else {
-		// 	r.HandleFunc(router.Pattern, router.Handler).Methods(router.Method)
-		// }
 		rt := r.Methods(router.Method).Subrouter()
 		rt.HandleFunc(router.Pattern, router.Handler)
 		if router.Middleware != nil {
 			rt.Use(router.Middleware)
 		}
+
 	}
 
 	return r