news 2025/12/24 8:03:21

手写简易Spring(八)

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
手写简易Spring(八)

参考小傅哥的教程:第09章:Aware感知容器对象 | 小傅哥 bugstack 虫洞栈

本期的目标是实现Aware功能,Aware是一个顶层接口,其拥有众多子类,如BeanClassLoaderAware、BeanNameAware、BeanFactoryAware、ApplicationContextAware接口,实现不同的Aware接口可以使Bean获得不一样的能力。

举个例子,假设我们的Bean实现了BeanFactoryAware接口,那意味着我们的Bean拥有获取Spring中的BeanFactory类对象的权力。

因此,Aware接口是赋予Bean感知容器中的相关对象,具体感知什么对象取决于实现了哪个Aware子类接口

实现

Aware

首先是顶层接口Aware,其只是一个标记性接口

/** * 标记类接口,实现该接口可以被Spring容器感知 */ public interface Aware { }

接下来是各个子接口,这些子接口有一个共同特点,都提供了相应的set方法

BeanFactoryAware

/** * 实现此接口,即能感知到所属的 BeanFactory */ public interface BeanFactoryAware extends Aware { void setBeanFactory(BeanFactory beanFactory) throws BeansException; }

BeanClassLoaderAware

/** * 实现此接口,既能感知到所属的 ClassLoader */ public interface BeanClassLoaderAware extends Aware { void setBeanClassLoader(ClassLoader classLoader); }

BeanNameAware

/** * 实现此接口,即能感知到所属的 BeanName */ public interface BeanNameAware extends Aware { void setBeanName(String name); }

ApplicationContextAware

/** * 实现此接口,即能感知到所属的 ApplicationContext */ public interface ApplicationContextAware extends Aware { void setApplicationContext(ApplicationContext applicationContext) throws BeansException; }

包装处理器

根据之前的学习,我们知道,管理Bean的流程是先通过XML获取对应的配置文件,然后在ApplicationContext中执行refresh操作,得到beanFactory,最后要回归到BeanFacotry中执行创建对象、依赖注入和各种前置、后置操作。

在BeanFacotry中,我们很容易能获取到相关的BeanName,BeanFactory,BeanClassLoader对象,但是无法获取到ApplicationContext对象,因为我们是先在ApplicationContext中执行了一些操作,再转到BeanFactory中执行剩余的操作。

为了解决在BeanFactory中无法获取ApplicationContext对象的问题,我们考虑在ApplicationContext中执行操作时,把ApplicationContext封装成一个BeanPostProcessor对象,直接把封装后的对象加入到BeanFactory中即可。

因此,就有了下面这个类

