C++ STL map多线程迭代器操作的线程安全性问题咨询

问题结论 #

1. 你的现有实现完全不具备线程安全性，存在严重的未定义行为风险 #

原因出在锁的粒度设计错误：你仅为单个成员函数（insert/erase/begin）加锁，锁会在每个函数调用结束后立即释放，而thread2中访问迭代器、拷贝键值对的操作是在无锁状态下执行的，中间存在非常大的竞争窗口：

• 调用my_map.begin()拿到迭代器后，锁已经释放，此时如果有其他线程执行erase操作删除了该迭代器指向的元素，你后续访问it->first、it->second就属于非法内存访问，直接触发未定义行为。
• 额外风险：如果调用begin()时map刚好为空，返回的迭代器等于end()，此时直接访问迭代器指向的内容也会直接崩溃。
• 即使没有其他线程做删除操作，你的访问逻辑也不符合线程安全要求：STL容器不允许并发的读和写操作同时执行，你读迭代器指向内容的时候没有锁，此时thread1的写入操作属于并发读写，同样是未定义行为。

2. 插入操作本身不会影响已合法拷贝的k、v值 #

std::map的插入操作不会导致已有元素的迭代器失效，也不会修改任何已存在的键值对内容。只要你是在迭代器合法、且持有锁的状态下完成了k、v的拷贝，那么拷贝得到的是独立于map存储的副本，后续thread1的任何插入操作都不会影响这部分副本的值。

正确实现方案 #

你需要将「获取迭代器、判断合法性、拷贝键值、删除元素」这一整个临界区都放在同一把锁的保护下，不要拆分锁的范围。推荐给封装的map类新增一个原子弹出首元素的方法，示例如下：

template <typename K, typename V>
class ThreadSafeMap {
private:
    std::map<K, V> map_;
    mutable std::mutex mutex_;
public:
    void insert(K k, V v) {
        std::lock_guard<std::mutex> lock(mutex_);
        map_.emplace(std::move(k), std::move(v));
    }

    // 原子弹出首元素，返回空optional表示map为空
    std::optional<std::pair<K, V>> try_pop_front() {
        std::lock_guard<std::mutex> lock(mutex_);
        if (map_.empty()) {
            return std::nullopt;
        }
        auto it = map_.begin();
        auto res = std::make_pair(it->first, it->second);
        map_.erase(it);
        return res;
    }
};

thread2的逻辑可以修改为：

while(1) {
    auto elem = my_map.try_pop_front();
    if (!elem) {
        // 空map可以做短暂sleep避免空转
        std::this_thread::sleep_for(std::chrono::milliseconds(10));
        continue;
    }
    auto& [k, v] = *elem;
    work(k, v);
}

这种实现下所有对map的操作都在锁的保护下完成，完全不存在竞争问题，符合线程安全要求。

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