Spring中11個(gè)最常用的擴(kuò)展點(diǎn)分享2

5.1 通用類

這種引入方式是最簡(jiǎn)單的，引入的類會(huì)被實(shí)例化為一個(gè)bean對(duì)象。

public class A {
}

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

}

通過(guò)@Import注解引入類A，spring可以自動(dòng)實(shí)例化A對(duì)象，然后在需要使用的地方通過(guò)注解@Autowired注入：

@Autowired
private A a;

5.2 配置類

這種引入方式是最復(fù)雜的，因?yàn)锧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注解的配置類通過(guò)@Import注解導(dǎo)入，配置類@Import、@ImportResource相關(guān)注解引入的類會(huì)一次性全部遞歸引入@PropertySource所在的屬性。

5.3 ImportSelector

該導(dǎo)入方法需要實(shí)現(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方法返回的是一個(gè)數(shù)組，也就是說(shuō)可以同時(shí)引入多個(gè)類，非常方便。

5.4 ImportBeanDefinitionRegistrar

該導(dǎo)入方法需要實(shí)現(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 {
}

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

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

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

好消息是 SpringBoot 提供了：

• CommandLineRunner
• ApplicationRunner

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

它們的用法很簡(jiǎn)單，以ApplicationRunner接口為例：

@Component
public class TestRunner implements ApplicationRunner {

    @Autowired
    private LoadDataService loadDataService;

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

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

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

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

7. 修改BeanDefinition

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

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

答案 ：我們可以實(shí)現(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)對(duì)象，修改對(duì)象的屬性。

8. 初始化 Bean 前和后

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

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

該接口目前有兩個(gè)方法：

• 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等注解都是通過(guò)AutowiredAnnotationBeanPostProcessor和CommonAnnotationBeanPostProcessor來(lái)實(shí)現(xiàn)的。

9. 初始化方法

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

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

9.1 使用 @PostConstruct

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

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

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

@Service
public class BService implements InitializingBean {

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

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

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

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

此時(shí)你可以實(shí)現(xiàn) DisposableBean 接口并重寫(xiě)它的 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容器銷毀之前，會(huì)調(diào)用destroy方法做一些額外的工作。

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

11. 自定義Bean的scope

我們都知道spring core默認(rèn)只支持兩種Scope：

• Singleton單例，從spring容器中獲取的每一個(gè)bean都是同一個(gè)對(duì)象。
• prototype多實(shí)例，每次從spring容器中獲取的bean都是不同的對(duì)象。

Spring Web 再次擴(kuò)展了 Scope，添加

• RequestScope：同一個(gè)請(qǐng)求中從spring容器中獲取的bean都是同一個(gè)對(duì)象。
• SessionScope：同一個(gè)session從spring容器中獲取的bean都是同一個(gè)對(duì)象。

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

比如我們?cè)谕粋€(gè)線程中要從spring容器中獲取的bean都是同一個(gè)對(duì)象，怎么辦？

答案 ：這需要一個(gè)自定義范圍。

1. 實(shí)現(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個(gè)擴(kuò)展點(diǎn)，可以在Bean創(chuàng)建、初始化到銷毀各個(gè)階段注入自己想要的邏輯，也有Spring MVC相關(guān)的攔截器等擴(kuò)展點(diǎn)，希望對(duì)大家有幫助。