Spring Boot中的那些条件判断

25次阅读

共计 7416 个字符,预计需要花费 19 分钟才能阅读完成。

Spring Boot 中的那些 Conditional
spring boot 中为我们提供了丰富的 Conditional 来让我们得以非常方便的在项目中向容器中添加 Bean。本文主要是对各个注解进行解释并辅以代码说明其用途。
所有 ConditionalOnXXX 的注解都可以放置在 class 或是 method 上,如果方式在 class 上,则会决定该 class 中所有的 @Bean 注解方法是否执行。
@Conditional
下面其他的 Conditional 注解均是语法糖,可以通过下面的方法自定义 ConditionalOnXXXConditional 注解定义如下,接收实现 Condition 接口的 class 数组。
public @interface Conditional {
Class<? extends Condition>[] value();
}
而 Condition 接口只有一个 matchs 方法,返回是否匹配的结果。
public interface Condition {
boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata);
}
通过操作系统进行条件判断,从而进行 Bean 配置。当 Window 时,实例化 Bill 的 Person 对象,当 Linux 时,实例化 Linus 的 Person 对象。
//LinuxCondition, 为方便起见,去掉判断代码,直接返回 true 了
public class LinuxCondition implements Condition {
@Override
public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata annotatedTypeMetadata) {
return true;
}
}

