Spring學習----------(4)@Conditional註解

Spring @Conditional註解 詳細講解及示例

@Conditional是Spring4新提供的註解，它的作用是按照一定的條件進行判斷，滿足條件給容器註冊bean。

@Conditional的定義：

//此註解可以標註在類和方法上
@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME) 
@Documented
public @interface Conditional {
    Class<? extends Condition>[] value();
}

從代碼中可以看到，需要傳入一個Class數組，並且需要繼承Condition接口：

public interface Condition {
    boolean matches(ConditionContext var1, AnnotatedTypeMetadata var2);
}

Condition是個接口，需要實現matches方法，返回true則注入bean，false則不注入。

示例：
首先，創建Person類：

public class Person {
 
    private String name;
    private Integer age;
 
    public String getName() {
        return name;
    }
 
    public void setName(String name) {
        this.name = name;
    }
 
    public Integer getAge() {
        return age;
    }
 
    public void setAge(Integer age) {
        this.age = age;
    }
 
    public Person(String name, Integer age) {
        this.name = name;
        this.age = age;
    }
 
    @Override
    public String toString() {
        return "Person{" + "name='" + name + '\'' + ", age=" + age + '}';
    }
}

創建MainConfig類，用於配置Person實例並注入，一個是張三，一個是李四。

@Configuration
public class MainConfig {

    //給aop容器註冊一個bean，類型爲返回值類型，默認bean id 是方法名，可以自定義
    @Bean
    public Person person(){

        System.out.println("給容器中 添加person........");
        return new Person("JJJ",12);
    }

    @Bean("zhangsan")//bean在容器中的id
    public Person zhangsan(){

        System.out.println("給容器中 添加zhangsan........");
        return new Person("zhangsan",12);
    }
    @Bean("lisi")
    public Person lisi(){

        System.out.println("給容器中 添加lisi.......");
        return new Person("lisi",56);
    }
}

接着寫一個測試類進行驗證這兩個Bean是否注入成功。

@Test
    public void test01(){

        ApplicationContext applicationContext = new AnnotationConfigApplicationContext(MainConfig.class);

        System.out.println("IOC容器創建完成.....");


    }

運行，輸出結果是這樣的，兩個Person實例被注入進容器。

這是一個簡單的例子，現在問題來了，如果我想根據當前操作系統來注入Person實例，windows下注入zhangsan，linux下注入lisi，怎麼實現呢？

這就需要我們用到@Conditional註解了，前言中提到，需要實現Condition接口，並重寫方法來自定義match規則。

首先，創建一個WindowsCondition類：

public class WindowsCondition implements Condition {

    /**
     *
     * @param conditionContext:判斷條件可以使用的上下文
     * @param annotatedTypeMetadata：註解的信息
     * @return
     */
    public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata annotatedTypeMetadata) {

        //能獲取到ioc容器正在使用的beanFactory
        ConfigurableListableBeanFactory beanFactory = conditionContext.getBeanFactory();
        System.out.println(beanFactory);
        //獲取當前環境變量
        Environment environment = conditionContext.getEnvironment();
        System.out.println(environment);
        String name = environment.getProperty("os.name");
        //包含Windows則說明是windows系統，返回true
        if(name.contains("Windows")){
            return  true;
        }
        return false;
    }
}

matches方法的兩個參數的意思在註釋中講述了，值得一提的是，conditionContext提供了多種方法，方便獲取各種信息，也是SpringBoot中 @ConditonalOnXX註解多樣擴展的基礎。

接着，創建LinuxCondition類：

public class LinuxCondition implements Condition {
    /**
     *
     * @param conditionContext:判斷條件可以使用的上下文
     * @param annotatedTypeMetadata：註解的信息
     * @return
     */
    public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata annotatedTypeMetadata) {

        //能獲取到ioc容器正在使用的beanFactory
        ConfigurableListableBeanFactory beanFactory = conditionContext.getBeanFactory();
        System.out.println(beanFactory);
        //獲取當前環境變量
        Environment environment = conditionContext.getEnvironment();
        System.out.println(environment);
        String name = environment.getProperty("os.name");
        if(name.contains("Linux")){
            return  true;
        }
        return false;
    }
}

接着就是使用這兩個類了，因爲此註解可以標註在方法上和類上，所以分開測試：

標註在方法上：

@Configuration
public class MainConfig {

    //給aop容器註冊一個bean，類型爲返回值類型，默認bean id 是方法名，可以自定義
    @Bean
    public Person person(){

        System.out.println("給容器中 添加person........");
        return new Person("JJJ",12);
    }

