SpringBoot 上下文获取注入的Bean

1import org.springframework.beans.BeansException;
2import org.springframework.context.ApplicationContext;
3import org.springframework.context.ApplicationContextAware;
4import org.springframework.stereotype.Component;
5
6/**
7 * 获取Spring上下文
8 *
9 * @author kelin.ll
10 * @date on 2019/7/18
11 */
12@Component
13public class ApplicationContextProvider implements ApplicationContextAware {
14    /**
15     * 上下文对象实例
16     */
17    private static ApplicationContext applicationContext;
18
19    @Override
20    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
21        this.applicationContext = applicationContext;
22    }
23
24    /**
25     * 获取applicationContext
26     *
27     * @return
28     */
29    public static ApplicationContext getApplicationContext() {
30        return applicationContext;
31    }
32
33    /**
34     * 通过name获取 Bean.
35     *
36     * @param name
37     * @return
38     */
39    public static Object getBean(String name) {
40        return getApplicationContext().getBean(name);
41    }
42
43    /**
44     * 通过class获取Bean.
45     *
46     * @param clazz
47     * @param <T>
48     * @return
49     */
50    public static <T> T getBean(Class<T> clazz) {
51        return getApplicationContext().getBean(clazz);
52    }
53
54    /**
55     * 通过name,以及Clazz返回指定的Bean
56     *
57     * @param name
58     * @param clazz
59     * @param <T>
60     * @return
61     */
62    public static <T> T getBean(String name, Class<T> clazz) {
63        return getApplicationContext().getBean(name, clazz);
64    }
65}

测试用例：

1import com.provider.ServiceProviderApplication;import lombok.extern.slf4j.Slf4j;
2import org.junit.Test;
3import org.junit.runner.RunWith;
4import org.springframework.boot.test.context.SpringBootTest;
5import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
6
7/**
8 * @author kelin.ll
9 * @date on 2019/7/18
10 */
11@RunWith(SpringJUnit4ClassRunner.class)
12@SpringBootTest(classes = ServiceProviderApplication.class)
13@Slf4j
14public class ApplicationContextProviderTest {
15    @Test
16    public void getBeanByNameTest(){
17        TaskService taskService = (TaskService)ApplicationContextProvider.getBean("taskService");
18        System.out.println(taskService.getAll());
19    }
20
21    @Test
22    public void getBeanByClassTest(){
23        TaskService taskService = ApplicationContextProvider.getBean(TaskService.class);
24        System.out.println(taskService.getAll());
25    }
26
27    @Test
28    public void getBeanByNameAndClassTest(){
29        TaskService taskService = ApplicationContextProvider.getBean("taskService",TaskService.class);
30        System.out.println(taskService.getAll());
31    }
32}