condition_variable：std::condition_variable::notify_all

引入

在多线程编程中，线程之间的等待与通知机制是确保程序高效与安全运行的关键。当一个或多个线程需要等待某个特定条件的实现时，C++的std::condition_variable提供了通用的解决方案。notify_all函数是条件变量的一部分，用于唤醒所有正在等待该条件变量的线程。这一机制在处理多个线程需要同步的复杂场景时尤其有效。本文将详细探讨std::condition_variable::notify_all的特性、使用方法，并提供完整示例代码和相关场景分析。

特性/函数/功能语法介绍

std::condition_variable是C++标准库中处理多线程条件同步的一个重要工具，定义在<condition_variable>头文件中。其成员函数notify_all的主要特性包括：

• 唤醒所有线程：与notify_one不同，notify_all会唤醒所有在条件变量上等待的线程，确保没有线程漏过通知。
• 高效的资源使用：通过唤醒所有线程，等待条件的多个线程可以同时继续执行，适合一些特定的同步场景。

语法

#include <condition_variable>
#include <mutex>

std::condition_variable cv;
std::mutex mtx;

// 一段需要条件满足的代码
void codeBlock() {
    std::unique_lock<std::mutex> lock(mtx);
    cv.wait(lock); // 等待条件
}

// 唤醒所有线程的函数
void notifyAll() {
    std::lock_guard<std::mutex> lock(mtx);
    cv.notify_all(); // 唤醒所有等待的线程
}

完整示例代码

以下示例代码展示了如何使用std::condition_variable::notify_all在多线程环境中进行通知：

#include <iostream>
#include <thread>
#include <condition_variable>
#include <chrono>
#include <vector>

std::mutex mtx;
std::condition_variable cv;
bool ready = false;
int resource = 0;

void worker(int id) {
    std::unique_lock<std::mutex> lock(mtx);
    while (!ready) { // 防止虚假唤醒
        cv.wait(lock);
    }
    std::cout << "Worker " << id << " processing resource: " << resource << std::endl;
}

void signal() {
    std::this_thread::sleep_for(std::chrono::seconds(2)); // 模拟长时间操作
    {
        std::lock_guard<std::mutex> lock(mtx);
        resource = 100; // 修改共享资源
        ready = true;   // 更改条件状态
        std::cout << "Signaling all workers...\n";
    }
    cv.notify_all(); // 唤醒所有等待的线程
}

int main() {
    const int numWorkers = 5;
    std::vector<std::thread> workers;

    for (int i = 0; i < numWorkers; ++i) {
        workers.emplace_back(worker, i);
    }

    std::thread notifier(signal);

    for (auto& worker : workers) {
        worker.join();
    }
    notifier.join();

    return 0;
}

代码解析

在以上代码示例中，我们定义了一个全局的条件变量cv和相应的互斥量mtx，以及一个共享资源resource和一个条件标志ready。

1. 工作线程：worker函数会试图获取mtx的独占锁，并在条件ready为false时被挂起。通过cv.wait(lock)保证了线程在等待期间能够释放对互斥量的持有，允许其他线程访问共享资源。
2. 信号线程：signal函数模拟了一个耗时的操作，完成后修改共享资源并设置ready为true，准备唤醒所有的工作线程。使用cv.notify_all()唤醒所有等待的线程，使其继续执行。
3. 高效的唤醒机制：在main函数中，我们创建了多个工作线程，并在信号线程中统一唤醒它们，让它们同时处理共享资源。

适用场景分析

notify_all在以下场景中尤为有效：

1. 多生产者/消费者模式：在复杂的生产者消费者模型中，当生产者向多个消费者提供资源时，唤醒所有在等待的消费者线程可以利用所有可用的线程资源快速度处理任务。
2. 批处理：在需要对多个线程同时做同样处理的情况下，notify_all可以高效地唤醒所有线程，让它们一起进行下一步的处理。
3. 全局状态更新：当一个线程完成了对共享资源的更新，并需要通知所有相关线程更新其行为或状态时，notify_all非常适合。

总结

std::condition_variable::notify_all是C++标准库中一个强大而灵活的工具，它简化了多线程间的同步和通信。通过唤醒所有等待的线程，notify_all确保了程序能够高效地管理线程状态。本文的示例和分析为开发者提供了如何运用这一工具来解决实际问题的视角。在面对多个线程需要协同工作和资源访问的场景时，合理运用notify_all将大大提高程序的效率和可维护性。