如何在Spring boot应用程序中为拦截器编写自定义启用注释？

Question

是否可以编写自定义启用注释以启用Spring Boot应用程序的拦截器。

这是我的密码，

配置

@Configuration
public class CustomConfig extends WebMvcConfigurerAdapter {

    @Autowired
    CustomInterceptor interceptor;

    @Override
      public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(interceptor);
    }

}

拦截器

@Component
public class CustomInterceptor implements HandlerInterceptor {

    @Override
    public void afterCompletion(HttpServletRequest arg0, HttpServletResponse arg1, Object arg2, Exception arg3)
            throws Exception {

        System.err.println("Executed!!!afterCompletion");

    }

    @Override
    public void postHandle(HttpServletRequest arg0, HttpServletResponse arg1, Object arg2, ModelAndView arg3)
            throws Exception {
        System.err.println("Executed!!!postHandle");

    }

    @Override
    public boolean preHandle(HttpServletRequest arg0, HttpServletResponse arg1, Object arg2) throws Exception {
        // TODO Auto-generated method stub
        return true;
    }

}

自定义批注

@Documented
@Retention(RUNTIME)
@Target(TYPE)
@Import(CustomConfig.class)
public @interface EnableCustomConfig {

}

仅当此批注

@启用CustomConfig

在主课堂上出现，

@SpringBootApplication
@EnableAutoConfiguration
@EnableCustomConfig
public class CustomEnableApplication {

    public static void main(String[] args) {
        SpringApplication.run(CustomEnableApplication.class, args);
    }
}

这可能在Spring启动中实现吗？如果是，请让我知道如何。

Answer 1

这里有几个选项。首先，我想我会完全跳过注释。您可以简单地扩展界面：

public CustomInterceptor extends HandlerInterceptor {}

让您所需的拦截器实现此接口，并使用ApplicationContext确定哪些Springbean实现此接口：

@Autowired
private ApplicationContext applicationContext;

@Override
public void addInterceptors(InterceptorRegistry registry) {
    Map<String, CustomInterceptor> interceptors = applicationContext.getBeansOfType(CustomInterceptor.class);
    for(CustomInterceptor interceptor : interceptors.values()){
          registry.addInterceptor(interceptor);
    }
}

如果设置为使用注释，则可以使用与上述基本相同的逻辑。您可以让所有bean实现HandlerInterceptor接口，然后检查它们的注释。可以通过以下方式检查bean的注释：

public boolean isAnnotatedWithMyAnnotation(Class clazz){
    return clazz.getAnnotation(EnableCustomConfig.class) == null ? false : true;
}