rust学习～tokio的io

await

Suspend execution until the result of a Future is ready.
暂停执行，直到一个 Future 的结果就绪。

.awaiting a future will suspend the current function’s execution until the executor has run the future to completion.
对一个 Future 使用 .await 操作会暂停当前函数的执行，直到执行器（executor）将该 Future 运行至完成

Read the async book for details on how async/await and executors work.
有关异步 / 等待（async/await）和执行器的工作原理的详细信息，请阅读《异步编程指南》（Async Book）。

Editions
await is a keyword from the 2018 edition onwards.
await 是从 2018 版及后续版本开始引入的关键字。

It is available for use in stable Rust from version 1.39 onwards.
从 1.39 版本及以后的稳定版 Rust 中可以使用它。

AsyncReadExt

use tokio::io::{self, AsyncReadExt};
use tokio::fs::File;#[tokio::main]
async fn main() -> io::Result<()> {let mut f = File::open("foo.txt").await?;let mut buffer = Vec::new();// 读取整个文件的内容f.read_to_end(&mut buffer).await?;// String::from_utf8 是 String 类型的一个关联函数// 专门用于把 Vec<u8> 类型的字节向量转换为 String 类型的 UTF - 8 字符串// 它会检查字节向量中的字节序列是否符合 UTF - 8 编码规则// 如果符合则返回一个 Ok(String)，若不符合则返回 Err(FromUtf8Error)// 适用于从字节数据（如文件读取、网络接收等）构建字符串，并且需要确保数据是有效的 UTF - 8 编码match String::from_utf8(buffer) {Ok(content) => {println!("文件内容如下：\n{}", content);}Err(e) => {eprintln!("将文件内容转换为字符串时出错: {}", e);}}Ok(())
}

use tokio::io::{self, AsyncReadExt};
use tokio::fs::File;#[tokio::main]
async fn main() -> io::Result<()> {let mut f = File::open("foo.txt").await?;let mut buffer = Vec::new();// 读取整个文件的内容f.read_to_end(&mut buffer).await?;Ok(())
}

为什么说 字节数组 &[u8] 实现了 AsyncRead ？

字节数组切片 &[u8] 实现了 AsyncRead 特征，这意味着它可以作为异步读取操作的数据源，允许以异步的方式从字节数组中读取数据

AsyncRead 特征的作用

AsyncRead 是 tokio 异步运行时库中定义的一个特征，它定义了异步读取操作的接口。其核心方法是 poll_read，该方法用于尝试从数据源中异步读取数据到指定的缓冲区。通过实现 AsyncRead 特征，类型可以参与到异步 I/O 操作中，利用异步运行时的调度机制，在等待数据可读时让出控制权，提高程序的并发性能。

AsyncRead 特征的简化定义如下：

