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
前面都是属于基本操作,这里才是扩展的核心。我们的实现逻辑是:
-
扫描 bean 的所有属性,然后找到定义了
IdGeneratorClient
注解的属性 -
获取注解的
value
值,作为 ID 生成器的分组标识 -
使用
IdGeneratorFactory
这个工厂类生成 ID 生成器实例,这里会返回新建的或已经定义的实例 -
通过反射将 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获取)
最后
还有Java核心知识点+全套架构师学习资料和视频+一线大厂面试宝典+面试简历模板可以领取+阿里美团网易腾讯小米爱奇艺快手哔哩哔哩面试题+Spring源码合集+Java架构实战电子书+2021年最新大厂面试题。
《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》,点击传送门即可获取!
img" style=“zoom: 33%;” />
最后
还有Java核心知识点+全套架构师学习资料和视频+一线大厂面试宝典+面试简历模板可以领取+阿里美团网易腾讯小米爱奇艺快手哔哩哔哩面试题+Spring源码合集+Java架构实战电子书+2021年最新大厂面试题。
[外链图片转存中…(img-bVCbI6RL-1711821265605)]
《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》,点击传送门即可获取!