写点什么

为什么你写的拦截器注入不了 Java bean?

用户头像
Java旅途
关注
发布于: 2020 年 11 月 23 日

一、如何实现拦截器


在 Spring Boot 项目中,拦截器经常被用来做登陆验证,日志记录等操作。拦截器是 Spring 提供的,所以可以将拦截器注成 bean,由 IOC 容器来管理。实现拦截器的方式很简单,主要由以下两个步骤:


  1. 自定义拦截器类实现HandlerInterceptor接口

  2. 自定义 WebMvc 配置类实现WebMvcConfigurer接口,添加自定义拦截器类


简要实现代码如下:


自定义拦截器 LoginInterceptor


public class LoginInterceptor implements HandlerInterceptor {
@Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
String token = request.getHeader("token"); if(StringUtils.isEmpty(token)){ ... return false; } return true; }}
复制代码


配置自定义拦截器:


@Configurationpublic class WebConfiguration implements WebMvcConfigurer {
@Override public void addInterceptors(InterceptorRegistry registry) { registry.addInterceptor(new LoginInterceptor()) // 拦截的请求 .addPathPatterns("/**") // 不用拦截的请求 .excludePathPatterns("/login"); }}
复制代码


这个拦截器的主要作用就是拦截所有访问请求,验证所带token是否有效,当token验证成功后,才能访问我们的业务接口。这时候就需要提供一个验证token有效性的接口,在拦截器中验证token,由于拦截器是 Spring 提供的,因此很容易想到使用@Component注解将拦截器注成一个 bean。然后使用@Autowired注解将验证token的类注入到拦截器进行验证。


改造完的代码如下:


验证 token 接口类:


@Componentpublic class TokenUtil {    /**     * 验证token 是否有效     */    public boolean checkToken(String token){        ...    }}
复制代码


改造完的拦截器代码如下:


@Componentpublic class LoginInterceptor implements HandlerInterceptor {
@Autowired private TokenUtil tokenUtil;
@Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { if(!tokenUtil.checkToken(token)){ ... return false; } return true; }}
复制代码


调用接口时发现,TokenUtil 并没有被注入进来!明明代码写的没问题,为什么不能正常注入TokenUtil呢?


仔细观察我们自定义的配置类WebConfiguration,在添加拦截器的时候用的是new LoginInterceptor(),**如果想要拦截器生效,必须将拦截器配置到 WebMvc 的配置类中,就是我们自定义的WebConfiguration类。现在添加拦截器的时候是 new 了一个拦截器,也就是说并没有将拦截器托管给 IOC 容器,所以就无法引入 Spring 的 bean 对象**。


二、如何将拦截器托管给 IOC 容器


解决问题的思路也很简单,就是将拦截器也托管给 IOC 容器,这样容器内的对象就可以相互注入了。总共有以下三种方式进行处理上述问题。


2.1 在 WebConfiguration 注入拦截器


拦截器代码不变,在拦截器上使用@Component,同时在WebConfiguration中使用@Autowired注解将拦截器注入。


拦截器代码:


@Componentpublic class LoginInterceptor implements HandlerInterceptor {}
复制代码


配置类代码:


@Configurationpublic class WebConfiguration implements WebMvcConfigurer {
@Autowired private LoginInterceptor loginInterceptor;
@Override public void addInterceptors(InterceptorRegistry registry) { registry.addInterceptor(loginInterceptor); }}
复制代码


2.2 在 WebConfiguration 将拦截器注成 bean


拦截器无需增加@Component注解,在WebConfiguration类中使用@Bean注解将拦截器注成 bean。


拦截器代码:


public class LoginInterceptor implements HandlerInterceptor {}
复制代码


配置类代码:


@Configurationpublic class WebConfiguration implements WebMvcConfigurer {
@Bean public LoginInterceptor loginInterceptor(){ return new LoginInterceptor(); }
@Override public void addInterceptors(InterceptorRegistry registry) { registry.addInterceptor( loginInterceptor()); }}
复制代码


2.3 通过构造器处理


思路是在WebConfiguration类中注入需要的验证token的业务类,然后在初始化拦截器的时候将业务类通过构造器带入拦截器中,这样就不用把拦截器注成 Spring Bean 对象了。


拦截器代码:


public class LoginInterceptor implements HandlerInterceptor {
private TokenUtil tokenUtil;
public LoginInterceptor(TokenUtil tokenUtil) { this.tokenUtil = tokenUtil; }}
复制代码


配置类代码:


@Configurationpublic class WebConfiguration implements WebMvcConfigurer {
@Autowired private TokenUtil tokenUtil;
@Override public void addInterceptors(InterceptorRegistry registry) { registry.addInterceptor(new LoginInterceptor(tokenUtil)); }}
复制代码


三、总结


网上关于拦截器的代码基本都是通过 new 一个拦截器进行配置的,这时候就会出现无法注入其他 bean 的情况。很多人想当然地直接在拦截器加@Component注解使其成为一个 bean 对象。这是一种错误的做法。我们需要保证的是在 WebMvc 配置类中添加的拦截器是 Spring 的一个 bean 对象,也就是说我们需要将拦截器注成一个 bean,同时将这个 bean 添加的 WebMvc 配置类中。


四、点关注、不迷路


如果觉得文章不错,欢迎关注、*点赞*、收藏,你们的支持是我创作的动力,感谢大家。


如果文章写的有问题,请不要吝惜文笔,欢迎留言指出,我会及时核查修改。


如果你还想看到更多别的东西,可以微信搜索「Java 旅途」进行关注。「Java 旅途」目前已经整理各种中间件的使用教程及各类 Java 相关的面试题。扫描下方二维码进行关注就可以得到这些资料。



发布于: 2020 年 11 月 23 日阅读数: 28
用户头像

Java旅途

关注

还未添加个人签名 2020.06.17 加入

公众号:Java旅途

评论

发布
暂无评论
为什么你写的拦截器注入不了 Java bean?