为什么你写的拦截器注入不了 Java bean?没想过原因吗?

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

HandlerInterceptor
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;
    }
}

配置自定义拦截器:

@Configuration
public 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接口类:

@Component
public class TokenUtil {
    /**
     * 验证token 是否有效
     */
    public boolean checkToken(String token){
        ...
    }
}

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

@Component
public 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 注解将拦截器注入。

拦截器代码:

@Component
public class LoginInterceptor implements HandlerInterceptor {
}

配置类代码:

@Configuration
public 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 {
}

配置类代码:

@Configuration
public 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;
    }
}

配置类代码:

@Configuration
public 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 bean?没想过原因吗?)