5.1 通用類

這種引入方式是最簡單的，引入的類會被實例化為一個bean對象。

public class A {
}

@Import(A.class)
@Configuration
public class TestConfiguration {

}

通過@Import注解引入類A，spring可以自動實例化A對象，然后在需要使用的地方通過注解@Autowired注入：

@Autowired
private A a;

5.2 配置類

這種引入方式是最復(fù)雜的，因為@Configuration支持還支持多種組合注解，比如：

• @Import
• @ImportResource
• @PropertySource

public class A {
}

public class B {
}

@Import(B.class)
@Configuration
public class AConfiguration {

    @Bean
    public A a() {
        return new A();
    }
}

@Import(AConfiguration.class)
@Configuration
public class TestConfiguration {
}

@Configuration注解的配置類通過@Import注解導(dǎo)入，配置類@Import、@ImportResource相關(guān)注解引入的類會一次性全部遞歸引入@PropertySource所在的屬性。

5.3 ImportSelector

該導(dǎo)入方法需要實現(xiàn)ImportSelector接口

public class AImportSelector implements ImportSelector {

    private static final String CLASS_NAME = "com.sue.cache.service.test13.A";

    public String[] selectImports(AnnotationMetadata importingClassMetadata) {
        return new String[]{CLASS_NAME};
    }
}

@Import(AImportSelector.class)
@Configuration
public class TestConfiguration {
}

這種方法的好處是selectImports方法返回的是一個數(shù)組，也就是說可以同時引入多個類，非常方便。

5.4 ImportBeanDefinitionRegistrar

該導(dǎo)入方法需要實現(xiàn)ImportBeanDefinitionRegistrar接口：

public class AImportBeanDefinitionRegistrar implements ImportBeanDefinitionRegistrar {
    @Override
    public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {
        RootBeanDefinition rootBeanDefinition = new RootBeanDefinition(A.class);
        registry.registerBeanDefinition("a", rootBeanDefinition);
    }
}

@Import(AImportBeanDefinitionRegistrar.class)
@Configuration
public class TestConfiguration {
}

這種方法是最靈活的。容器注冊對象可以在registerBeanDefinitions方法中獲取，可以手動創(chuàng)建BeanDefinition注冊到BeanDefinitionRegistry種。

6. 當(dāng)工程啟動時

有時候我們需要在項目啟動的時候自定義一些額外的功能，比如加載一些系統(tǒng)參數(shù)，完成初始化，預(yù)熱本地緩存等。我們應(yīng)該做什么？

好消息是 SpringBoot 提供了：

• CommandLineRunner
• ApplicationRunner

這兩個接口幫助我們實現(xiàn)了上面的需求。

它們的用法很簡單，以ApplicationRunner接口為例：

@Component
public class TestRunner implements ApplicationRunner {

    @Autowired
    private LoadDataService loadDataService;

    public void run(ApplicationArguments args) throws Exception {
        loadDataService.load();
    }
}

實現(xiàn)ApplicationRunner接口，重寫run方法，在該方法中實現(xiàn)您的自定義需求。

如果項目中有多個類實現(xiàn)了ApplicationRunner接口，如何指定它們的執(zhí)行順序？

答案是使用@Order(n)注解，n的值越小越早執(zhí)行。當(dāng)然，順序也可以通過@Priority注解來指定。

7. 修改BeanDefinition

在實例化Bean對象之前，Spring IOC需要讀取Bean的相關(guān)屬性，保存在BeanDefinition對象中，然后通過BeanDefinition對象實例化Bean對象。

如果要修改BeanDefinition對象中的屬性怎么辦？

答案 ：我們可以實現(xiàn) BeanFactoryPostProcessor 接口。

@Component
public class MyBeanFactoryPostProcessor implements BeanFactoryPostProcessor {

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException {
        DefaultListableBeanFactory defaultListableBeanFactory = (DefaultListableBeanFactory) configurableListableBeanFactory;
        BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(User.class);
        beanDefinitionBuilder.addPropertyValue("id", 123);
        beanDefinitionBuilder.addPropertyValue("name", "Tom");
        defaultListableBeanFactory.registerBeanDefinition("user", beanDefinitionBuilder.getBeanDefinition());
    }
}

在postProcessBeanFactory方法中，可以獲取BeanDefinition的相關(guān)對象，修改對象的屬性。

8. 初始化 Bean 前和后

