SpringBoot 项目鉴权的 4 种方式

点击上方“码农突围”，马上关注
这里是码农充电第一站，回复“666”，获取一份专属大礼包
真爱，请设置“星标”或点个“在看”

前言

传统AOP

实现

1. 使用 @Aspect 声明一下切面类 WhitelistAspect；

2. 在切面类内添加一个切点 whitelistPointcut()，为了实现此切点灵活可装配的能力，这里不使用 execution 全部拦截，而是添加一个注解 @Whitelist，被注解的方法才会校验白名单。

3. 在切面类中使用 spring 的 AOP 注解 @Before 声明一个通知方法 checkWhitelist() 在 Controller 方法被执行之前校验白名单。

@Aspect  
public class WhitelistAspect {  
     
 @Before(value = "whitelistPointcut() && @annotation(whitelist)")  
 public void checkAppkeyWhitelist(JoinPoint joinPoint, Whitelist whitelist) {  
     checkWhitelist();  
     // 可使用 joinPoint.getArgs() 获取Controller方法的参数  
     // 可以使用 whitelist 变量获取注解参数  
 }  
     
     
 @Pointcut("@annotation(com.zhenbianshu.Whitelist)")  
 public void whitelistPointCut() {  
 }  
}  

扩展

Interceptor

实现

1. 定义拦截器类 AppkeyInterceptor 类并实现 HandlerInterceptor 接口。

2. 实现其 preHandle() 方法；

3. 在 preHandle 方法内通过注解和参数判断是否需要拦截请求，拦截请求时接口返回 false；

4. 在自定义的 WebMvcConfigurerAdapter 类内注册此拦截器；

@Component  
public class WhitelistInterceptor implements HandlerInterceptor {  
  
    @Override  
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {  
        Whitelist whitelist = ((HandlerMethod) handler).getMethodAnnotation(Whitelist.class);  
        // whitelist.values(); 通过 request 获取请求参数，通过 whitelist 变量获取注解参数  
        return true;  
    }  
  
    @Override  
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {  
  // 方法在Controller方法执行结束后执行  
    }  
  
    @Override  
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {  
  // 在view视图渲染完成后执行  
    }  
}  

扩展

@Configuration  
public class MvcConfiguration extends WebMvcConfigurerAdapter {  
    @Override  
    public void addInterceptors(InterceptorRegistry registry) {  
        registry.addInterceptor(new WhitelistInterceptor()).addPathPatterns("/*").order(1);  
        // 这里可以配置拦截器启用的 path 的顺序，在有多个拦截器存在时，任一拦截器返回 false 都会使后续的请求方法不再执行  
    }  
}  

ArgumentResolver

实现

1. 定义自定义参数类型 AuthParam，类内有 appkey 相关字段；

2. 定义 AuthParamResolver 并实现 HandlerMethodArgumentResolver 接口；

3. 实现 supportsParameter() 接口方法将 AuthParam 与 AuthParamResolver 适配起来；

4. 实现 resolveArgument() 接口方法解析 reqest 对象生成 AuthParam 对象，并在此校验 AuthParam ，确认 appkey 是否在白名单内；

5. 在 Controller Action 方法上签名内添加 AuthParam 参数以启用此 Resolver;

@Component  
public class AuthParamResolver implements HandlerMethodArgumentResolver {  
  
    @Override  
    public boolean supportsParameter(MethodParameter parameter) {  
        return parameter.getParameterType().equals(AuthParam.class);  
    }  
  
    @Override  
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {  
        Whitelist whitelist = parameter.getMethodAnnotation(Whitelist.class);  
        // 通过 webRequest 和 whitelist 校验白名单  
        return new AuthParam();  
    }  
}  

扩展

@Configuration  
public class MvcConfiguration extends WebMvcConfigurerAdapter {  
  
    @Override  
    public void addArgumentResolvers(List argumentResolvers) {  
        argumentResolvers.add(new AuthParamResolver());  
    }  
}  

Filter

public class WhitelistFilter implements javax.servlet.Filter {  
  
    @Override  
    public void init(FilterConfig filterConfig) throws ServletException {  
  // 初始化后被调用一次  
    }  
  
    @Override  
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {  
     // 判断是否需要拦截  
       chain.doFilter(request, response); // 请求通过要显示调用  
    }  
  
    @Override  
    public void destroy() {  
     // 被销毁时调用一次  
    }  
}  

扩展

@Configuration  
public class FilterConfiguration {  
  
    @Bean  
    public FilterRegistrationBean someFilterRegistration() {  
        FilterRegistrationBean registration = new FilterRegistrationBean();  
        registration.setFilter(new WhitelistFilter());  
        registration.addUrlPatterns("/*");  
        registration.setName("whitelistFilter");  
        registration.setOrder(1); // 设置过滤器被调用的顺序  
        return registration;  
    }  
}  

小结

来源：https://zhenbianshu.github.io