JVector项目中DenseIntMap并发修改导致size计数异常问题分析
2025-07-10 19:04:28作者:申梦珏Efrain
背景
在JVector项目的DenseIntMap实现中,存在一个并发场景下的size计数异常问题。DenseIntMap是一种基于数组实现的密集键值映射结构,它使用乐观读锁机制来提高并发性能,但在特定并发操作组合下会导致size计数永久性不一致。
问题本质
DenseIntMap的size计数机制在并发环境下存在同步漏洞。具体表现为:
- 当put和remove操作并发执行于同一个未存在的键时
- put操作首先修改了底层数组
- remove操作获取写锁并删除值(减少size计数)
- put操作检测到版本戳失效后重试
- 重试的put操作再次插入值(增加size计数)
最终结果是map中存在新插入的值,但size计数没有相应增加,导致永久性的计数不准确。
技术细节分析
DenseIntMap的实现采用了乐观并发控制策略:
- 读操作使用乐观锁,通过版本戳(stamp)检测并发修改
- 写操作(remove和扩容)使用悲观写锁
- put操作在检测到版本戳变化时会重试
这种设计在大多数情况下能提供良好的并发性能,但在特定时序下会出现问题:
// 伪代码展示问题场景
void put(K key, V value) {
do {
stamp = readStamp(); // 获取当前版本
oldValue = map.get(key);
isInsert = (oldValue == null);
// 此时另一个线程执行remove操作
} while (!validate(stamp)); // 检测到版本变化重试
if (isInsert) size.increment();
}
当重试发生时,虽然操作成功插入了值,但size计数可能已经因为并发remove而减少,导致最终计数不准确。
解决方案探讨
项目维护者考虑了多种解决方案:
- 悲观锁方案:改用ReadWriteLock完全同步,虽然实现简单但会降低并发性能
- 弱化size保证:
- 保留O(1)的isEmpty()检查
- 提供O(1)的modificationCount
- 将size()改为O(N)操作(遍历计数非空键)
- 保持乐观锁但修复计数:通过更精细的版本控制确保计数准确
最终采用了第一种方案,因为:
- 实际性能测试显示乐观/悲观锁对put操作影响不大
- 实现简单可靠
- 在JVector的使用场景中,构建索引时的节点添加操作本身开销较大
经验总结
这个案例展示了并发编程中的典型陷阱:
- 乐观并发控制虽然能提高性能,但需要仔细考虑所有可能的执行路径
- 复合操作(检查-执行)在并发环境下需要特别处理
- 有时简单的同步方案反而比复杂的无锁方案更可靠
- 性能优化应该基于实际场景的基准测试,而非理论假设
对于类似的高并发数据结构实现,开发者需要:
- 全面分析所有可能的并发操作组合
- 考虑重试机制对状态的影响
- 在性能与正确性之间做出合理权衡
登录后查看全文
热门项目推荐
相关项目推荐
atomcodeClaude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed. Get StartedRust0214
cann-learning-hubCANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。Jupyter Notebook0138
uni-appA cross-platform framework using Vue.jsJavaScript08
GLM-5.2智谱开源 GLM-5.2,这是针对长文本任务的最新旗舰模型。相较于前代产品 GLM-5.1,它在长文本任务处理能力上实现了显著飞跃,并且首次在稳定的 100 万 token 上下文中提供这一能力。Jinja00
SwanLab⚡️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / LLaMA Factory / veRL/ Swift / Ultralytics / MMEngine / Keras etc.Python00
tiny-universe《大模型白盒子构建指南》:一个全手搓的Tiny-UniverseJupyter Notebook03
项目优选
收起
deepin linux kernel
C
32
16
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
469
465
暂无描述
Dockerfile
778
5.08 K
本项目是CANN提供的transformer类大模型算子库,实现网络在NPU上加速计算。
C++
877
2.03 K
Ascend Extension for PyTorch
Python
758
968
本项目是CANN提供的神经网络类计算算子库,实现网络在NPU上加速计算。
C++
697
1.4 K
昇腾LLM分布式训练框架
Python
185
231
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
1.1 K
1.14 K
本仓库是 Flutter SDK 与 Flutter Engine 的 OpenHarmony 适配版本,由 CPF-Flutter 团队维护。开发者可使用熟悉的 Flutter 技术栈开发 OpenHarmony 应用,3.35.7 及以后的适配版本可基于本仓库源码构建支持 OpenHarmony 的 Flutter Engine。
Dart
1.04 K
271
JiuwenSwarm 是一款基于openJiuwen开发的智能AI Agent,它能够将大语言模型的强大能力,通过你日常使用的各类通讯应用,直接延伸至你的指尖。
Python
2.25 K
677