有時，您想在 bean 初始化前后實現(xiàn)一些您自己的邏輯。

這時候就可以實現(xiàn)：BeanPostProcessor接口。

該接口目前有兩個方法：

• postProcessBeforeInitialization：應(yīng)該在初始化方法之前調(diào)用。
• postProcessAfterInitialization：此方法在初始化方法之后調(diào)用。

@Component
    public class MyBeanPostProcessor implements BeanPostProcessor {

        @Override
        public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
            if (bean instanceof User) {
                ((User) bean).setUserName("Tom");
            }
            return bean;
        }
    }

我們經(jīng)常使用的@Autowired、@Value、@Resource、@PostConstruct等注解都是通過AutowiredAnnotationBeanPostProcessor和CommonAnnotationBeanPostProcessor來實現(xiàn)的。

9. 初始化方法

目前在Spring中初始化bean的方式有很多種：

1. 使用@PostConstruct注解
2. 實現(xiàn)InitializingBean接口

9.1 使用 @PostConstruct

@Service
public class AService {
    @PostConstruct
    public void init() {
        System.out.println("===init===");
    }
}

為需要初始化的方法添加注解@PostConstruct，使其在Bean初始化時執(zhí)行。

9.2 實現(xiàn)初始化接口InitializingBean

@Service
public class BService implements InitializingBean {

    @Override
    public void afterPropertiesSet() throws Exception {
        System.out.println("===init===");
    }
}

實現(xiàn)InitializingBean接口，重寫afterPropertiesSet方法，在該方法中可以完成初始化功能。

10. 關(guān)閉Spring容器前

有時候，我們需要在關(guān)閉spring容器之前做一些額外的工作，比如關(guān)閉資源文件。

此時你可以實現(xiàn) DisposableBean 接口并重寫它的 destroy 方法。

@Service
public class DService implements InitializingBean, DisposableBean {

    @Override
    public void destroy() throws Exception {
        System.out.println("DisposableBean destroy");
    }

    @Override
    public void afterPropertiesSet() throws Exception {
        System.out.println("InitializingBean afterPropertiesSet");
    }
}

這樣，在spring容器銷毀之前，會調(diào)用destroy方法做一些額外的工作。

通常我們會同時實現(xiàn)InitializingBean和DisposableBean接口，重寫初始化方法和銷毀方法。

11. 自定義Bean的scope

我們都知道spring core默認只支持兩種Scope：

• Singleton單例，從spring容器中獲取的每一個bean都是同一個對象。
• prototype多實例，每次從spring容器中獲取的bean都是不同的對象。

Spring Web 再次擴展了 Scope，添加

• RequestScope：同一個請求中從spring容器中獲取的bean都是同一個對象。
• SessionScope：同一個session從spring容器中獲取的bean都是同一個對象。

盡管如此，有些場景還是不符合我們的要求。

比如我們在同一個線程中要從spring容器中獲取的bean都是同一個對象，怎么辦？

答案 ：這需要一個自定義范圍。

1. 實現(xiàn) Scope 接口

public class ThreadLocalScope implements Scope {
    private static final ThreadLocal THREAD_LOCAL_SCOPE = new ThreadLocal();

    @Override
    public Object get(String name, ObjectFactory? objectFactory) {
        Object value = THREAD_LOCAL_SCOPE.get();
        if (value != null) {
            return value;
        }

        Object object = objectFactory.getObject();
        THREAD_LOCAL_SCOPE.set(object);
        return object;
    }

    @Override
    public Object remove(String name) {
        THREAD_LOCAL_SCOPE.remove();
        return null;
    }

    @Override
    public void registerDestructionCallback(String name, Runnable callback) {
    }

    @Override
    public Object resolveContextualObject(String key) {
        return null;
    }

    @Override
    public String getConversationId() {
        return null;
    }
}

2. 將新定義的Scope注入到Spring容器中

@Component
public class ThreadLocalBeanFactoryPostProcessor implements BeanFactoryPostProcessor {
    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
        beanFactory.registerScope("threadLocalScope", new ThreadLocalScope());
    }
}

3. 使用新定義的Scope

@Scope("threadLocalScope")
@Service
public class CService {
    public void add() {
    }
}

總結(jié)

本文總結(jié)了Spring中很常用的11個擴展點，可以在Bean創(chuàng)建、初始化到銷毀各個階段注入自己想要的邏輯，也有Spring MVC相關(guān)的攔截器等擴展點，希望對大家有幫助。