Spring Boot 基于 ApplicationContextAware 扩展点的应用

在某些情况下，开发者可能需要更深入地与 Spring 容器进行交互，以实现一些特定的功能或需求。ApplicationContextAware 接口正是为此而设计的，它允许 Bean 获取到当前的 ApplicationContext，从而能够在运行时访问容器中的其他 Bean 或执行一些依赖于容器的操作.

什么是 ApplicationContextAware？

ApplicationContextAware 是 Spring 框架提供的一个接口，它允许 Bean 获取到当前的 ApplicationContext。通过实现这个接口，你可以访问到 Spring 容器中的其他 Bean 或者执行一些依赖于容器的操作。

public interface ApplicationContextAware extends Aware {
    void setApplicationContext(ApplicationContext applicationContext) throws BeansException;
}

扩展示例

假设我们希望设计一个工具类，在无法标记 @Service 或 @Component 注解的情况下，仍然可以获取到容器中的其他 Bean。这时，我们可以利用 ApplicationContextAware 接口来实现这一功能。

创建 ApplicationContextAware 实现类

首先，我们需要创建一个实现 ApplicationContextAware 接口的类，并将其标记为 @Component，以便 Spring 容器能够管理它。

@Component
public class SpringContextUtil implements ApplicationContextAware {

    private static ApplicationContext applicationContext;

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

    public static ApplicationContext getApplicationContext(){
        return applicationContext;
    }

    public static <T> T getBean(Class<T> type) {
        return applicationContext.getBean(type);
    }


    public static Object getBean(String beanName){
        return applicationContext.getBean(beanName);
    }
}

在这个类中，我们创建了一个静态的 ApplicationContext 变量，并在 setApplicationContext 方法中将其设置为传入的 ApplicationContext。这样，我们就可以在任何地方通过 SpringContextUtil 类来获取到 ApplicationContext 并从中获取 Bean。

使用示例

假设我们有一个服务类 TestService，我们需要在某个类中获取到这个服务类的实例。可以采用如下方式

 TestService service = SpringContextUtil.getBean(TestService.class);

总结

通过实现 ApplicationContextAware 接口，你可以在 Spring Boot 应用中方便地获取到 ApplicationContext。