//WindowsCondition, 为方便起见,去掉判断代码,直接返回 false 了
public class WindowsCondition implements Condition {
@Override
public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata metadata) {
return false;
}
}
@Data
@ToString
@AllArgsConstructor
@NoArgsConstructor
public class Person {
private String name;
private Integer age;
}
// 配置类
@Configuration
public class BeanConfig {

@Bean(name = “bill”)
@Conditional({WindowsCondition.class})
public Person person1(){
return new Person(“Bill Gates”,62);
}

@Bean(“linus”)
@Conditional({LinuxCondition.class})
public Person person2(){
return new Person(“Linus”,48);
}
}
public class AppTest {
AnnotationConfigApplicationContext applicationContext = new AnnotationConfigApplicationContext(BeanConfig.class);

@Test
public void test(){
String osName = applicationContext.getEnvironment().getProperty(“os.name”);
System.out.println(“ 当前系统为:” + osName);
Map<String, Person> map = applicationContext.getBeansOfType(Person.class);
System.out.println(map);
}
}
输出的结果:
当前系统为:Mac OS X{linus=Person(name=Linus, age=48)}
@ConditionalOnBean & @ConditionalOnMissingBean
这两个注解会对 Bean 容器中的 Bean 对象进行判断,使用的例子是配置的时候,如果发现如果没有 Computer 实例,则实例化一个备用电脑。
@Data
@AllArgsConstructor
@ToString
public class Computer {
private String name;
}
@Configuration
public class BeanConfig {
@Bean(name = “notebookPC”)
public Computer computer1(){
return new Computer(“ 笔记本电脑 ”);
}

@ConditionalOnMissingBean(Computer.class)
@Bean(“reservePC”)
public Computer computer2(){
return new Computer(“ 备用电脑 ”);
}
}
public class TestApp {
AnnotationConfigApplicationContext applicationContext = new AnnotationConfigApplicationContext(BeanConfig.class);
@Test
public void test1(){
Map<String,Computer> map = applicationContext.getBeansOfType(Computer.class);
System.out.println(map);
}
}
修改 BeanConfig, 如果注释掉第一个 @Bean, 会实例化备用电脑,否则就不会实例化备用电脑
@ConditionalOnClass & @ConditionalOnMissingClass
这个注解会判断类路径上是否有指定的类,一开始看到的时候比较困惑,类路径上如果没有指定的 class, 那编译也通过不了啊 … 这个主要用于集成相同功能的第三方组件时用,只要类路径上有该组件的类,就进行自动配置,比如 spring boot web 在自动配置视图组件时,是用 Velocity,还是 Thymeleaf, 或是 freemaker 时,使用的就是这种方式。例子是两套盔甲 A(光明套装) 和 B(暗黑套装),如果 A 不在则配置 B。
public interface Fighter {
void fight();
}
public class FighterA implements Fighter {
@Override
public void fight() {
System.out.println(“ 使用光明套装 ”);
}
}
public class FighterB implements Fighter {
@Override
public void fight() {
System.out.println(“ 使用暗黑套装 ”);
}
}
Van 是武士,使用套装进行战斗
@Data
@AllArgsConstructor
@NoArgsConstructor
public class Van {
private Fighter fighter;
public void fight(){
fighter.fight();
}
}
VanConfigA/ B 实例化武士
@Configuration
@ConditionalOnClass({FighterA.class})
public class VanConfigA {
@Primary
@Bean
public Van vanA(){
return new Van(new FighterA());
}
}
@Configuration
@ConditionalOnClass({FighterB.class})
public class VanConfigB {
@Bean
public Van vanB(){
return new Van(new FighterB());
}
}
测试类, 默认情况,如果套装 AB 都在类路径上,两套都会加载,A 会设置为 PRIMARY,如果在 target class 中将 FightA.class 删除,则只会加载套装 B。
@SpringBootApplication
public class TestApp implements CommandLineRunner {
@Autowired
private Van van;
public static void main(String[] args) {
SpringApplication.run(TestApp.class, args);
}
@Override
public void run(String… args) throws Exception {
//do something
van.fight();
}
}
另外,尝试将两个 VanConfigA/ B 合并,将注解 ConditionalOnClass 放到方法上,如果删除一个套装就会运行出错。
@ConditionalOnExpress
依据表达式进行条件判断,这个作用和 @ConditionalOnProperty 大部分情况可以通用,表达式更灵活一点, 因为可以使用 SpEL。例子中会判断 properties 中 test.enabled 的值进行判断。BeanConfig 分别对布尔,字符串和数字三种类型进行判断。数字尝试了很多其他的方式均不行,比如直接使用 ==,貌似配置的属性都会当成字符串来处理。
@Data
public class TestBean {
private String name;
}
@Configuration
@ConditionalOnExpression(“#{${test.enabled:true} }”)
//@ConditionalOnExpression(“‘zz’.equalsIgnoreCase(‘${test.name2}’)”)
//@ConditionalOnExpression(“new Integer(‘${test.account}’)==1”)
public class BeanConfig {
@Bean
public TestBean testBean(){
return new TestBean(“ 我是美猴王 ”);
}
}
@SpringBootApplication
public class TestAppCommand implements CommandLineRunner {
@Autowired
private TestBean testBean;

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

@Override
public void run(String… args) throws Exception {
System.out.println(testBean.getName());
}
}
@ConditionalOnProperty
适合对单个 Property 进行条件判断,而上面的 @ConditionalOnExpress 适合面对较为复杂的情况,比如多个 property 的关联比较。这个例子也给了三种基本类型的条件判断,不过貌似均当成字符串就可以 …
@Data
@AllArgsConstructor
@NoArgsConstructor
public class TestBean {
private String name;
}
@Configuration
@ConditionalOnProperty(prefix = “test”, name=”enabled”, havingValue = “true”,matchIfMissing = false)
//@ConditionalOnProperty(prefix = “test”, name=”account”, havingValue = “1”,matchIfMissing = false)
//@ConditionalOnProperty(prefix = “test”, name=”name1″, havingValue = “zz”,matchIfMissing = false)
public class BeanConfig {

@Bean
public TestBean testBean(){
return new TestBean(“ 我是美猴王 ”);
}
}
@SpringBootApplication
public class TestAppCommand implements CommandLineRunner {
@Autowired
private TestBean testBean;
public static void main(String[] args) {
SpringApplication.run(TestAppCommand.class, args);
}
@Override
public void run(String… args) throws Exception {
System.out.println(testBean.getName());

}
}
@ConditionalOnJava
可以通过 java 的版本进行判断。
@Data
public class TestBean {
}
@Configuration
@ConditionalOnJava(JavaVersion.EIGHT)
public class BeanConfig {

@Bean
public TestBean testBean(){
return new TestBean();
}
}
public class TestApp {
AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(BeanConfig.class);
@Test
public void test(){
Map<String,TestBean> map = context.getBeansOfType(TestBean.class);
System.out.println(map);
}
}
@ConditionalOnResource
通过指定的资源文件是否存在进行条件判断,比如判断 ehcache.properties 来决定是否自动装配 ehcache 组件。
@Data
public class TestBean {
}
@Configuration
@ConditionalOnResource(resources = “classpath:application.yml”)
public class BeanConfig {

@Bean
public TestBean testBean(){
return new TestBean();
}
}
public class TestApp {
AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(BeanConfig.class);

@Test
public void test(){
Map<String,TestBean> map = context.getBeansOfType(TestBean.class);
System.out.println(map);
}
}
@ConditionalOnSingleCandidate
这个还没有想到应用场景,条件通过的条件是:1 对应的 bean 容器中只有一个 2. 对应的 bean 有多个,但是已经制定了 PRIMARY。例子中,BeanB 装配的时候需要看 BeanA 的装配情况,所以 BeanBConfig 要排在 BeanAConfig 之后. 可以修改 BeanAConfig, 将 @Primary 注解去掉,或者把三个 @Bean 注解去掉,BeanB 就不会实例化了。
@Data
@AllArgsConstructor
@NoArgsConstructor
public class BeanA {
private String name;
}
@Configuration
public class BeanAConfig {

@Bean
@Primary
public BeanA bean1(){
return new BeanA(“bean1”);
}
@Bean(autowireCandidate = false)
public BeanA bean2(){
return new BeanA(“bean2”);
}
//@Bean(autowireCandidate = false)
public BeanA bean3(){
return new BeanA(“bean3”);
}
}
@Data
public class BeanB {
}
@Configuration
@AutoConfigureAfter(BeanAConfig.class)
@ConditionalOnSingleCandidate(BeanA.class)
public class BeanBConfig {

@Bean
public BeanB targetBean(){
return new BeanB();
}
}
public class TestApp {
AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(BeanAConfig.class, BeanBConfig.class);

@Test
public void test(){
Map<String,BeanA> map = context.getBeansOfType(BeanA.class);
System.out.println(map);
Map<String,BeanB> map2 = context.getBeansOfType(BeanB.class);
System.out.println(map2);
}
}
@ConditionalOnNotWebApplication & @ConditionalOnWebApplication
判断当前环境是否是 Web 应用。

正文完
 0