springboot使用AOP基于注解实现token拦截

最新推荐文章于 2023-11-23 16:16:30 发布

zyy_demon

最新推荐文章于 2023-11-23 16:16:30 发布

阅读量1.8k

点赞数 5

文章标签： java spring aop

本文链接：https://blog.csdn.net/sy_white/article/details/122307627

版权

本文介绍了如何通过自定义@NeedToken注解和Spring AOP实现接口请求的Token验证。当接口带有@NeedToken注解时，如果请求中未携带Token，系统将抛出异常；反之，则允许请求。主要步骤包括定义注解、创建拦截器、配置拦截规则以及测试用例。AOP在权限验证中的应用简化了代码并提高了可维护性。

摘要由CSDN通过智能技术生成

往往在一些后台管理需求中，后台请求接口一般都需要token参数才能访问，下面介绍以下自定义@NeedToken注解使用切面aop技术决定请求接口是否需要token参数

目的

接口若有@NeedToken注解，若请求不携带token，则会抛出无token异常

接口若无该注解，不携带token，可正常请求该接口

思路

1.首先定义一个@NeedToken注解，作用于方法上

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface NeedToken {
}

2.定义切面拦截

继承HandlerInterceptorAdapter类，重写preHandle方法，尝试分别从请求头，请求参数，cookie中获取token，若最后token为null，且需要token，则抛出无token，无权访问异常

@Component
public class LoginTokenInterceptor extends HandlerInterceptorAdapter {

    private static final String TOKEN_KEY = "token";

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

        // 判断访问的方法上是否有@needToken注解
        HandlerMethod handlerMethod = (HandlerMethod) handler;
        boolean needToken = handlerMethod.hasMethodAnnotation(NeedToken.class);

        // 尝试从请求头从获取token
        String token = request.getHeader(TOKEN_KEY);
        if (StringUtils.isEmpty(token)) {
            // 尝试从请求参数中获取token
            token = request.getParameter(token);
        }

        if (StringUtils.isEmpty(token)) {
            // 尝试从cookie获取token
            if (request.getCookies() != null && request.getCookies().length != 0) {
                for (Cookie cookie : request.getCookies()) {
                    if (TOKEN_KEY.equals(cookie.getName())) {
                        token = cookie.getValue();
                    }
                }

            }
        }

        // 若token为空,且需要token
        if (StringUtils.isEmpty(token)) {
            if (needToken) {
                throw new RuntimeException("无token,无权访问");
            }
        }

        return super.preHandle(request, response, handler);
    }
}

3.添加拦截器，将该拦截点放入拦截器

@Configuration
public class SpringMvcConfig extends WebMvcConfigurerAdapter {

    @Resource
    private LoginTokenInterceptor tokenInterceptor;

    @Override
    public void addInterceptors(InterceptorRegistry registry) {

        // 注册Token拦截器
        registry.addInterceptor(tokenInterceptor)
                .addPathPatterns("/**");

        super.addInterceptors(registry);
    }
}

4.测试用例

@RestController
public class TestController {

    /**
     * 测试,需要token
     */
    @GetMapping("/test")
    @NeedToken
    public String test() {
        return "访问成功";
    }

    /**
     * 测试,无token
     */
    @GetMapping("/test1")
    public String test1() {
        return "访问成功";
    }
}