C++stop_callback注册停止回调

2026-04-02 00:35:24 1583阅读 0评论

C++中的`std::stop_callback`：优雅地处理线程池关闭

在现代C++编程中，线程池是一种非常常见的工具，用于管理一组线程，提高程序的并发性能和资源利用率。然而，在多线程环境中，如何优雅地处理线程池的关闭是一个不容忽视的问题。std::stop_callback正是为了解决这个问题而设计的。

理解`std::stop_token`

在C++20中引入了std::stop_token，它允许你在多个线程间共享一个停止信号。当某个线程检测到停止信号时，可以执行清理操作或其他必要的任务。std::stop_token通常与std::jthread一起使用，后者是std::thread的智能包装器，支持异常安全的线程管理。

注册停止回调

std::stop_token的一个重要功能就是注册停止回调。你可以通过调用std::stop_token::attach_stop_callback方法来注册一个回调函数。这个回调函数会在线程池关闭时被自动调用，从而确保所有线程都能正确地处理关闭操作。

C++stop_callback注册停止回调

以下是一个简单的示例，展示了如何使用std::stop_token和std::stop_callback：

#include <iostream>
#include <thread>
#include <vector>
#include <stop_token>

void worker(std::stop_token stoken) {
    while (!stoken.stop_requested()) {
        // 执行一些工作
        std::cout << "Working..." << std::endl;
        std::this_thread::sleep_for(std::chrono::seconds(1));
    }
    std::cout << "Stopping..." << std::endl;
}

int main() {
    std::vector<std::jthread> threads;

    for (int i = 0; i < 5; ++i) {
        threads.emplace_back(worker);
    }

    // 让主线程休眠一段时间，以便子线程开始工作
    std::this_thread::sleep_for(std::chrono::seconds(3));

    // 请求停止所有线程
    for (auto& thread : threads) {
        if (thread.joinable()) {
            thread.request_stop();
        }
    }

    // 等待所有线程完成
    for (auto& thread : threads) {
        if (thread.joinable()) {
            thread.join();
        }
    }

    return 0;
}

在这个示例中，我们创建了一个包含五个线程的线程池，并让它们执行一些工作。当主线程请求停止所有线程时，每个线程都会检查停止信号并执行相应的清理操作。

实际应用中的挑战

虽然std::stop_token和std::stop_callback提供了强大的功能，但在实际应用中仍然会遇到一些挑战。例如，如果线程池中的某些任务需要很长时间才能完成，那么即使线程收到了停止信号，也可能无法立即退出。为了解决这个问题，你可以在任务中定期检查停止信号，并在必要时提前退出。

以下是一个改进的示例，展示了如何在任务中定期检查停止信号：

#include <iostream>
#include <thread>
#include <vector>
#include <stop_token>

void long_running_task(std::stop_token stoken) {
    int progress = 0;
    while (!stoken.stop_requested() && progress < 100) {
        // 模拟长时间运行的任务
        std::cout << "Progress: " << progress << "%" << std::endl;
        std::this_thread::sleep_for(std::chrono::seconds(1));
        progress += 10;
    }
    if (progress == 100) {
        std::cout << "Task completed successfully." << std::endl;
    } else {
        std::cout << "Task stopped prematurely." << std::endl;
    }
}

int main() {
    std::vector<std::jthread> threads;

    for (int i = 0; i < 5; ++i) {
        threads.emplace_back(long_running_task);
    }

    // 让主线程休眠一段时间，以便子线程开始工作
    std::this_thread::sleep_for(std::chrono::seconds(3));

    // 请求停止所有线程
    for (auto& thread : threads) {
        if (thread.joinable()) {
            thread.request_stop();
        }
    }

    // 等待所有线程完成
    for (auto& thread : threads) {
        if (thread.joinable()) {
            thread.join();
        }
    }

    return 0;
}

在这个示例中，long_running_task函数模拟了一个长时间运行的任务。每隔一秒钟，任务会检查一次停止信号，并根据进度更新状态。这样可以确保即使任务还在运行，线程也能在接收到停止信号后及时退出。