1. 简介
在使用 Spring 进行开发时，我们经常会用到 ApplicationContext 来获取 Spring 管理的 Bean 实例。但是，有些情况下，我们需要在自定义的 Bean 中获取到当前的 ApplicationContext 实例以及它的一些属性，这时就可以使用 ApplicationContextAware 接口。
ApplicationContextAware 接口是 Spring 提供的一个回调接口，实现该接口的 Bean 可以获取到当前上下文的实例，并进行相应的操作。
2. 实现过程
要实现 ApplicationContextAware 接口，需要完成以下几个步骤：
- 实现接口
ApplicationContextAware； - 重写接口方法
setApplicationContext，将上下文实例保存到成员变量中；
接下来，我们看一下 ApplicationContextAware 的源码实现。
3. 源码解析
首先，我们在 ApplicationContextAware 接口中可以看到以下方法定义：
public interface ApplicationContextAware extends Aware {
void setApplicationContext(ApplicationContext applicationContext) throws BeansException;
}
可以看到，ApplicationContextAware 接口继承了 Aware 接口，并定义了一个方法 setApplicationContext。
setApplicationContext 方法用于将当前上下文的实例传递给实现该接口的 Bean，让它可以获取当前的上下文信息。在 Spring 容器初始化时，会自动调用实现了 ApplicationContextAware 接口的 Bean 的 setApplicationContext 方法。
接下来，我们来看一下 ApplicationContextAwareProcessor 类中对 ApplicationContextAware 接口的处理逻辑。
public class ApplicationContextAwareProcessor implements BeanPostProcessor {
...
@Override
public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
if (!(bean instanceof ApplicationContextAware)) {
return bean;
}
// 强制类型转换
ApplicationContextAware aware = (ApplicationContextAware) bean;
// 获取当前的应用上下文
ApplicationContext applicationContext = getApplicationContext();
// 调用 setApplicationContext 方法
aware.setApplicationContext(applicationContext);
return bean;
}
...
}
在 ApplicationContextAwareProcessor 的 postProcessBeforeInitialization 方法中，会判断当前 Bean 是否实现了 ApplicationContextAware 接口，如果实现了，则将当前上下文传递给该 Bean。
通过 getApplicationContext 方法获取当前的应用上下文实例，然后将该实例传递给实现了 ApplicationContextAware 接口的 Bean 的 setApplicationContext 方法。
4. 使用示例
下面，我们通过一个示例来演示如何使用 ApplicationContextAware 接口。
首先，我们创建一个实现了 ApplicationContextAware 接口的 Bean：
@Component
public class MyBean implements ApplicationContextAware {
private ApplicationContext applicationContext;
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
this.applicationContext = applicationContext;
}
public void doSomething() {
// 获取当前上下文的一些信息
String appName = applicationContext.getEnvironment().getProperty("spring.application.name");
System.out.println("当前应用名称：" + appName);
}
}
在 setApplicationContext 方法中，将传入的上下文实例存储到成员变量中。
接下来，我们创建一个使用了 MyBean 的服务类：
@Service
public class MyService {
private MyBean myBean;
@Autowired
public MyService(MyBean myBean) {
this.myBean = myBean;
}
public void doSomething() {
myBean.doSomething();
}
}
在这个服务类中，我们使用了 MyBean，并通过构造函数注入到该服务类中。
最后，我们定义一个启动类，并运行：
@SpringBootApplication
public class Application {
public static void main(String[] args) {
ApplicationContext context = SpringApplication.run(Application.class, args);
MyService myService = context.getBean(MyService.class);
myService.doSomething();
}
}
在启动类中，我们获取 MyService 的实例，并调用 doSomething 方法。由于在 MyBean 中实现了 setApplicationContext 方法，所以在调用 MyService 的 doSomething 方法时，能够获取到当前应用的名称。
5. 总结
通过 ApplicationContextAware 接口，我们可以在实现了该接口的 Bean 中获取到当前的应用上下文实例，并进行相应的操作。这在某些场景下非常有用，例如获取应用配置信息、获取 Bean 实例等。
希望通过本文对 Spring 的 ApplicationContextAware 接口有一个更深入的了解，并能够在实际开发中灵活应用。
本文来自极简博客，作者：深海游鱼姬，转载请注明原文链接：Spring ApplicationContextAware 相关源码
