Spring運(yùn)行時(shí)手動(dòng)注入bean的方法實(shí)例

更新時(shí)間：2022年05月31日 10:21:14 作者：xiuyuandashen

spring給我們提供了IOC服務(wù),讓我們可以用注解的方式,方便的使用bean的相互引用,下面這篇文章主要給大家介紹了關(guān)于Spring運(yùn)行時(shí)手動(dòng)注入bean的相關(guān)資料,需要的朋友可以參考下

有時(shí)候，會(huì)有這樣一個(gè)需求，在程序運(yùn)行時(shí)動(dòng)態(tài)生成的對(duì)象，需要注入到Spring容器中進(jìn)行管理。

下面是獲取Bean以及注入Bean的工具類

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

import java.lang.reflect.Field;
import java.util.Map;

@Component
public class SpringUtils implements ApplicationContextAware {

    private static ApplicationContext applicationContext;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        SpringUtils.applicationContext = applicationContext;
    }

    public static ApplicationContext getApplicationContext() {
        return applicationContext;
    }

    public static Object getBean(String name) throws BeansException {
        return applicationContext.getBean(name);
    }

    public static Object getBean(Class name) throws BeansException {
        return applicationContext.getBean(name);
    }

    /**
     * bean注入spring容器
     * map[id,className,...]
     * id 為 bean的標(biāo)識(shí)
     * className 為 類的全限定類名
     * ... 為其他屬性
     * @param map
     */
    public static void injectBean(Map<String, String> map) {
        String className = map.get("className");
        Class<?> aClass;
        if (className == null) {
            throw new RuntimeException("map參數(shù)缺少className");
        }

        try {
            aClass = Class.forName(className);
            if (aClass == null) {
                throw new RuntimeException("className不存在");
            }
        } catch (ClassNotFoundException e) {
            throw new RuntimeException(e);
        }
        BeanDefinitionBuilder builder = BeanDefinitionBuilder.genericBeanDefinition(aClass);
        Field[] declaredFields = aClass.getDeclaredFields();
        for (int i = 0; i < declaredFields.length; i++) {
            String fieldName = declaredFields[i].getName();
            if (map.get(fieldName) != null) {
                // 必須設(shè)置可訪問,否則下面的操作做不了
                // declaredFields[i].setAccessible(true);
                builder.addPropertyValue(fieldName, map.get(fieldName));
            }
        }
        BeanDefinitionRegistry registry = (BeanDefinitionRegistry) applicationContext;
        // 注冊(cè)bean 第一個(gè)參數(shù)為 name ，第二個(gè)為 bean定義類
        String id = map.get("id");
        if (id == null) {
            registry.registerBeanDefinition(className, builder.getBeanDefinition());
            return;
        }
        registry.registerBeanDefinition(id, builder.getBeanDefinition());
    }
}

測試

@Test
   public void test2() {
      HashMap<String, String> map = new HashMap<>();
      map.put("id", "helloTask");
      map.put("className", "com.example.demo.Task.HelloTask4");
      // 注冊(cè)bean
      SpringUtils.injectBean(map);
//    HelloTask4 helloTask =(HelloTask4) SpringUtils.getBean(HelloTask4.class);
      HelloTask4 helloTask = (HelloTask4) SpringUtils.getBean("helloTask");
      System.out.println(helloTask.getClass());
   }

附：利用注解向Spring容器中注入Bean

常用注解包含：@Controller、@Service、@Repository、@Component，其中@Controller、@Service、@Repository都是基于@Component的擴(kuò)展。通常的@Controller用于標(biāo)識(shí)處理前端請(qǐng)求的類，@Service用于標(biāo)識(shí)業(yè)務(wù)邏輯類，@Repository用于標(biāo)識(shí)DAO層的類，@Component為通用注解，可以標(biāo)注在任何想要注入容器中的類上面；

@Component
//@Controller
//@Service
//@Repository
class Stu{
    
}

也可以利用@Bean與@Configuration注入，其中@Configuration用于標(biāo)注一個(gè)配置類，@Bean用于標(biāo)注返回需注入Bean的方法；

@Configuration
class MyConfig{
    @Bean
    public Student getStudent(){
        return new Student();
    }
}

總結(jié)

到此這篇關(guān)于Spring運(yùn)行時(shí)手動(dòng)注入bean的文章就介紹到這了,更多相關(guān)Spring運(yùn)行手動(dòng)注入bean內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: