当前位置：首页 > 科技 > 软件

Rust Tokio取消任务的几种模式，你知道吗？

来源：责编：时间：2024-05-20 17:53:30 235观看

导读Rust提供了对异步编程的支持，它可以生成异步任务，然后通过运行时执行器在操作系统线程之间调度执行。与Rust中的所有东西一样，异步编程必须是内存安全的，因此需要确保借用检查器可以编译通过。这篇文章是关于任务取消模式

Rust提供了对异步编程的支持，它可以生成异步任务，然后通过运行时执行器在操作系统线程之间调度执行。

与Rust中的所有东西一样，异步编程必须是内存安全的，因此需要确保借用检查器可以编译通过。

这篇文章是关于任务取消模式的，下面我们来介绍Tokio任务的取消模式。

Select 和 Channels

所有这些模式的核心是两个tokio特性：

channel：用于任务间通信
select：用于等待多个异步计算(不一定是任务!)

Tokio channel看起来有点复杂，但同时就程序的内存安全和弹性而言，它很强大。Tokio channel创建了两个不同的对象，用于任务之间的通信，不能同时使用一个通道对象来接收和发送。

Tokio提供的频道实际上有四种：

mpsc：多个生产者，单一消费者
oneshot：用于发送和接收单个值，发送后，通道关闭。
broadcast：多个发送者，多个消费者
watch：单一生产者，多个消费者

Drop JoinHandle不会取消任务

JoinHandle在删除关联的任务时将其分离，这意味着不再有任何任务句柄，也没有办法对其进行连接。

每次在tokio中生成任务时，都会返回JoinHandle。可以使用join句柄来等待任务完成，但是认为可以使用它来简单地通过删除任务来强制终止任务是错误的。这里有一个愚蠢的例子：

use tokio::time::{self, Duration};#[tokio::main]async fn main() {    let handle = tokio::spawn(async {        // do some work        tokio::time::sleep(Duration::from_secs(10)).await;        println!("Task completed");    });    // 100毫秒后取消任务    time::sleep(Duration::from_millis(100)).await;    drop(handle);    println!("Task was cancelled");}

丢弃句柄并不会取消正在运行的任务！

Abort任务

这是取消任务的最极端的方式，没有清理的空间：

use tokio::time::{self, Duration};#[tokio::main]async fn main() {    let handle = tokio::spawn(async {        // do some work        tokio::time::sleep(Duration::from_secs(1)).await;        println!("Task completed");    });    // 100毫秒后取消任务    time::sleep(Duration::from_millis(100)).await;    handle.abort();    time::sleep(Duration::from_secs(2)).await;    println!("Task was cancelled");}

使用oneshot channel

oneshot channel允许通道上的发送单个值，可以由多个接收器侦听。与drop模式不同，此模式允许通道执行一些清理工作。这里有一个例子：

use tokio::sync::oneshot;use tokio::time::Duration;#[tokio::main]async fn main() {    let (tx, rx) = oneshot::channel();    let task = tokio::spawn(async move {        tokio::select! {            _ = rx => {                println!("Task is cancelling...");            }            _ = tokio::time::sleep(Duration::from_secs(10)) => {                println!("Task completed normally");            }        }        println!("Task is cleaning up");    });    tokio::time::sleep(Duration::from_millis(100)).await;    // 发送取消信号    let _ = tx.send(());    // 等待任务完成    let _ = task.await;}

运行结果如下：

Task is cancelling...Task is cleaning up

oneshot channel的限制是你不能用它来取消多个任务。

使用broadcast channel取消多个任务

如果要取消多个任务，可以使用broad channel。可以有多个生产者向通道发送信息，也可以有多个消费者从通道接收信息。每个接收方都可以看到在通道上发送的每个值。

这里有一个简单的例子，来演示如何使用它来取消多个任务：

