SpringBoot原始碼@ConditionalOnBean與@ConditionalOnClass你清楚嗎

碼上程式碼發表於2020-11-27
@ConditionalOnBean         //	當給定的在bean存在時,則例項化當前Bean
@ConditionalOnMissingBean  //	當給定的在bean不存在時,則例項化當前Bean
@ConditionalOnClass        //	當給定的類名在類路徑上存在,則例項化當前Bean
@ConditionalOnMissingClass //	當給定的類名在類路徑上不存在,則例項化當前Bean

下面我通過案例深入講下@ConditionalOnBean 註解,這個理解其它也就理解了。
一、@ConditionalOnBean概念
需求場景 比如下面一種場景,我在例項化People物件的時候,需要注入一個City物件。這個時候問題來了,如果city沒有例項化,那麼下面就會報空指標或者直接報錯。
所以這裡需求很簡單,就是當前city存在則例項化people,如果不存在則不例項化people,這個時候@ConditionalOnBean 的作用來了。

   @Bean
    public People people(City city) {
        //這裡如果city實體沒有成功注入 這裡就會報空指標
        city.setCityName("千島湖");
        city.setCityCode(301701);
        return new People("小小", 3, city);
    }
1@ConditionalOnBean註解定義
@Target({ ElementType.TYPE, ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Conditional(OnBeanCondition.class)
public @interface ConditionalOnBean {
    /**
     * 需要作為條件的類的Class物件陣列
     */
    Class<?>[] value() default {};
    /**
     * 需要作為條件的類的Name,Class.getName()
     */
    String[] type() default {};
    /**
     *  (用指定註解修飾的bean)條件所需的註解類
     */
    Class<? extends Annotation>[] annotation() default {};
    /**
     * spring容器中bean的名字
     */
    String[] name() default {};
    /**
     * 搜尋容器層級,當前容器,父容器
     */
    SearchStrategy search() default SearchStrategy.ALL;
    /**
     * 可能在其泛型引數中包含指定bean型別的其他類
     */
    Class<?>[] parameterizedContainer() default {};
}

下面舉例說明。

二、@ConditionalOnBean示例
1、Bean實體
1)City類

@Data
@ToString
@AllArgsConstructor
@NoArgsConstructor
public class City {
    /**
     * 城市名稱
     */
    private String cityName;
    /**
     * 城市code
     */
    private Integer cityCode;
}

2)People類
這裡City作為People一個屬性欄位。
@Data
@ToString
@NoArgsConstructor
@AllArgsConstructor
public class People {
  /**
     * 姓名
     */
    private String name;
    /**
     * 年齡
     */
    private Integer age;
    /**
     *  城市資訊
     */
    private City city;
}

2、Config類
這裡寫個正常的配置類,City成功注入到IOC容器中。
@Slf4j
@Configuration
public class Config {
    @Bean
    public City city() {
        City city = new City();
        city.setCityName("千島湖");
        return city;
    }
    @Bean
    public People people(City city) {
        //這裡如果city實體沒有成功注入 這裡就會報空指標
        city.setCityCode(301701);
        return new People("小小", 3, city);
    }
}

3、Test測試類
@SpringBootTest(classes = Application.class)
@RunWith(SpringRunner.class)
public class TestConditionOn {

    @Autowired(required=false)
    private People people;
    
    @Test
    public void test() {
        System.out.println("= = = = = = = = = = = = = ");
        System.out.println("people = " + people);
        System.out.println("= = = = = = = = = = = = = ");
    }
}

執行結果

一切正常,這個很符合我們實際開發中的需求。但是如果有一種情況,就是我的city並沒有被注入。我把上面這部分注視掉。

//    @Bean
//    public City city() {
//        City city = new City();
//        city.setCityName("千島湖");
//        return city;
//    }

再執行測試類

發現啟動直接報錯了,這當然不是我們希望看到的,我們是要當city已經注入那麼例項化people,如果沒有注入那麼不例項化people。

@Slf4j
@Configuration
public class Config {
//    @Bean
//    public City city() {
//        City city = new City();
//        city.setCityName("千島湖");
//        return city;
//    }
/**
     * 這裡加了ConditionalOnBean註解,就代表如果city存在才例項化people
     */
    @Bean
    @ConditionalOnBean(name = "city")
    public People people(City city) {
        //這裡如果city實體沒有成功注入 這裡就會報空指標
        city.setCityCode(301701);
        return new People("小小", 3, city);
    }
}

再執行測試類

很明顯,上面因為city已經註釋調,所以也導致無法例項化people,所以people為null。
注意有點要注意的,就是一旦使用@Autowired那就預設代表當前Bean一定是已經存在的,如果為null,會報錯。所以這裡要修改下。
@Autowired(required=false) //required=false 的意思就是允許當前的Bean物件為null。

總結講了這個註解,其它三個註解的意思大致差不多,在實際開發過程中可以根據實際情況使用該註解。

相關文章