use std::pin::Pin;
use std::task::{Context, Poll};
use tokio::io::Result;pub trait AsyncRead {fn poll_read(self: Pin<&mut Self>,cx: &mut Context<'_>,buf: &mut [u8],) -> Poll<Result<usize>>;
}

self 是实现该特征的类型的可变引用，使用 Pin 确保在异步操作过程中对象的内存位置不会改变
cx 是任务上下文，包含了任务的唤醒器等信息，用于在数据准备好时唤醒任务
buf 用于存储读取数据的缓冲区
Poll 枚举表示操作的结果，可能是 Poll::Ready 表示操作已完成，返回实际读取的字节数；也可能是 Poll::Pending 表示操作还未完成，需要等待。

&[u8] 实现 AsyncRead 的原因

灵活性和通用性：字节数组切片 &[u8] 是一种非常常见且灵活的数据表示方式，它可以表示内存中的一段连续字节数据。实现 AsyncRead 特征后，&[u8] 可以作为异步读取操作的数据源，使得很多使用 AsyncRead 的代码可以直接处理字节数组，无需额外的转换。
例如，在测试代码中，可以使用字节数组模拟文件或网络数据进行异步读取测试。
异步编程的一致性：在异步编程中，希望不同的数据源（如文件、网络套接字、内存缓冲区等）都能以统一的方式进行异步读取操作。通过让 &[u8] 实现 AsyncRead 特征，保持了异步读取操作的一致性，使得代码更加简洁和易于维护。

示例代码

use tokio::io::{self, AsyncReadExt};#[tokio::main]
async fn main() -> io::Result<()> {// 定义一个字节数组，存储了字符串 "Hello, World!" 的 UTF - 8 编码字节数据let data = b"Hello, World!";let mut buffer = [0; 5];// 创建一个字节数组切片 &[u8]，作为异步读取的数据源let mut reader = &data[..];// 调用 read 方法（该方法是基于 AsyncRead 特征实现的），异步地从字节数组切片中读取数据到 buffer 中// await 关键字用于等待读取操作完成，最终返回实际读取的字节数let n = reader.read(&mut buffer).await?;println!("Read {} bytes: {:?}", n, &buffer[..n]);Ok(())
}

实际应用场景

测试：在编写异步 I/O 代码的单元测试时，可以使用字节数组模拟不同的数据源，方便进行测试。例如，测试一个异步数据解析器时，可以使用字节数组提供测试数据。
内存数据处理：当需要对内存中的字节数据进行异步处理时，如对加密数据、压缩数据等进行异步解密或解压缩操作，可以直接使用字节数组作为数据源。

字节数组切片 &[u8] 实现 AsyncRead 特征，为异步编程提供了更多的灵活性和一致性，使得字节数组可以方便地作为异步读取操作的数据源。

AsyncWriteExt

use tokio::io::{self, AsyncWriteExt};
use tokio::fs::File;#[tokio::main]
async fn main() -> io::Result<()> {let mut file = File::create("foo.txt").await?;// 将一个 &str 字符串转变成一个字节数组：&[u8;10]// 然后 write 方法又会将这个 &[u8;10] 的数组类型隐式强转为数组切片: &[u8]let n = file.write(b"some bytes").await?;println!("Wrote the first {} bytes of 'some bytes'.", n);Ok(())
}

AsyncWriteExt::write_all 将缓冲区的内容全部写入到写入器中

use tokio::io::{self, AsyncWriteExt};
use tokio::fs::File;#[tokio::main]
async fn main() -> io::Result<()> {let mut file = File::create("foo.txt").await?;file.write_all(b"some bytes").await?;Ok(())
}

tokio::io处理标准的输入/输出/错误

use tokio::fs::File;
use tokio::io;#[tokio::main]
async fn main() -> io::Result<()> {// &[u8] 是字节数组切片let mut reader: &[u8] = b"hello";let mut file = File::create("foo.txt").await?;// 异步的将读取器( reader )中的内容拷贝到写入器( writer )中// 字节数组 &[u8] 实现了 AsyncRead，所以这里可直接使用 readerio::copy(&mut reader, &mut file).await?;Ok(())
}

tokio::io分离读写器

错误写法

io::copy(&mut socket, &mut socket).await

读取器和写入器都是同一个 socket，因此需要对其进行两次可变借用，这明显违背了 Rust 的借用规则

用同一个 socket 是不行的，为了实现目标功能，必须将 socket 分离成一个读取器和写入器
任何一个读写器( reader + writer )都可以使用 io::split 方法进行分离，最终返回一个读取器和写入器，两者可以独自使用，例如可以放入不同的任务中。

回声服务端

use tokio::io;
use tokio::net::TcpListener;#[tokio::main]
async fn main() -> io::Result<()> {let listener = TcpListener::bind("127.0.0.1:6142").await?;loop {let (mut socket, _) = listener.accept().await?;tokio::spawn(async move {// split 操作和 io::copy 调用是在同一个异步任务上下文中执行的// 由于它们处于同一个任务中，所以不存在不同任务之间的数据传递开销和同步问题// 任务的执行是连贯的，避免了因为跨任务操作而引入的额外复杂性和性能损耗let (mut rd, mut wr) = socket.split();if io::copy(&mut rd, &mut wr).await.is_err() {eprintln!("failed to copy");}});}
}

回声客户端

use tokio::io::{self, AsyncReadExt, AsyncWriteExt};
use tokio::net::TcpStream;#[tokio::main]
async fn main() -> io::Result<()> {let socket = TcpStream::connect("127.0.0.1:6142").await?;let (mut rd, mut wr) = io::split(socket);// 创建异步任务，在后台写入数据tokio::spawn(async move {wr.write_all(b"hello\r\n").await?;wr.write_all(b"world\r\n").await?;// 有时，我们需要给予 Rust 一些类型暗示，它才能正确的推导出类型Ok::<_, io::Error>(())});let mut buf = vec![0; 128];loop {let n = rd.read(&mut buf).await?;if n == 0 {break;}println!("GOT {:?}", &buf[..n]);}Ok(())
}

tokio::spawn 是 Tokio 运行时提供的一个函数，用于创建一个新的异步任务并将其放入任务队列中等待执行。这个新任务会在 Tokio 运行时的调度下异步执行，与当前代码所在的任务是并发关系，而不是顺序执行关系。
当执行到 tokio::spawn 时，它会立即将传入的异步闭包包装成一个新的异步任务并放入 Tokio 运行时的任务队列中，然后代码会继续往下执行，不会等待这个新任务开始执行。
因此，let mut buf = vec![0; 128]; 这行代码会紧接着 tokio::spawn 之后执行，而 tokio::spawn 内部的异步任务会在 Tokio 运行时调度到它时才开始执行。
Tokio 运行时的调度是基于事件驱动和任务优先级的，它会根据系统资源和任务的状态动态地决定哪个任务先执行。所以，tokio::spawn 内部的任务可能在 let mut buf = vec![0; 128]; 之前执行，也可能在之后执行，甚至可能与后续代码并发执行
假设 tokio::spawn 内部的任务执行需要一些时间（例如网络延迟），而创建 buf 向量的操作很快，那么很可能 let mut buf = vec![0; 128]; 会先执行，然后才轮到 tokio::spawn 内部的任务开始执行

C-S修正

cargo run --bin server.rs
cargo run --bin client.rs

上述代码跑起来之后，服务端不退出的话，客户端会一直卡住，客户端加如下函数即可解决

wr.shutdown().await ? ;

手动实现io

use tokio::io::{self, AsyncReadExt, AsyncWriteExt};
use tokio::net::TcpListener;#[tokio::main]
async fn main() -> io::Result<()> {let listener = TcpListener::bind("127.0.0.1:6142").await?;loop {let (mut socket, _) = listener.accept().await?;tokio::spawn(async move {// 此处的缓冲区是一个 Vec 动态数组，它的数据是存储在堆上，而不是栈上// 若改成 let mut buf = [0; 1024];，则存储在栈上// 一个数据如果想在 .await 调用过程中存在，那它必须存储在当前任务内// buf 会在 .await 调用过程中被使用，因此它必须要存储在任务内let mut buf = vec![0; 1024];loop {match socket.read(&mut buf).await {// 返回值 `Ok(0)` 说明对端已经关闭Ok(0) => return,Ok(n) => {// Copy the data back to socket// 将数据拷贝回 socket 中if socket.write_all(&buf[..n]).await.is_err() {// 非预期错误，由于我们这里无需再做什么，因此直接停止处理return;}}Err(_) => {// 非预期错误，由于我们无需再做什么，因此直接停止处理return;}}}});}
}

若该缓冲区数组创建在栈上，那每条连接所对应的任务的内部数据结构看上去可能如下所示

struct Task {task: enum {AwaitingRead {socket: TcpStream,buf: [BufferType],},AwaitingWriteAll {socket: TcpStream,buf: [BufferType],}}
}

栈数组要被使用，就必须存储在相应的结构体内，其中两个结构体分别持有了不同的栈数组 [BufferType]
这种方式会导致任务结构变得很大
一般选择缓冲区长度往往会使用分页长度(page size)，因此使用栈数组会导致任务的内存大小变得很奇怪甚至糟糕：
$page-size + 一些额外的字节

编译器会进一步优化 async 语句块的布局，而不是像上面一样简单的使用 enum
在实践中，变量也不会在枚举成员间移动。但是再怎么优化，任务的结构体至少也会跟其中的栈数组一样大
因此通常情况下，使用堆上的缓冲区会高效实用的多

当任务因为调度在线程间移动时，存储在栈上的数据需要进行保存和恢复，过大的栈上变量会带来不小的数据拷贝开销
因此，存储大量数据的变量最好放到堆上

处理 EOF

当 TCP 连接的读取端关闭后，再调用 read 方法会返回 Ok(0)。此时，再继续下去已经没有意义，因此需要退出循环。
忘记在 EOF 时退出读取循环，是网络编程中一个常见的 bug :

loop {match socket.read(&mut buf).await {Ok(0) => return,// ... 其余错误处理}
}

一旦读取端关闭后，那后面的 read 调用就会立即返回 Ok(0)，而不会阻塞等待，因此这种无阻塞循环会最终导致 CPU 立刻跑到 100%，并将一直持续下去，直到程序关闭。

小节

实际上，io::split 可以用于任何同时实现了 AsyncRead 和 AsyncWrite 的值，它的内部使用了 Arc 和 Mutex 来实现相应的功能。如果大家觉得这种实现有些重，可以使用 Tokio 提供的 TcpStream，它提供了两种方式进行分离:

TcpStream::split 会获取字节流的引用，然后将其分离成一个读取器和写入器。但由于使用了引用的方式，它们俩必须和 split 在同一个任务中。 优点就是，这种实现没有性能开销，因为无需 Arc 和 Mutex。
TcpStream::into_split 还提供了一种分离实现，分离出来的结果可以在任务间移动，内部是通过 Arc 实现

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.mzph.cn/bicheng/72178.shtml

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

腾讯2025年软件测试面试题

腾讯2025年软件测试面试题

以下是基于腾讯等一线互联网公司软件测试岗位的面试趋势和技术要求，025年出现的软件测试面试题。这些问题涵盖了基础知识、自动化测试、性能测试、安全测试、编程能力等多个方面，供参考和准备。 一、基础知识 软件测试的基本概念
阅读更多...
数据结构（陈越，何钦铭） 第四讲 树（中）

数据结构（陈越，何钦铭） 第四讲 树（中）

4.1 二叉搜索树 4.1.1 二叉搜索树及查找 Position Find(ElementTyoe X,BinTree BST){if(!BST){return NULL;}if(X>BST->Data){return Find(X,BST->Right)}else if(X<BST->Data){return Find(X,BST->Left)}else{return BST;} } Position IterFind(ElementTyp…
阅读更多...
GEE学习笔记 28：基于Google Earth Engine的Landsat8缨帽变换土壤指数反演——亮度、绿度与湿度分量的提取

GEE学习笔记 28：基于Google Earth Engine的Landsat8缨帽变换土壤指数反演——亮度、绿度与湿度分量的提取

1.缨帽变换介绍 缨帽变换（Tasseled Cap Transformation，TCT），也称为缨帽特征空间或缨帽系数，是一种用于遥感图像分析的线性变换方法。它最初由美国农业部的研究人员E. Kauth和G. Thomas在1976年提出，用于增强陆地卫星（Landsat）图像中的特定地表特征，如植被、土壤和城市…
阅读更多...
【现代Web布局与动画技术：卡片组件实战分享】

【现代Web布局与动画技术：卡片组件实战分享】

&#x1f4f1; 现代Web布局与动画技术&#xff1a;卡片组件实战分享 &#x1f680; 引言 &#x1f31f; 在过去的开发过程中&#xff0c;我们共同实现了一个功能丰富的卡片组件&#xff0c;它不仅美观&#xff0c;还具有交互性和响应式设计。这篇文章将分享这个组件背后的技术…
阅读更多...
学习路之PHP --TP6异步执行功能 （无需安装任何框架）

学习路之PHP --TP6异步执行功能 （无需安装任何框架）

学习路之PHP --异步执行功能 &#xff08;无需安装任何框架&#xff09; 简介一、工具类二、调用三、异步任务的操作四、效果&#xff1a; 简介 执行异步任务是一种很常见的需求&#xff0c;如批量发邮箱&#xff0c;短信等等执行耗时任务时&#xff0c;需要程序异步执行&…
阅读更多...
STM32之影子寄存器

STM32之影子寄存器

预分频寄存器计数到一半的时候&#xff0c;改变预分频值&#xff0c;此时不会立即生效&#xff0c;会等到计数完成&#xff0c;再从影子寄存器即预分频缓冲器里装载修改的预分频值。 如上图&#xff0c;第一行是内部时钟72M&#xff0c;第二行是时钟使能&#xff0c;高电平启动…
阅读更多...
Deepseek API接入IDE【VSCode Cline Cursor ChatBox Deepseek deepseek-reasoner】

Deepseek API接入IDE【VSCode Cline Cursor ChatBox Deepseek deepseek-reasoner】

本文解决以下疑难杂症: 使用deepseek的最新接模型接入ide 使用deepseek的最新接模型接入vscode 使用deepseek的最新接模型接入vscode中的Cline 使用deepseek的最新接模型接入Cline 使用deepseek的最新接模型接入ChatBox 使用cursor接入Deepseek官方的的deepseek-reasoner…
阅读更多...
微信小程序读取写入NFC文本，以及NFC直接启动小程序指定页面

微信小程序读取写入NFC文本，以及NFC直接启动小程序指定页面

一、微信小程序读取NFC文本(yyy优译小程序实现)，网上有很多通过wx.getNFCAdapter方法来监听读取ＮＦＣ卡信息，但怎么处理读取的message文本比较难找，现用下面方法来实现，同时还解决几个问题，1、在回调方法中this.setData不更新信息，因为this的指向问题，2、在退出页面时，…
阅读更多...
在Linux桌面上创建Idea启动快捷方式

在Linux桌面上创建Idea启动快捷方式

1、在桌面新建idea.desktop vim idea.desktop [Desktop Entry] EncodingUTF-8 NameIntelliJ IDEA CommentIntelliJ IDEA Exec/home/software/idea-2021/bin/idea.sh Icon/home/software/idea-2021/bin/idea.svg Terminalfalse TypeApplication CategoriesApplication;Developm…
阅读更多...
VUE2生命周期页面加载顺序

VUE2生命周期页面加载顺序

使用 Vue CLI 4.5 运行 vue create myvue 创建项目&#xff0c;并通过 npm run serve 运行后&#xff0c;会生成一个标准的 Vue 项目目录结构。以下是生成目录的详细说明&#xff0c;以及运行 localhost:8080 后 Vue 页面的加载顺序。 1. 生成目录结构 运行 vue create myvue …
阅读更多...
SV基础（一）：System Verilog与Verilog核心区别详解

SV基础（一）：System Verilog与Verilog核心区别详解

文章目录 **1. 设计增强功能****数据类型扩展****接口（Interface）****2. 验证功能增强****断言（Assertions）****约束随机测试****功能覆盖率****3. 面向对象编程（OOP）****4. 测试平台（Testbench）改进****5. 语法简化****6. 其他关键区别****学习建议**System Verilog 是…
阅读更多...
如何用 Python 进行机器学习

如何用 Python 进行机器学习

文章目录 前言1. 环境准备Python安装选择Python开发环境安装必要库 2. 数据收集与加载3. 数据探索与可视化4. 数据预处理5. 模型选择与训练6. 模型评估7. 模型调优8. 模型部署 前言 使用 Python 进行机器学习一般可以按照以下步骤进行&#xff0c;下面将详细介绍每个步骤及对应…
阅读更多...
2021-05-27 C++找出矩阵数组中值最大的元素和它在数组中的位置

2021-05-27 C++找出矩阵数组中值最大的元素和它在数组中的位置

缘由各位大佬&#xff0c;这个应该怎么做_编程语言-CSDN问答 void 找出数组中值最大的元素和它在数组中的位置() {//缘由https://ask.csdn.net/questions/7436585?spm1005.2025.3001.5141int a[4][4], aa 0, aaa 0, d 0, x 0;while (aa < 4 && aaa < 4)std…
阅读更多...
在 IntelliJ IDEA 中启动多个注册到 Nacos 的服务

在 IntelliJ IDEA 中启动多个注册到 Nacos 的服务

使用场景&#xff1a;边改代码&#xff0c;边和前端联调。 在微服务架构中&#xff0c;服务注册与发现是核心功能之一。Nacos 作为一款流行的开源服务注册与配置管理工具&#xff0c;被广泛应用于微服务架构中。本文将介绍如何在 IntelliJ IDEA 中配置并启动多个注册到 Nacos …
阅读更多...
DeepSeek开源周Day2：DeepEP - 专为 MoE 模型设计的超高效 GPU 通信库

DeepSeek开源周Day2：DeepEP - 专为 MoE 模型设计的超高效 GPU 通信库

项目地址&#xff1a;https://github.com/deepseek-ai/DeepEP 开源日历&#xff1a;2025-02-24起 每日9AM(北京时间)更新&#xff0c;持续五天 (2/5)&#xff01; ​ ​ 引言 在大模型训练中&#xff0c;混合专家模型&#xff08;Mixture-of-Experts, MoE&#xff09;因其动…
阅读更多...
HTTP学习——————（四）TLS等加密算法

HTTP学习——————（四）TLS等加密算法

前文学习&#xff1a; 一、二、三 学习来源网站 &#xff1a; 极客时间 TLS 目的&#xff1a;身份验证、保密性、完整性 解决问题&#xff1a; Record记录协议——对称加密 Handshake握手协议———1.验证通讯双方身份 2.交换加解密安全套件 3.协商加密参数 有密钥交换算法…
阅读更多...
FastExcel vs EasyExcel vs Apache POI：三者的全面对比分析

FastExcel vs EasyExcel vs Apache POI：三者的全面对比分析

一、核心定位与历史沿革 Apache POI&#xff08;1990s-&#xff09; 作为Java生态中最古老的Excel处理库&#xff0c;提供对.xls/.xlsx文件的全功能支持。其核心价值在于对Excel规范的完整实现&#xff0c;包括单元格样式、公式计算、图表操作等深度功能。但存在内存消耗大&…
阅读更多...
辛格迪客户案例 | 鼎康生物电子合约系统(eSign)项目

辛格迪客户案例 | 鼎康生物电子合约系统(eSign)项目

01 案例企业 鼎康(武汉)生物医药有限公司于2013年06月19日成立 &#xff0c;是一家总部位于湖北武汉的CDMO公司&#xff0c;坚持以客户为中心&#xff0c;以及时、经济和高质量为服务导向。鼎康生物拥有先进的150,000平方英尺的生产厂房&#xff0c;生产设施位于中国武汉的Bio…
阅读更多...
multer 依赖详解

multer 依赖详解

multer 是一个用于处理 multipart/form-data 类型表单数据的 Node.js 中间件&#xff0c;主要用于文件上传。它基于 busboy 构建&#xff0c;使用起来非常方便。 一、安装 npm install multer 二、基本使用 const express require("express");const multer req…
阅读更多...
点云配准技术的演进与前沿探索：从传统算法到深度学习融合(4)

点云配准技术的演进与前沿探索：从传统算法到深度学习融合(4)

4、点云配准面临的挑战与应对策略 4.1 点云配准面临的主要挑战 在点云配准的实际应用中&#xff0c;尽管已经取得了显著的研究成果&#xff0c;但仍然面临着诸多复杂而严峻的挑战&#xff0c;这些挑战严重制约了点云配准技术在更多领域的广泛应用和深入发展。 在自动驾驶场景…
阅读更多...
最新文章

Copyright @ 2022~2023