C++stop_token检查停止请求

2026-04-02 00:40:24 327阅读 0评论

C++中的`std::stop_token`：优雅地处理异步任务取消

在现代软件开发中，异步编程已经成为一种不可或缺的技术。然而，如何优雅地处理异步任务的取消是一个复杂的问题。C++20引入了std::stop_token，这是一个强大的工具，可以帮助我们实现这一目标。

引言

想象一下，你正在编写一个多线程程序，其中包含多个异步任务。如果用户希望提前终止这些任务，传统的做法可能是设置一个标志位，然后在每个任务中定期检查这个标志位。这种方法虽然简单，但存在一些问题：

代码耦合度高：每个任务都需要检查标志位，这会导致代码耦合度增加。
难以扩展：如果需要添加更多的异步任务，需要在每个任务中重复相同的逻辑。
效率低下：频繁地检查标志位会影响性能。

std::stop_token正是为了解决这些问题而设计的。它提供了一种更优雅的方式来处理异步任务的取消。

`std::stop_token`的基本概念

std::stop_token是C++20引入的一个新特性，用于在异步任务中传递取消信号。它的主要用途是在多线程环境中安全地取消任务。

创建和使用`std::stop_token`

要使用std::stop_token，你需要创建一个std::jthread对象，因为std::jthread会自动管理任务的生命周期并支持取消操作。

#include <iostream>
#include <thread>
#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 << "Task canceled!" << std::endl;
}

int main() {
    std::jthread t(worker);
    std::this_thread::sleep_for(std::chrono::seconds(5));
    t.request_stop();
    t.join();
    return 0;
}

在这个例子中，worker函数接受一个std::stop_token参数。在循环中，它会定期检查stoken.stop_requested()是否返回true。如果返回true，表示任务被取消，循环将退出。

请求取消

你可以通过调用std::jthread的request_stop()方法来请求取消任务。这个方法不会立即终止任务，而是设置一个取消标志，任务会在下一次检查时响应这个标志。

t.request_stop();

等待任务完成

为了确保任务完全终止后再继续执行主线程，你可以调用t.join()方法。

t.join();

处理取消请求的策略

在实际应用中，处理取消请求的策略非常重要。以下是一些常见的策略：

1. 清理资源

当任务被取消时，通常需要清理一些资源，例如关闭文件、释放内存等。

void worker(std::stop_token stoken) {
    try {
        while (!stoken.stop_requested()) {
            // 执行任务
            std::cout << "Working..." << std::endl;
            std::this_thread::sleep_for(std::chrono::seconds(1));
        }
    } catch (...) {
        // 处理异常
    } finally {
        // 清理资源
        cleanup();
    }
}

2. 中断当前操作

如果你的任务正在进行某些耗时的操作（如I/O操作），可以尝试中断这些操作。

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 << "Task canceled!" << std::endl;

    // 中断当前操作
    interrupt_current_operation();
}

3. 使用`std::future`和`std::promise`

有时候，你可以使用std::future和std::promise来更好地控制任务的取消。

#include <future>

std::promise<void> cancel_promise;
std::future<void> cancel_future = cancel_promise.get_future();

void worker(std::stop_token stoken) {
    while (!stoken.stop_requested() && !cancel_future.wait_for(std::chrono::milliseconds(0)).ready()) {
        // 执行任务
        std::cout << "Working..." << std::endl;
        std::this_thread::sleep_for(std::chrono::seconds(1));
    }
    std::cout << "Task canceled!" << std::endl;
}

int main() {
    std::jthread t(worker);
    std::this_thread::sleep_for(std::chrono::seconds(5));
    cancel_promise.set_value();
    t.join();
    return 0;
}

在这个例子中，cancel_future用于等待取消信号。如果取消信号到达，任务将立即退出。