/** * 包装处理器 * 负责在 Bean 初始化之前,把 Spring 的 ApplicationContext 注入到实现了 * ApplicationContextAware 接口的 Bean 中,使这些 Bean 能够感知并访问容器本身。 */ public class ApplicationContextAwareProcessor implements BeanPostProcessor { // 要注入的 Spring 应用上下文 private final ApplicationContext applicationContext; public ApplicationContextAwareProcessor(ApplicationContext applicationContext) { this.applicationContext = applicationContext; } /** * 在 Bean 初始化(init-method、@PostConstruct 等)之前调用。 * 如果当前 Bean 实现了 ApplicationContextAware 接口,就把容器注入进去。 */ @Override public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException { // 如果bean 实现了 ApplicationContextAware 接口 if (bean instanceof ApplicationContextAware) { // 注入 ApplicationContext,让 Bean 可以访问容器资源 ((ApplicationContextAware) bean).setApplicationContext(applicationContext); } return bean; } /** * 在 Bean 初始化之后调用,这里不做任何处理,直接返回原 Bean。 */ @Override public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException { return bean; } }

最后由AbstractAutowireCapableBeanFactory.applyBeanPostProcessorsBeforeInitialization方法调用即可。

注册BeanPostProcessor

@Override public void refresh() throws BeansException { // 1. 子类负责创建/刷新 BeanFactory 并加载 BeanDefinition(模板方法) refreshBeanFactory(); // 2. 拿到已经完成定义的 BeanFactory(模板方法) ConfigurableListableBeanFactory beanFactory = getBeanFactory(); // 3. 添加 ApplicationContextAwareProcessor,让继承自 ApplicationContextAware 的 Bean 对象都能感知所属的 ApplicationContext // 由于创建Bean的时候无法获取到对应的ApplicationContext, 因此这里先把ApplicationContext做一个包装, 直接在这里获取 // 其余的Aware接口, 包括BeanFactory BeanName 都可以在创建Bean那块获取 beanFactory.addBeanPostProcessor(new ApplicationContextAwareProcessor(this)); // 4. 先执行 BeanFactoryPostProcessor,允许动态修改/补充 BeanDefinition invokeBeanFactoryPostProcessors(beanFactory); // 5. 再把所有 BeanPostProcessor 提前实例化并注册到工厂,后续创建的 Bean 才能被拦截 registerBeanPostProcessors(beanFactory); // 6. 一次性提前实例化所有单例 Bean(lazy-init=false 的) beanFactory.preInstantiateSingletons(); }

注意,这一块就是前文提到的有一些操作先在ApplicationContext中执行,在这个环境下执行的时候,我直接把当前ApplicationContext对象封装为BeanPostProcessor,交给BeanFactory。

感知调用操作

private Object initializeBean(String beanName, Object bean, BeanDefinition beanDefinition) { // 先检查Aware接口 // invokeAwareMethods if (bean instanceof Aware) { if (bean instanceof BeanFactoryAware) { ((BeanFactoryAware) bean).setBeanFactory(this); } if (bean instanceof BeanClassLoaderAware){ ((BeanClassLoaderAware) bean).setBeanClassLoader(getBeanClassLoader()); } if (bean instanceof BeanNameAware) { ((BeanNameAware) bean).setBeanName(beanName); } } // 1. 执行 BeanPostProcessor Before 处理 Object wrappedBean = applyBeanPostProcessorsBeforeInitialization(bean, beanName); // 执行 Bean 对象的初始化方法 try { invokeInitMethods(beanName, wrappedBean, beanDefinition); } catch (Exception e) { throw new BeansException("Invocation of init method of bean[" + beanName + "] failed", e); } // 2. 执行 BeanPostProcessor After 处理 wrappedBean = applyBeanPostProcessorsAfterInitialization(wrappedBean, beanName); return wrappedBean; }

这一块,在初始化阶段会检查当前Bean是否实现了相应的Aware接口,如果有,就把相关的对象给到Bean,这个在后面的例子中会更好理解。

而对于ApplicationContext对象,则会在执行BeanPostProcessor的前置操作时执行,因为ApplicationContext的封装对象重写了这个方法:

/** * 在 Bean 初始化(init-method、@PostConstruct 等)之前调用。 * 如果当前 Bean 实现了 ApplicationContextAware 接口,就把容器注入进去。 */ @Override public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException { // 如果bean 实现了 ApplicationContextAware 接口 if (bean instanceof ApplicationContextAware) { // 注入 ApplicationContext,让 Bean 可以访问容器资源 ((ApplicationContextAware) bean).setApplicationContext(applicationContext); } return bean; }

测试

定义UserService类

public class UserService implements BeanNameAware, BeanClassLoaderAware, ApplicationContextAware, BeanFactoryAware { private ApplicationContext applicationContext; private BeanFactory beanFactory; private String uId; private String company; private String location; private UserDao userDao; @Override public void setBeanFactory(BeanFactory beanFactory) throws BeansException { this.beanFactory = beanFactory; } @Override public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { this.applicationContext = applicationContext; } @Override public void setBeanName(String name) { System.out.println("Bean Name is:" + name); } @Override public void setBeanClassLoader(ClassLoader classLoader) { System.out.println("ClassLoader:" + classLoader); } public String queryUserInfo() { return userDao.queryUserName(uId) + "," + company + "," + location; } // 其余 get set 方法 }

这里的处理方式是,首先实现了各个Aware接口,同时在Bean中加入了相关字段,用来存储获取到的对象。之后,重写接口中的相关set方法,获取对象,将对象地址给到Bean中的相关字段。这也就赋予了Bean获取Spring中相关容器的能力。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2025/12/18 12:26:27

Kotaemon + GPU算力加速:释放大模型推理极致性能

Kotaemon GPU算力加速:释放大模型推理极致性能 在企业级智能对话系统日益复杂的今天,一个核心挑战始终摆在开发者面前:如何让大语言模型既“懂行”又“快答”?尤其是在面对专业领域的高频问答场景时,用户不会容忍长达…

作者头像 李华
网站建设 2025/12/18 12:24:57

【完整源码+数据集+部署教程】水上浮球定位系统源码分享[一条龙教学YOLOV8标注好的数据集一键训练_70+全套改进创新点发刊_Web前端展示]

一、背景意义 随着水上运动和水上活动的普及,水上浮球的定位与识别在安全管理、赛事组织以及环境监测等领域变得愈发重要。水上浮球作为水域标识的重要工具,其准确定位不仅有助于提高水上活动的安全性,还能为水域环境保护提供数据支持。传统的…

作者头像 李华
网站建设 2025/12/18 12:23:36

本体+知识图谱:RAG真正读懂了复杂工业技术文档

LLM 单啃工业标准会“消化不良”? 船舶、海工、能源等行业的工业标准(ASTM、API、ISO 等)往往长这样: 一份文档 60 页,层层嵌套“1-1.1-1.1.1-Table 3-Note b”;一段句子包含条件-例外-数值-单位四连击&…

作者头像 李华
网站建设 2025/12/20 2:11:35

GitHub 狂揽 6.3k Star!AI Agent 系统学习教程爆火!

如果说 2024 年是“百模大战”的元年,那么 2025 年无疑是“Agent 元年”。技术的焦点正从“训练更大的模型”转向**“构建更聪明的智能体应用”**。 那有没有一个开源、免费、系统性的智能体学习教程呢?答案是有的! Hello-Agents 就是这样一…

作者头像 李华
网站建设 2025/12/18 12:22:25

docker 新手入门:10分钟搞定基础使用

上周我第一次正儿八经用 Docker,不是光跑个 hello-world,而是把一个 Spring Boot 项目塞进去跑起来了。整个过程其实没想象中那么玄乎,就是几个命令来回敲。今天我就用大白话,把我踩过的坑和走通的路写下来,保证你跟着…

作者头像 李华
网站建设 2025/12/18 12:21:55

虾分发平台有哪些功能需要付费呢

虾分发平台部分功能需付费使用,主要包括以下方面: 高级套餐与增值服务:虾分发xiafenfa.com平台提供多种价格套餐,基础套餐(如30元/1000虾币)可满足小范围测试需求,而更高级的套餐则可能包含更多…

作者头像 李华