kkc-go/router/Interceptor.go

package router

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"strings"
)

func LoginInterceptor() gin.HandlerFunc {
	return func(c *gin.Context) {
		id := GetUserIdByToken(c)
		fmt.Println("拦截器", id)
		if id == 0 {
			fmt.Println("拦截器:用户未登录")
			//这里终止后续请求访问
			c.Abort()
			return
		}
	}
}

func Interceptor(InterceptorName string) gin.HandlerFunc {
	switch InterceptorName {
	case "login":
		return LoginInterceptor()
	case "log":
		return LogInterceptor()
	default:
		return func(context *gin.Context) {}
	}
}
func AddUseInterceptor(routes gin.IRoutes, AuthorityVerification string) {
	split := strings.Split(AuthorityVerification, ",")
	for i := range split {
		routes.Use(Interceptor(split[i]))
	}
}

func LogInterceptor() gin.HandlerFunc {
	return func(c *gin.Context) {
		fmt.Println("日志系统")
		c.Next()
		fmt.Println("日志系统666")
	}
}