SpringBoot获取上下文中的Bean对象

版权申明:本文为原创文章,转载请注明原文出处

原文链接:https://blog.it-follower.com/posts/4236301417.html

本文将讲解如何在SpringBoot中,以工具类的形式获取被Spring管理的Bean对象

方式一:

实现ApplicationContextAware,重写setApplicationContext方法。这个方式下,工具类也被注册成了Bean,既然这样,那就必须确保该类能被Spring自动扫描到。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
@Component
public class SpringContextUtils implements ApplicationContextAware {

private static ApplicationContext applicationContext;

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

public static <T> T getBean(Class<T> cls) {
if (applicationContext == null) {
throw new RuntimeException("applicationContext注入失败");
}
return applicationContext.getBean(cls);
}

public static Object getBean(String name) {
if (applicationContext == null) {
throw new RuntimeException("applicationContext注入失败");
}
return applicationContext.getBean(name);
}

public static <T> T getBean(String name, Class<T> cls) {
if (applicationContext == null) {
throw new RuntimeException("applicationContext注入失败");
}
return applicationContext.getBean(name, cls);
}

public static HttpServletRequest getRequest() {
ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder
.getRequestAttributes();
return requestAttributes == null ? null : requestAttributes.getRequest();
}
}

方式二:

我想要的工具类,往往会部署在公共jar中,一般情况下不能被SpringBoot程序扫描到。所以就有手动注入上下文的方式。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
@Slf4j
public class SpringUtils {

private SpringUtils() {
}

private static ApplicationContext context = null;

/**
* 初始化Spring上下文
*
* @param ctx 上下文对象
*/
public static void initContext(ApplicationContext ctx) {
if(ctx == null) {
log.warn("ApplicationContext is null.");
return;
}
context = ctx;
}


/**
* 根据类型获取Bean
*
* @param cls Bean类
* @param <T> Bean类型
* @return Bean对象
*/
public static <T> T getBean(Class<T> cls) {
return context == null ? null : context.getBean(cls);
}

/**
* 根据名称获取Bean
*
* @param name Bean名称
* @return Bean对象
*/
public static Object getBean(String name) {
return context == null ? null : context.getBean(name);
}

/**
* 根据Bean名称和类获取Bean对象
*
* @param name Bean名称
* @param cls Bean类
* @param <T> Bean类型
* @return Bean对象
*/
public static <T> T getBean(String name, Class<T> cls) {
return context == null ? null : context.getBean(name, cls);
}
}

此种方式不用实现ApplicationContextAware接口,但是需要手动设置上下文。所以在程序入口处,需要调用initContext方法,完成上下文的初始化。代码如下:

1
2
3
4
public static void main(String[] args) {
ApplicationContext context = SpringApplication.run(YCloudsServiceBApplication.class, args);
SpringUtils.initContext(context);
}

具体内容可参照我的github项目https://github.com/ye17186/myhelper-spring-boot-starter

评论