C/C++/POSIX如何同时唤醒多线程 规避notify_all顺序唤醒导致的饥饿问题

核心思路 #

• 采用世代版本号机制标记生产完成状态，避免线程反复争抢锁
• 收窄条件变量互斥锁的临界区范围，仅用做版本号校验，唤醒后线程可立即并行执行
• 领导线程完成生产后需等待所有消费者完成消费，再释放领导权，从根源避免抢占饥饿

完整C++实现代码 #

#include <atomic>
#include <mutex>
#include <condition_variable>
#include <thread>
#include <vector>

std::atomic_flag leader_flag = ATOMIC_FLAG_INIT;
std::atomic<int> generation = 0;
std::atomic<int> waiter_count = 0;
std::mutex mtx;
std::condition_variable cv;

void do_something() { /* 领导线程生产逻辑 */ }
void do_some_other_thing() { /* 工作线程消费逻辑 */ }

void example() {
    if (!leader_flag.test_and_set()) {
        // 领导线程逻辑
        do_something();
        
        // 更新世代号，通知所有等待的工作线程
        {
            std::lock_guard<std::mutex> lk(mtx);
            generation++;
        }
        cv.notify_all();
        
        // 等待所有工作线程完成消费后再释放领导权，避免连续抢占
        while (waiter_count.load() > 0) {
            std::this_thread::yield();
        }
        
        leader_flag.clear();
    } else {
        // 工作线程逻辑
        waiter_count++;
        int cur_gen = generation.load();
        
        // 等待世代更新，临界区仅做版本号校验
        std::unique_lock<std::mutex> lk(mtx);
        cv.wait(lk, [&]{ return generation.load() > cur_gen; });
        lk.unlock();
        
        // 消费逻辑在锁外执行，可完全并行
        do_some_other_thing();
        waiter_count--;
    }
}

void runner() {
    std::vector<std::thread> threads;
    for (int i = 0; i < 10; ++i) {
        threads.emplace_back([]() {
            while (1) {
                example();
            }
        });
    }
    // 后续线程join逻辑
}

关键逻辑说明 #

1. 版本号校验逻辑：每个等待线程进入等待逻辑前先记录当前世代号，只有当世代号更新时才结束等待，天然规避虚假唤醒问题
2. 防抢占设计：领导线程生产完成后不会立刻释放领导标记，而是等待所有等待的消费者都完成消费后再清空标记，此时所有线程公平争抢下一轮领导权，不会出现领导线程连续抢占的问题
3. 低CPU占用：线程等待时阻塞在条件变量上，不会空转消耗CPU，领导线程的do_something如果是IO阻塞操作也不会占用CPU资源
4. 并行执行优化：传统notify_all序列化执行的原因是消费逻辑放在临界区内，唤醒后的线程需要争抢互斥锁才能执行逻辑，本方案将消费逻辑移至临界区外，所有线程校验完版本号后即可并行执行消费任务

POSIX C实现思路 #

如果使用纯C POSIX API，逻辑与上述实现完全一致，仅需替换对应组件即可：

• std::atomic<int>替换为C11标准的_Atomic int，或GCC内置原子操作
• std::mutex替换为pthread_mutex_t
• std::condition_variable替换为pthread_cond_t

该方案完全适配动态单生产者多消费者场景，可稳定满足需求。

内容的提问来源于stack exchange，提问作者陈泽霖