use tokio::sync::broadcast;use tokio::time::Duration;#[tokio::main]async fn main() {    let (tx, mut rx1) = broadcast::channel(1);    let mut rx2 = tx.subscribe();    let task1 = tokio::spawn(async move {        tokio::select! {            _ = rx1.recv() => {                println!("Task 1 is cancelling...");            }            _ = tokio::time::sleep(Duration::from_secs(10)) => {                println!("Task 1 completed normally");            }        }        println!("Task 1 is cleaning up");    });    let task2 = tokio::spawn(async move {        tokio::select! {            _ = rx2.recv() => {                println!("Task 2 is cancelling...");            }            _ = tokio::time::sleep(Duration::from_secs(10)) => {                println!("Task 2 completed normally");            }        }        println!("Task 2 is cleaning up");    });    tokio::time::sleep(Duration::from_millis(100)).await;    // 发送取消信号    let _ = tx.send(());    // 等待任务完成    let _ = tokio::join!(task1, task2);}

运行结果如下：

Task 2 is cancelling...Task 2 is cleaning upTask 1 is cancelling...Task 1 is cleaning up

取消的顺序可能会有所不同，因为任务可能会以不同的顺序取消！

如果只想从单个任务向多个任务发送取消信号，那么broad channel可能有点过度，因为它提供了在多个任务之间传递消息的所有机制。

如果既需要消息传递又需要消息取消，这很方便。但如果只需要消息取消，还有更好的方法，开销更少：watch channel。

使用watch channel取消多个任务

watch channel是多个消费者频道的单一生产者。watch channel给了任务清理自己的机会。缺点是，消费者只能看到通道上发送的最近的值——这意味着如果任务在通道上发送了一个值之后启动，它可能会错过它，因此不会被取消，所以要小心这一点。这里有一个简单的例子：

use tokio::sync::watch;use tokio::time::Duration;#[tokio::main]async fn main() {    let (tx, mut rx1) = watch::channel(false);    let mut rx2 = tx.subscribe();    let task1 = tokio::spawn(async move {        loop {            tokio::select! {                _ = rx1.changed() => {                    if *rx1.borrow() {                        println!("Task 1 is cancelling...");                        break;                    }                }                _ = tokio::time::sleep(Duration::from_secs(10)) => {                    println!("Task 1 completed normally");                    break;                }            }        }        println!("Task 1 is cleaning up");    });    let task2 = tokio::spawn(async move {        loop {            tokio::select! {                _ = rx2.changed() => {                    if *rx2.borrow() {                        println!("Task 2 is cancelling...");                        break;                    }                }                _ = tokio::time::sleep(Duration::from_secs(10)) => {                    println!("Task 2 completed normally");                    break;                }            }        }        println!("Task 2 is cleaning up");    });    tokio::time::sleep(Duration::from_millis(100)).await;    // 发送取消信号    let _ = tx.send(true);    // 等待任务完成    let _ = tokio::join!(task1, task2);}

取消令牌

官方的Tokio文档中列出了一种名为CancellationToken的东西，用于优雅关机。这在tokio crate本身中不可用，但在相关的toko_util crate中可用。

use tokio::time::{sleep, Duration};use tokio_util::sync::CancellationToken;#[tokio::main]async fn main() {    // Create a CancellationToken    let token = CancellationToken::new();    let token1 = token.clone();    let token2 = token.clone();    let task1 = tokio::spawn(async move {        loop {            tokio::select! {                _ = token1.cancelled() => {                        println!("Task 1 is cancelling...");                        break;                }                _ = tokio::time::sleep(Duration::from_secs(10)) => {                    println!("Task 1 completed normally");                    break;                }            }        }        println!("Task 1 is cleaning up");    });    let task2 = tokio::spawn(async move {        loop {            tokio::select! {                _ = token2.cancelled() => {                        println!("Task 2 is cancelling...");                        break;                }                _ = tokio::time::sleep(Duration::from_secs(10)) => {                    println!("Task 2 completed normally");                    break;                }            }        }        println!("Task 2 is cleaning up");    });    sleep(Duration::from_millis(100)).await;    // 发送取消信号    token.cancel();    // 等待任务完成    let _ = tokio::join!(task1, task2);}

