C++11的condition_variable實現WaitForSingleObject功能

• 在多線程程序開發中，我們有時會希望等待其他線程結束後在繼續執行某些任務。比如常見的，主線程需要等待其他所有子線程結束後再結束程序。在Windows下我們一般使用WaitForSingleObject，或者WaitForMultipleObjects來實現，它們都會等待Object被設置爲有信號時才返回的。這樣我們在主線程結束的地方wait，在子線程結束的地方設置信號，就能實現上述功能。
• 但是這個API是Windows下的，無法跨平臺使用。還好C++11爲我們帶來了多線程的支持。在C++11中我們可以通過mutex和condition_variable來達到相似效果。下面是我寫的小例子。

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

std::mutex mx;

std::mutex g_mx;
std::condition_variable g_cv;
uint32_t g_thread_num = 0;

void function_1()
{
    for (int i = 0; i < 100; i++)
    {
        std::lock_guard<std::mutex> loc(mx);
        std::cout << "from function 1:" << i << std::endl;
    }

    std::unique_lock<std::mutex> guard(g_mx);
    --g_thread_num;
    g_cv.notify_one();
    guard.unlock();

    std::cout << "function 1 end" << std::endl;
}
void function_2()
{
    for (int i = 0; i < 100; i++)
    {
        std::lock_guard<std::mutex> loc(mx);
        std::cout << "from function 2:" << i << std::endl;
    }

    std::unique_lock<std::mutex> guard(g_mx);
    --g_thread_num;
    g_cv.notify_one();
    guard.unlock();

    std::cout << "function 2 end" << std::endl;
}
void function_3()
{
    for (int i = 0; i < 100; i++)
    {
        std::lock_guard<std::mutex> loc(mx);
        std::cout << "from function 3:" << i << std::endl;
    }
    std::this_thread::sleep_for(std::chrono::seconds(3));

    std::unique_lock<std::mutex> guard(g_mx);
    --g_thread_num;
    g_cv.notify_one();
    guard.unlock();

    std::cout << "function 3 end" << std::endl;
}

int main()
{
    std::thread t1(function_1);
    {
        std::unique_lock<std::mutex> guard(g_mx);
        ++g_thread_num;
    }
    t1.detach();
    std::thread t2(function_2);
    {
        std::unique_lock<std::mutex> guard(g_mx);
        ++g_thread_num;
    }
    t2.detach();
    std::thread t3(function_3);
    {
        std::unique_lock<std::mutex> guard(g_mx);
        ++g_thread_num;
    }
    t3.detach();

    for (int i = 0; i < 100; i++)
    {
        std::lock_guard<std::mutex> loc(mx);
        std::cout << "from function main:" << i << std::endl;
    }
    

    {
        std::unique_lock<std::mutex> guard(g_mx);
        g_cv.wait(guard, []() { return g_thread_num == 0; });
    }

    std::cout << "main end" << std::endl;

    return 0;
}

• 例子寫的很簡單，其實可以簡單封裝下會更好用。大家如果有什麼更好的解決方案也歡迎交流。