AsyncReadExt::read_u64是否支持取消操作安全?
创始人
2024-09-21 08:30:25
0

AsyncReadExt::read_u64函数并不直接支持取消操作,但可以通过使用tokio::select!宏和一个额外的取消通道来实现取消。具体的实现方法如下:

use tokio::io::AsyncReadExt;
use tokio::sync::mpsc::{channel, Sender};
use std::io::Error;

async fn read_with_cancel(mut reader: R, cancel: Sender<()>) -> Result
where
    R: AsyncReadExt + Unpin,
{
    let mut buf = [0; 8];
    let mut n = 0usize;
    
    tokio::select! {
        res = reader.read(&mut buf) => {
            let mut bytes_read = res?;
            n += bytes_read;
            
            while bytes_read > 0 && n < 8 {
                bytes_read = reader.read(&mut buf[n..]).await?;
                n += bytes_read;
            }
        }
        _ = cancel.recv() => {
            return Err(Error::new(std::io::ErrorKind::Interrupted, "cancelled"));
        }
    }
    
    Ok(u64::from_be_bytes(buf))
}

这个函数将读取操作包含在了tokio::select!宏中,并在取消通道上等待。如果收到了一个取消请求,函数将立即返回带有一个Error的Result。否则,函数将连续尝试读取充分的字节数,之后将这些字节解析成一个u64值并返回。

相关内容

热门资讯

5分钟辅助!新玄龙辅助(辅助挂... 5分钟辅助!新玄龙辅助(辅助挂)原来确实有挂(必备教程开挂辅助平台)新玄龙辅助ai黑科技系统规律教程...
第三分钟辅助!有没有哈糖大菠萝... 第三分钟辅助!有没有哈糖大菠萝攻略推荐(辅助挂)一贯确实有挂(德州论坛开挂辅助平台);亲,有没有哈糖...
1分钟详情!广东老友辅助(辅助... 1分钟详情!广东老友辅助(辅助挂)一贯确实有挂(AI教程开挂辅助下载)【无需打开直接搜索加薇1367...
第二分钟明白!海盗来了大白辅助... 第二分钟明白!海盗来了大白辅助(辅助挂)原来确实有挂(详细教程开挂辅助安装);亲,海盗来了大白辅助这...
6分钟讲解!战皇大厅辅助那个可... 6分钟讲解!战皇大厅辅助那个可靠(辅助挂)一直真的有挂(透视教程开挂辅助脚本)《详细加薇136704...
一分钟讲究!欢欢辅助软件怎么样... 一分钟讲究!欢欢辅助软件怎么样(辅助挂)果然是有挂的(黑科技教程开挂辅助软件)【无需打开直接搜索加薇...
8分钟辅助!wepokerpl... 8分钟辅助!wepokerplus辅助辅助(辅助挂)果然确实有挂(详细教程开挂辅助平台) >>您好:...
9分钟讲究!大咖娱乐挂脚本(辅... 9分钟讲究!大咖娱乐挂脚本(辅助挂)果然真的有挂(曝光教程开挂辅助工具)您好:大咖娱乐挂脚本这款游戏...
第2分钟辅助!小闲川南辅助(辅... 第2分钟辅助!小闲川南辅助(辅助挂)原来确实有挂(扑克教程!开挂辅助下载);无需打开直接搜索薇:13...
第七分钟发现!蜀山四川免费辅助... 第七分钟发现!蜀山四川免费辅助软件下载(辅助挂)一贯有挂(安装教程开挂辅助软件)>>您好:软件加薇1...