请注意我们是如何克隆令牌的，以便将其移动到各个异步任务中。

本文链接：http://www.28at.com/showinfo-26-89399-0.htmlRust Tokio取消任务的几种模式，你知道吗？

声明：本网页内容旨在传播知识，若有侵权等问题请及时与本网联系，我们将在第一时间删除处理。邮件：2376512515@qq.com

上一篇：基于Jenkins Pipeline构建企业级CI/CD

下一篇：小米面试：如何实现优先级线程池？

标签：

热门焦点

Find N3入网：最高支持16+1TB

OPPO将于近期登场的Find N3折叠屏目前已经正式入网，型号为PHN110。本次Find N3在外观方面相比前两代有很大的变化，不再是小号的横向折叠屏，而是跟别的厂商一样采用了较为常见的
卢伟冰长文解析K60至尊版对Redmi有着里程碑式的意义

在今天的Redmi后性能时代战略发布会结束之后，Redmi总经理卢伟冰又带来了一篇长文，详解了为什么 Redmi 要开启后性能时代？为什么选择和 MediaTek、Pixelworks 深度合作？以及后性
6月iOS设备好评榜：第一蝉联榜首近一年

作为安兔兔各种榜单里变化最小的那个，2023年6月的iOS好评榜和上个月相比没有任何排名上的变化，仅仅是部分设备好评率的下降，长年累月的用户评价和逐渐退出市场的老款机器让这
印度登月最关键一步！月船三号今晚进入环月轨道

8月5日消息，据印度官方消息，月船三号将于北京时间今晚21时30分左右开始近月制动进入环月轨道。这是该探测器能够成功的最关键步骤之一，如果成功将开始围
梁柱接棒两年，腾讯音乐闯出新路子

文丨田静出品丨牛刀财经（niudaocaijing）7月5日，企鹅FM发布官方公告称由于业务调整，将于9月6日正式停止运营，这意味着腾讯音乐长音频业务走向消亡。腾讯在长音频领域还在摸索。为
当家的盒马，加速谋生

来源 | 价值星球Planet作者 | 归去来自己“当家”的盒马，开始加速谋生了。据盒马官微消息，盒马计划今年开放生鲜供应链，将其生鲜商品送往食堂。目前，盒马在上海已经与
华为Mate60标准版细节曝光：经典星环相机模组回归

这段时间以来，关于华为新旗舰的爆料日渐密集。据此前多方爆料，今年华为将开始恢复一年双旗舰战略，除上半年推出的P60系列外，往年下半年的Mate系列也将
OPPO K11样张首曝：千元机影像“卷”得真不错！

一直以来，OPPO K系列机型都保持着较为均衡的产品体验，历来都是2K价位的明星机型，去年推出的OPPO K10和OPPO K10 Pro两款机型凭借各自的出色配置，堪称有
利用职权私自解除被封帐号 Meta开除20多名员工

11月18日消息，据外媒援引知情人士表示，过去一年时间内，Facebook母公司Meta解雇或处罚了20多名员工以及合同工，指控这些人通过内部系统以不当方式重置用户帐号，其

Rust Tokio取消任务的几种模式，你知道吗？

Select 和 Channels

Drop JoinHandle不会取消任务

Abort任务

使用oneshot channel

使用broadcast channel取消多个任务

使用watch channel取消多个任务

取消令牌

Find N3入网：最高支持16+1TB

卢伟冰长文解析K60至尊版对Redmi有着里程碑式的意义

6月iOS设备好评榜：第一蝉联榜首近一年

印度登月最关键一步！月船三号今晚进入环月轨道

梁柱接棒两年，腾讯音乐闯出新路子

当家的盒马，加速谋生

华为Mate60标准版细节曝光：经典星环相机模组回归

OPPO K11样张首曝：千元机影像“卷”得真不错！

利用职权私自解除被封帐号 Meta开除20多名员工

最新推荐

猜你喜欢

热门推荐

相关资讯