    //如果WindowsCondition的實現方法返回true，則注入這個bean
    @Conditional(WindowsCondition.class)
    @Bean("zhangsan")//bean在容器中的id
    public Person zhangsan(){

        System.out.println("給容器中 添加zhangsan........");
        return new Person("zhangsan",12);
    }
    //如果LinuxCondition的實現方法返回true，則注入這個bean
    @Conditional(LinuxCondition.class)
    @Bean("lisi")
    public Person lisi(){

        System.out.println("給容器中 添加lisi.......");
        return new Person("lisi",56);
    }
}

修改測試方法，使其可以打印當前系統名：

@Test
    public void test01(){

        ApplicationContext applicationContext = new AnnotationConfigApplicationContext(MainConfig.class);

        System.out.println("IOC容器創建完成.....");
        System.out.println("當前系統爲....."+applicationContext.getEnvironment().getProperty("os.name"));

    }

運行結果如下：

我是運行在windows上的所以只注入了zhansan。

接着實驗linux下的情況，不能運行在linux下，但可以修改運行時參數：

修改後啓動測試方法：

一個方法只能注入一個bean實例，所以@Conditional標註在方法上只能控制一個bean實例是否注入。

標註在類上：
一個類中可以注入很多實例，@Conditional標註在類上就決定了一批bean是否注入。

我們試一下，將MainConfig 改寫，這時，如果WindowsCondition返回true，則所有Person實例將被注入(注意：上一個測試將os.name改爲linux，這是我將把這個參數去掉)：

@Conditional(WindowsCondition.class)
//配置類 == 配置文件xml
@Configuration
public class MainConfig {

    //給aop容器註冊一個bean，類型爲返回值類型，默認bean id 是方法名，可以自定義
    @Bean
    public Person person(){

        System.out.println("給容器中 添加person........");
        return new Person("JJJ",12);
    }

    //如果WindowsCondition的實現方法返回true，則注入這個bean
    //@Conditional(WindowsCondition.class)
    @Bean("zhangsan")//bean在容器中的id
    public Person zhangsan(){

        System.out.println("給容器中 添加zhangsan........");
        return new Person("zhangsan",12);
    }
    //如果LinuxCondition的實現方法返回true，則注入這個bean
    //@Conditional(LinuxCondition.class)
    @Bean("lisi")
    public Person lisi(){

        System.out.println("給容器中 添加lisi.......");
        return new Person("lisi",56);
    }
}

結果兩個實例都被注入：

如果將類上的WindowsCondition.class改爲LinuxCondition.class，結果應該可以猜到：

結果就是空的，類中所有bean都沒有注入。

多個條件類：

前言中說，@Conditional註解傳入的是一個Class數組，存在多種條件類的情況。

這種情況貌似判斷難度加深了，測試一波，新增新的條件類，實現的matches返回false（這種寫死返回false的方法純屬測試用，沒有實際意義O(∩_∩)O）

public class ObstinateCondition implements Condition {

    //直接返回false,僅做測試使用，無任何實際意義
    public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata annotatedTypeMetadata) {
        return false;
    }
}

MainConfig修改一下：

@Conditional({WindowsCondition.class,ObstinateCondition.class})
//配置類 == 配置文件xml
@Configuration
public class MainConfig {

    //給aop容器註冊一個bean，類型爲返回值類型，默認bean id 是方法名，可以自定義
    @Bean
    public Person person(){

        System.out.println("給容器中 添加person........");
        return new Person("JJJ",12);
    }

    //如果WindowsCondition的實現方法返回true，則注入這個bean
    //@Conditional(WindowsCondition.class)
    @Bean("zhangsan")//bean在容器中的id
    public Person zhangsan(){

        System.out.println("給容器中 添加zhangsan........");
        return new Person("zhangsan",12);
    }
    //如果LinuxCondition的實現方法返回true，則注入這個bean
    //@Conditional(LinuxCondition.class)
    @Bean("lisi")
    public Person lisi(){

        System.out.println("給容器中 添加lisi.......");
        return new Person("lisi",56);
    }
}

結果：

現在如果將ObstinateCondition的matches方法返回值改成true，兩個bean就被注入進容器：

結論得：

第一個條件類實現的方法返回true，第二個返回false，則結果false，不注入進容器。

第一個條件類實現的方法返回true，第二個返回true，則結果true，注入進容器中。

多個條件類之間是&關係。

原文參考：https://blog.csdn.net/xcy1193068639/article/details/81491071