Spring Boot如何让自己的bean优先加载

Posted 2023-03-12 amcomputer

背景介绍

在一些需求中，可能存在某些场景，比如先加载自己的bean，然后自己的bean做一些DB操作，初始化配置问题，然后后面的bean基于这个配置文件，继续做其他的业务逻辑。因此有了本文的这个题目。

实现方法

DependsOn注解

这个@DependsOn网上实现方法很多，依赖的bean数目较少的话，比较好弄，但数目变多后，就比较麻烦了，每个类都需要重新写一遍，因此推荐第二种方法。

ApplicationContextInitializer

通过注册 ApplicationContextInitializer 后，就可以注册 BeanDefinitionRegistryPostProcessor 到 Spring里面。最后实现 BeanDefinitionRegistryPostProcessor ，注册目标 bean。

 class DemoApplicationContextInitializer implements ApplicationContextInitializer<ConfigurableApplicationContext> 

  @Override
  public void initialize(ConfigurableApplicationContext applicationContext) 
    
      applicationContext.addBeanFactoryPostProcessor(new DemoBeanDefinitionRegistryPostProcessor());
  

实现 BeanDefinitionRegistryPostProcessor:

public class DemoBeanDefinitionRegistryPostProcessor implements BeanDefinitionRegistryPostProcessor, PriorityOrdered 
  // from BeanDefinitionRegistryPostProcessor interface
  @Override
  public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry beanDefinitionRegistry) throws BeansException 
// 重点在这里，这里可以把自己的 想要提起 实现的 或者初始化的 bean  加到这里
beanDefinitionRegistry.registerBeanDefinition("DemoService",new RootBeanDefinition(DemoService.class));
  

  // from BeanDefinitionRegistryPostProcessor interface
  @Override
  public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException 

  

// from PriorityOrdered  interface
  @Override
  public int getOrder() 
  // 排在 ConfigurationClassPostProcessor 之前即可
      return Ordered.LOWEST_PRECEDENCE - 1;
  

这里需要注意的是，不能使用 @Componet 或者其他的注解进行注册 BeanDefinitionRegistryPostProcessor。

因为@Componet 注解方式能注册的前提是 被 ConfigurationClassPostProcessor 扫描到，而现在，我们需要考虑我们的bean注册，要在这些bean之前，所以一定不能被“它-ConfigurationClassPostProcessor”管理 。 换一个角度思考，如果被 “它” 管理类 注册出来的bean 一定不可能排在 ConfigurationClassPostProcessor 的前面。

注意：@Order 这个只能控制 spring 自身 bean的顺序，对于 @Service @Component 、@Repository这些注解是不能控制的。

简单Demo

需求：笔者想让TestService 提前被注册，并且执行后，其他bean 才能被注册。

public class TestService 
    // 存放系统配置
    private static Map<String, String> GLOBAL_CONF = new HashMap<>();

    @PostConstruct
    public void init() 
        // 先做初始化 GLOBAL_CONF 或者其他 IO操作
        // GLOBAL_CONF.put(key, value);
    
    //其他 bean 通过这个方法获得系统配置
    public static String getGlobalConfig(String key) 
        return GLOBAL_CONF.get(key);
    

实现：（为了简单，直接实现上述的3个接口）

public class DemoBeanDefinitionRegistryPostProcessor implements BeanDefinitionRegistryPostProcessor, 
PriorityOrdered, 
ApplicationContextInitializer<ConfigurableApplicationContext> 

    /**
     *  第二步： 注册 自己的 bean
     *
     * @param beanDefinitionRegistry
     */
    // from BeanDefinitionRegistryPostProcessor interface
    @Override
    public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry beanDefinitionRegistry) throws BeansException 
        beanDefinitionRegistry.registerBeanDefinition("TestService",new RootBeanDefinition(TestService.class));
    

    // from BeanDefinitionRegistryPostProcessor interface
    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException 

    

// from PriorityOrdered  interface
    @Override
    public int getOrder() 
        return Ordered.LOWEST_PRECEDENCE - 1;
    

    /**
     *  第一步 先注册 到 configurableApplicationContext 中
     *
     * @param configurableApplicationContext
     */
    // from ApplicationContextInitializer  interface
    @Override
    public void initialize(ConfigurableApplicationContext configurableApplicationContext) 
        configurableApplicationContext.addBeanFactoryPostProcessor(new DemoBeanDefinitionRegistryPostProcessor());
    

思考： 那如何得到 或者 看到 spring bean 加载的顺序呢？
见下一篇。