Spring原始碼解析之基礎應用(三)
組合Java配置
在XML中,我們可以使用<import/>標籤,在一個XML檔案中引入另一個XML檔案,在Java類中,我們同樣可以在一個配置類中用@Import引入另一個配置類,被引入的配置類中的@Bean也會載入到spring容器。程式碼如下:
@Configuration public class ConfigA { @Bean public A a() { return new A(); } } @Configuration @Import(ConfigA.class) public class ConfigB { @Bean public B b() { return new B(); } }
我們將ConfigB作為配置類傳給AnnotationConfigApplicationContext進行spring容器初始化,我們不但可以從spring容器中獲得A和B所對應的bean,還可以得到ConfigA和ConfigB所對應的bean:
public static void main(String[] args) { ApplicationContext ctx = new AnnotationConfigApplicationContext(ConfigB.class); A a = ctx.getBean(A.class); B b = ctx.getBean(B.class); ConfigA configA = ctx.getBean(ConfigA .class); ConfigB configB = ctx.getBean(ConfigB.class); }
事實上,@Import並不要求我們引入一個配置類,我們也可以引入一個普通的類,spring容器同樣會幫我們把這個類初始化成一個bean:
package org.example.config; import org.example.beans.D; import org.springframework.context.annotation.Import; @Import(D.class) public class MyConfig3 { } package org.example.beans; public class D { }
測試用例:
@Test public void test10() { AnnotationConfigApplicationContext ac = new AnnotationConfigApplicationContext(MyConfig3.class); System.out.println(ac.getBean(D.class)); }
執行結果:
org.example.beans.D@134593bf
利用@Import,我們可以把一些第三方的外掛引入spring容器初始化為bean,當然,我們也可以在一個方法裡返回外掛的例項,方法用@Bean註解標註。不過,筆者認為,用@Import或者@Bean來引入bean還是分情況的,如果是引入一個類似DataSource的資料來源的bean,我們要在方法裡指定資料來源的地址、使用者名稱、密碼,那麼我們可以用@Bean,如果我們需要引入的第三方外掛不需要設定額外的屬性,則可以用@Import。
我們還可以用@Import來引入一個ImportBeanDefinitionRegistrar介面的實現,spring容器會幫我們回撥ImportBeanDefinitionRegistrar介面的實現,在ImportBeanDefinitionRegistrar介面的實現內,我們可以往spring容器註冊一個BeanDefinition,spring容器會根據BeanDefinition生成一個bean,BeanDefinition用來描述一個bean的class物件,比如這個bean的class是什麼?作用域是單例還是原型?是否懶載入?等等。幾乎可以認為,只要是存在於spring容器中的bean,都會有一個對應的BeanDefinition來描述這個bean。我們先來看下BeanDefinition介面的部分方法:
package org.springframework.beans.factory.config; …… public interface BeanDefinition extends AttributeAccessor, BeanMetadataElement { …… //設定bean所對應的class void setBeanClassName(@Nullable String beanClassName); //獲取bean所對應的class String getBeanClassName(); //設定bean的作用域 void setScope(@Nullable String scope); //獲取bean的作用域 String getScope(); //設定bean是否懶載入 void setLazyInit(boolean lazyInit); //判斷bean是否懶載入 boolean isLazyInit(); //設定bean在初始化前需要的依賴項 void setDependsOn(@Nullable String... dependsOn); //獲取bean在初始化前需要的依賴項 String[] getDependsOn(); …… }
ImportBeanDefinitionRegistrar可以讓我們重寫兩個方法,從方法名可以看出,註冊BeanDefinition,第一個方法僅比第二個多一個引數importBeanNameGenerator,為beanName生成器。
public interface ImportBeanDefinitionRegistrar { default void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry, BeanNameGenerator importBeanNameGenerator) { registerBeanDefinitions(importingClassMetadata, registry); } default void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) { } }
我們在MyImportBeanDefinitionRegistrar裡重寫了registerBeanDefinitions,由於BeanDefinition是介面,所以我們用spring編寫好的實現類GenericBeanDefinition進行註冊,我們聲明瞭一個GenericBeanDefinition後並設定其對應的屬性,然後將beanName和BeanDefinition註冊進registry:
package org.example.beans; import org.springframework.beans.factory.support.BeanDefinitionRegistry; import org.springframework.beans.factory.support.GenericBeanDefinition; import org.springframework.context.annotation.ImportBeanDefinitionRegistrar; import org.springframework.core.type.AnnotationMetadata; public class MyImportBeanDefinitionRegistrar implements ImportBeanDefinitionRegistrar { @Override public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) { GenericBeanDefinition beanDefinition = new GenericBeanDefinition(); beanDefinition.setBeanClass(D.class); registry.registerBeanDefinition("d", beanDefinition); } } package org.example.beans; public class D { }
我們在MyConfig4中@Import進MyImportBeanDefinitionRegistrar,並在後面的測試用例中把MyConfig4傳入應用上下文進行初始化。
package org.example.config; import org.example.beans.MyImportBeanDefinitionRegistrar; import org.springframework.context.annotation.Import; @Import(MyImportBeanDefinitionRegistrar.class) public class MyConfig4 { }
測試用例中我們根據在MyImportBeanDefinitionRegistrar註冊的beanName獲取bean,從執行結果可以看到,spring會根據我們傳入的BeanDefinition建立bean。
@Test public void test11() { AnnotationConfigApplicationContext ac = new AnnotationConfigApplicationContext(MyConfig4.class); System.out.println(ac.getBean("d")); }
執行結果:
org.example.beans.D@4b53f538
關於上面說的BeanDefinition如果看不懂的話可以先略過,後面還會再詳講BeanDefinition。
@DependsOn
有時候我們存在這樣的需求:AService初始化之前,BService必須先初始化,AService不通過屬性或建構函式引數顯式依賴於BService。比如:AService負責產出訊息,BService負責消費訊息,最理想的做法是BService必須先AService初始化,讓訊息能及時消費掉,而spring建立bean的順序是不固定的,所以我們可以使用@DependsOn來要求在初始化AService之前必須先初始化BService。
package org.example.beans; import org.springframework.context.annotation.DependsOn; import org.springframework.stereotype.Component; @Component @DependsOn({"c"}) public class B { public B() { System.out.println("B construct..."); } } package org.example.beans; import org.springframework.stereotype.Component; @Component public class C { public C() { System.out.println("C construct..."); } } package org.example.config; import org.springframework.context.annotation.ComponentScan; @ComponentScan("org.example.beans") public class MyConfig5 { }
測試用例:
@Test public void test12() { AnnotationConfigApplicationContext ac = new AnnotationConfigApplicationContext(MyConfig5.class); }
執行結果:
C construct... B construct...
我們在B類宣告對C類的依賴,所以spring在建立bean的先建立C再建立B,大家也可以嘗試下,把B類上的@DependsOn去掉,spring會先建立B再建立C。
FactoryBean
當建立一個bean的邏輯十分複雜,不適合以@Bean來標註方法返回時,我們可以建立一個類並實現FactoryBean介面,然後在FactoryBean的實現中編寫複雜的建立邏輯。我們來看下FactoryBean介面:
public interface FactoryBean<T> { //返回工廠所建立的例項,例項可被共享,取決於這個例項是單例還是原型 T getObject() throws Exception; //返回bean的class物件 Class<?> getObjectType(); //返回bean是否單例 default boolean isSingleton() { return true; } }
我們編寫一個機器人FactoryBean(RobotFactoryBean)用來產生機器人(Robot)物件:
package org.example.beans; import org.springframework.beans.factory.FactoryBean; import org.springframework.stereotype.Component; @Component public class RobotFactoryBean implements FactoryBean<Robot> { @Override public Robot getObject() throws Exception { return new Robot(); } @Override public Class<?> getObjectType() { return Robot.class; } @Override public boolean isSingleton() { return true; } } package org.example.beans; public class Robot { }
一般我們通過beanName來獲取bean時,都是通過類名來獲取,但FactoryBean有點特殊,如果我們要獲取Robot這個bean,需要傳入它的工廠名稱robotFactoryBean,如果我們需要獲取工廠物件本身,則在robotFactoryBean前面加一個'&'符號。
@Test public void test13() { AnnotationConfigApplicationContext ac = new AnnotationConfigApplicationContext(MyConfig5.class); Object robot = ac.getBean("robotFactoryBean"); System.out.println(robot); System.out.println(robot.getClass()); Object factoryBean = ac.getBean("&robotFactoryBean"); System.out.println(factoryBean); System.out.println(factoryBean.getClass()); }
執行結果:
org.example.beans.Robot@6b1274d2 class org.example.beans.Robot org.example.beans.RobotFactoryBean@7bc1a03d class org.example.beans.RobotFactoryBean
&n