SpringBoot 实战：通过 BeanPostProcessor 动态注入 ID 生成器

synchronized (ID_CACHE) {

ID_CACHE.computeIfAbsent(groupName, key -> new AtomicLong(1));

}

}

@Override

public String groupName() {

return this.groupName;

}

@Override

public long nextId() {

return ID_CACHE.get(this.groupName).getAndIncrement();

}

}

如前面设计的，我们需要一个工厂类来创建 ID 生成器，示例中使用最简单的实现，我们真正使用的时候，还可以通过更加灵活的 SPI 实现（关于 SPI 的实现，这里挖个坑，后面专门写一篇填坑）：

public enum IdGeneratorFactory {

INSTANCE;

private static final Map<String, IdGenerator> ID_GENERATOR_MAP = new ConcurrentHashMap<>(new HashMap<>());

public synchronized IdGenerator create(final String groupName) {

return ID_GENERATOR_MAP.computeIfAbsent(groupName, key -> new DefaultIdGenerator(groupName));

}

}

定义 BeanPostProcessor

前面都是属于基本操作，这里才是扩展的核心。我们的实现逻辑是：

1. 扫描 bean 的所有属性，然后找到定义了IdGeneratorClient注解的属性

2. 获取注解的value值，作为 ID 生成器的分组标识

3. 使用IdGeneratorFactory这个工厂类生成 ID 生成器实例，这里会返回新建的或已经定义的实例

4. 通过反射将 ID 生成器实例写入 bean

public class IdGeneratorBeanPostProcessor implements BeanPostProcessor {

@Override

public Object postProcessBeforeInitialization(final Object bean, final String beanName) throws BeansException {

return bean;

}

@Override

public Object postProcessAfterInitialization(final Object bean, final String beanName) throws BeansException {

parseFields(bean);

return bean;

}

private void parseFields(final Object bean) {

if (bean == null) {

return;

}

Class<?> clazz = bean.getClass();

parseFields(bean, clazz);

while (clazz.getSuperclass() != null && !clazz.getSuperclass().equals(Object.class)) {

clazz = clazz.getSuperclass();

parseFields(bean, clazz);

}

}

private void parseFields(final Object bean, Class<?> clazz) {

if (bean == null || clazz == null) {

return;

}

for (final Field field : clazz.getDeclaredFields()) {

try {

final IdGeneratorClient annotation = AnnotationUtils.getAnnotation(field, IdGeneratorClient.class);

if (annotation == null) {

continue;

}

final String groupName = annotation.value();

final Class<?> fieldType = field.getType();

if (fieldType.equals(IdGenerator.class)) {

final IdGenerator idGenerator = IdGeneratorFactory.INSTANCE.create(groupName);

invokeSetField(bean, field, idGenerator);

continue;

}

throw new RuntimeException("未知字段类型无法初始化，bean: " + bean + "，field: " + field);

} catch (Throwable t) {

throw new RuntimeException(“初始化字段失败，bean=” + bean + “，field=” + field, t);

}

}

}

private void invokeSetField(final Object bean, final Field field, final Object param) {

ReflectionUtils.makeAccessible(field);

ReflectionUtils.setField(field, bean, param);

}

}

实现BeanPostProcessor接口需要完成postProcessBeforeInitialization和postProcessAfterInitialization两个方法的定义。下图是 Spring 中 Bean 的实例化过程：

Spring 中 Bean 的实例化过程图示

从图中可以知道，Spring 调用BeanPostProcessor的这两个方法时，bean 已经被实例化，所有能注入的属性都已经被注入了，是一个完整的 bean。而且两个方法的返回值，可以是原来的 bean 实例，也可以是包装后的实例，这就要看我们的定义了。

测试我们的代码

写一个测试用例，验证我们的实现是否生效：

@SpringBootTest

class SpringBeanPostProcessorApplicationTests {

@IdGeneratorClient

private IdGenerator defaultIdGenerator;

@IdGeneratorClient(“group1”)

private IdGenerator group1IdGenerator;

@Test

void contextLoads() {

Assert.notNull(defaultIdGenerator, “注入失败”);

System.out.println(defaultIdGenerator.groupName() + " => " + defaultIdGenerator.nextId());

Assert.notNull(group1IdGenerator, “注入失败”);

for (int i = 0; i < 5; i++) {
自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注Java获取）

最后

2020年在匆匆忙忙慌慌乱乱中就这么度过了，我们迎来了新一年，互联网的发展如此之快，技术日新月异，更新迭代成为了这个时代的代名词，坚持下来的技术体系会越来越健壮，JVM作为如今是跳槽大厂必备的技能，如果你还没掌握，更别提之后更新的新技术了。

更多JVM面试整理：

《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》，点击传送门即可获取！
如此之快，技术日新月异，更新迭代成为了这个时代的代名词，坚持下来的技术体系会越来越健壮，JVM作为如今是跳槽大厂必备的技能，如果你还没掌握，更别提之后更新的新技术了。

[外链图片转存中…(img-YU1FZPOv-1712075931923)]

更多JVM面试整理：

[外链图片转存中…(img-l2hMaZ19-1712075931923)]

《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》，点击传送门即可获取！