主路由和子路由同时添加权限控制

Author: yumaojun03

http/router/entry.go

@@ -25,11 +25,21 @@ func (e *Entry) EnableAuth() {
 	e.AuthEnable = true
 }
 
+// DisableAuth 不启用身份验证
+func (e *Entry) DisableAuth() {
+	e.AuthEnable = false
+}
+
 // EnablePermission 启用授权验证
 func (e *Entry) EnablePermission() {
 	e.PermissionEnable = true
 }
 
+// DisablePermission 禁用授权验证
+func (e *Entry) DisablePermission() {
+	e.PermissionEnable = false
+}
+
 // NewEntrySet 实例
 func NewEntrySet() *EntrySet {
 	return &EntrySet{}

http/router/httprouter/httprouter.go

@@ -17,12 +17,14 @@ type httpRouter struct {
 	r *httprouter.Router
 	l logger.Logger
 
-	middlewareChain []router.Middleware
-	entrySet        *entrySet
-	auther          router.Auther
-	mergedHandler   http.Handler
-	labels          []*router.Label
-	notFound        http.Handler
+	middlewareChain  []router.Middleware
+	entrySet         *entrySet
+	auther           router.Auther
+	mergedHandler    http.Handler
+	labels           []*router.Label
+	notFound         http.Handler
+	authEnable       bool
+	permissionEnable bool
 }
 
 // New 基于社区的httprouter进行封装
@@ -52,10 +54,12 @@ func (r *httpRouter) Use(m router.Middleware) {
 func (r *httpRouter) Handle(method, path string, h http.HandlerFunc) router.EntryDecorator {
 	e := &entry{
 		Entry: &router.Entry{
-			Method:       method,
-			Path:         path,
-			FunctionName: router.GetHandlerFuncName(h),
-			Labels:       map[string]string{},
+			Method:           method,
+			Path:             path,
+			FunctionName:     router.GetHandlerFuncName(h),
+			Labels:           map[string]string{},
+			AuthEnable:       r.authEnable,
+			PermissionEnable: r.permissionEnable,
 		},
 		h: h,
 	}
@@ -64,6 +68,14 @@ func (r *httpRouter) Handle(method, path string, h http.HandlerFunc) router.Entr
 	return e.Entry
 }
 
+func (r *httpRouter) Auth(isEnable bool) {
+	r.authEnable = isEnable
+}
+
+func (r *httpRouter) Permission(isEnable bool) {
+	r.permissionEnable = isEnable
+}
+
 func (r *httpRouter) SetAuther(at router.Auther) {
 	r.auther = at
 	return

http/router/httprouter/subrouter.go

@@ -21,6 +21,8 @@ type subRouter struct {
 	root             *httpRouter
 	labels           []*router.Label
 	middlewareChain  []router.Middleware
+	authEnable       bool
+	permissionEnable bool
 }
 
 func (r *subRouter) Use(m router.Middleware) {
@@ -43,25 +45,33 @@ func (r *subRouter) With(m ...router.Middleware) router.SubRouter {
 func (r *subRouter) Handle(method, path string, h http.HandlerFunc) router.EntryDecorator {
 	e := &entry{
 		Entry: &router.Entry{
-			Resource:     r.resourceName,
-			Method:       method,
-			Path:         path,
-			FunctionName: router.GetHandlerFuncName(h),
-			Labels:       map[string]string{},
-			AuthEnable:   true,
+			Resource:         r.resourceName,
+			Method:           method,
+			Path:             path,
+			FunctionName:     router.GetHandlerFuncName(h),
+			Labels:           map[string]string{},
+			AuthEnable:       r.authEnable,
+			PermissionEnable: r.permissionEnable,
 		},
 		h: h,
 	}
 
 	r.add(e)
-
 	return e.Entry
 }
 
 func (r *subRouter) SetLabel(labels ...*router.Label) {
 	r.labels = append(r.labels, labels...)
 }
 
+func (r *subRouter) Auth(isEnable bool) {
+	r.authEnable = isEnable
+}
+
+func (r *subRouter) Permission(isEnable bool) {
+	r.permissionEnable = isEnable
+}
+
 func (r *subRouter) ResourceRouter(resourceName string, labels ...*router.Label) router.ResourceRouter {
 	return &subRouter{
 		resourceName:     resourceName,

http/router/router.go

@@ -11,6 +11,12 @@ type Router interface {
 	// 添加中间件
 	Use(m Middleware)
 
+	// 是否启用用户身份验证
+	Auth(isEnable bool)
+
+	// 是否启用用户权限验证
+	Permission(isEnable bool)
+
 	// 添加受认证保护的路由
 	Handle(method, path string, h http.HandlerFunc) EntryDecorator
 
@@ -44,6 +50,10 @@ type ResourceRouter interface {
 
 // SubRouter 子路由或者分组路由
 type SubRouter interface {
+	// 是否启用用户身份验证
+	Auth(isEnable bool)
+	// 是否启用用户权限验证
+	Permission(isEnable bool)
 	// 添加中间件
 	Use(m Middleware)
 	// SetLabel 设置路由标签, 作用于Entry上
@@ -61,5 +71,7 @@ type EntryDecorator interface {
 	// SetLabel 设置子路由标签, 作用于Entry上
 	AddLabel(...*Label) EntryDecorator
 	EnableAuth()
+	DisableAuth()
 	EnablePermission()
+	DisablePermission()
 }