spring boot 自定义@EnableXXX注解

前言

spring boot 自带了很多@EnableXXX这样的注解，通过这些注解我们可以很方便地启用某些功能，比如@EnableAutoConfiguration用来开启自动装配的功能。内部实现主要是通过@Import注解将指定的类实例注入之Spring IOC Container中，从下面代码可以看到@EnableAutoConfiguration的@Import

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
@AutoConfigurationPackage
@Import(AutoConfigurationImportSelector.class)
public @interface EnableAutoConfiguration {…………
}

`@Import`注解

spring boot 的@Import注解可以配置三种不同的class，根据不同的场景来选择不同的注入方式

普通的bean 或者带有@Configuration的bean 直接注入
实现ImportSelector接口注入
实现 ImportBeanDefinitionRegistrar接口注入

下面使用这三种不同方式的实例来演示一下

先创建三个需要被注入的类

public class LoggerService {public void saveLog(String log){System.out.println("log is saved");}
}
public class MonitorService {public  void saveMonitor(){System.out.println("cpu、memory is saved");}
}
public class CounterService {public void add(int count ){System.out.println("count is added");}
}

针对MonitorServic我们使用ImportSelector的方式来注入，这里需要创建ImportSelector的实现类

public class MonitorImportSelector implements ImportSelector {@Overridepublic String[] selectImports(AnnotationMetadata importingClassMetadata) {AnnotationAttributes annotationAttributes = AnnotationAttributes.fromMap(importingClassMetadata.getAnnotationAttributes(EnableCustService.class.getName()));//在这里可以拿到所有注解的信息，可以根据不同注解的和注解的属性来返回不同的class,// 从而达到开启不同功能的目的return new String[]{MonitorService.class.getName()};}
}

针对CounterService我们使用ImportBeanDefinitionRegistrar的方式来注入，这里需要创建ImportSelector的实现类

public class CounterDefinitionRegistrar implements ImportBeanDefinitionRegistrar {@Overridepublic void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {Class beanClass = CounterService.class;RootBeanDefinition beanDefinition = new RootBeanDefinition(beanClass);String beanName = StringUtils.uncapitalize(beanClass.getSimpleName());//在这里可以拿到所有注解的信息，可以根据不同注解来返回不同的class,从而达到开启不同功能的目的//通过这种方式可以自定义beanNameregistry.registerBeanDefinition(beanName, beanDefinition);}
}

自定`@EnableXXX`注解

准备工作做好后，我们开始创建自定义@EnableCustService

@Import({LoggerService.class,MonitorImportSelector.class,CounterDefinitionRegistrar.class})
public @interface EnableCustService {}

模拟使用`@EnableCustService`

@EnableCustService
@SpringBootApplication
public class SpringbootCodeMain {public static void main(String[] args) {ConfigurableApplicationContext applicationContext  = SpringApplication.run(SpringbootCodeMain.class, args);System.err.println("-->"+applicationContext.getBean(LoggerService.class));System.err.println("-->"+applicationContext.getBean(MonitorService.class));System.err.println("-->"+applicationContext.getBean(CounterService.class));}
}

启动程序输出：
–>com.maple.learn.enable.LoggerService@4943defe
–>com.maple.learn.enable.MonitorService@5eefa415
–>com.maple.learn.enable.CounterService@181d7f28

本文的项目代码git地址：https://github.com/amapleleaf/springboot-code.git ,本文只是讲了如何使用@EnableXXX，如果想了解其原理可以参考本人的另一篇文章<<spring boot自动装配之@EnableAutoConfiguration详解>>