如何在C++的folly::ConcurrentHashMap中原子化更新字符串频率计数器？

你遇到的两个核心问题：

• 编译错误：it->second是const引用，这是folly::ConcurrentHashMap的设计限制——它的迭代器返回的value不可直接修改，避免非原子操作引发线程安全问题。
• 原子性问题：你写的emplace+自增不是原子操作，两个步骤之间存在竞态窗口，多线程下可能出现计数不准确的情况。

正确的原子化实现方式 #

要实现**“键不存在则插入1，存在则原子化加1”**，需要结合原子类型作为哈希表的值，搭配folly提供的线程安全API：

1. 先定义哈希表类型，将值设为原子类型：

folly::ConcurrentHashMap<std::string, std::atomic<size_t>> numOccurences;

2. 使用computeIfAbsent完成线程安全的“查缺补漏”，再通过原子操作自增：

// 键不存在时插入初始值1，存在则返回已有原子变量的引用
auto& count = numOccurences.computeIfAbsent(str, []() {
    return std::atomic<size_t>{1};
});
// 原子化自增，memory_order_relaxed适合计数场景（无需强内存屏障）
count.fetch_add(1, std::memory_order_relaxed);

方案说明 #

• computeIfAbsent是folly::ConcurrentHashMap的线程安全方法，保证同一个键只会被插入一次，避免多线程重复插入的竞态。
• std::atomic的fetch_add是原子操作，确保计数自增的过程不会被其他线程打断，完全满足频率统计的原子性要求。

替代方案（不推荐） #

如果不想使用原子类型，也可以用insertOrUpdate方法，它会在锁保护下执行插入或更新逻辑：

numOccurences.insertOrUpdate(
    str,
    []() { return 1; },  // 键不存在时的初始值
    [](auto& existing) { existing += 1; }  // 键存在时的更新逻辑
);

但这种方式依赖锁实现线程安全，性能通常不如原子类型方案，且非原子值在并发场景下仍有潜在风险。

内容的提问来源于stack exchange，提问作者alexlipa