快捷導(dǎo)航

Spring之動(dòng)態(tài)注冊(cè)bean的實(shí)現(xiàn)方法

更新時(shí)間：2018年08月06日 10:46:57 作者：小灰灰Blog

這篇文章主要介紹了Spring之動(dòng)態(tài)注冊(cè)bean的實(shí)現(xiàn)方法,小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧

Spring之動(dòng)態(tài)注冊(cè)bean

什么場(chǎng)景下，需要主動(dòng)向Spring容器注冊(cè)bean呢？

如我之前做個(gè)的一個(gè)支持掃表的基礎(chǔ)平臺(tái)，使用者只需要添加基礎(chǔ)配置 + Groovy任務(wù)，就可以丟到這個(gè)平臺(tái)上面來(lái)運(yùn)行了，而這個(gè)基礎(chǔ)平臺(tái)是一直都在運(yùn)行的，所以在新來(lái)任務(wù)時(shí)，最直觀(guān)需要注冊(cè)的就是 DataSource 數(shù)據(jù)源這個(gè)bean了，那么可以怎么玩？

I. 主動(dòng)注冊(cè)Bean支持

借助BeanDefinition來(lái)實(shí)現(xiàn)bean的定義，從最終的使用來(lái)看，代碼比較少，幾行而已

public <T> T registerBean(String name, Class<T> clazz, Object... args) {
   BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(clazz);
   if (args.length > 0) {
     for (Object arg : args) {
       beanDefinitionBuilder.addConstructorArgValue(arg);
     }
   }
   BeanDefinition beanDefinition = beanDefinitionBuilder.getRawBeanDefinition();
 
   BeanDefinitionRegistry beanFactory = (BeanDefinitionRegistry) applicationContext.getBeanFactory();
   beanFactory.registerBeanDefinition(name, beanDefinition);
   return applicationContext.getBean(name, clazz);
}

測(cè)試如下

import com.github.hui.story.quickstory.server.VisitService;
import lombok.ToString;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.stereotype.Component;

@Component
public class BeanHolder {
  private final ConfigurableApplicationContext applicationContext;

  public BeanHolder(ConfigurableApplicationContext applicationContext) {
    this.applicationContext = applicationContext;
    initSer();
  }

  public void initSer() {
    InrSer ser = registerBean("test", InrSer.class);
    ser.name = "一灰";
    ser.uid = 22;
    System.out.println(ser);

    InrSer ser2 = registerBean("test2", InrSer.class, "一灰灰Blog", 20);
    System.out.println(ser2);
  }

  @ToString
  public static class InrSer {
    private String name;
    private Integer uid;

    @Autowired
    private VisitService visitService;

    public InrSer() {
    }

    public InrSer(String name, Integer uid) {
      this.name = name;
      this.uid = uid;
    }
  }

  private <T> T registerBean(String name, Class<T> clazz, Object... args) {
    BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(clazz);
    if (args.length > 0) {
      for (Object arg : args) {
        beanDefinitionBuilder.addConstructorArgValue(arg);
      }
    }
    BeanDefinition beanDefinition = beanDefinitionBuilder.getRawBeanDefinition();

    BeanDefinitionRegistry beanFactory = (BeanDefinitionRegistry) applicationContext.getBeanFactory();
    beanFactory.registerBeanDefinition(name, beanDefinition);
    return applicationContext.getBean(name, clazz);
  }
}

輸出如下