Tokio强大的Rust异步框架

一、简介

异步是现在编写高性能应用的核心要素之一，很多语言都已经纷纷加入异步的特效。C#应该是最先做出响应的语言。连微软这位软件大佬都非常重视异步，可见异步编程重要性。

在Rust中同样支持异步编程，只不过一般我们不会使用Rust官方，而是使用第三方依赖Tokio。

二、Tokio

Tokio 是一个用于 Rust 编程语言的异步运行时，它提供了一种在单线程中管理多个并发任务的方式。它是基于 Rust 的异步编程模型构建的，这个模型利用了 Rust 的所有权（ownership）和生命周期（lifetimes）特性来保证内存安全。

Tokio 的核心是它的异步 I/O 模型，它允许程序在执行 I/O 操作（如网络请求或文件读写）时不需要阻塞线程。

如果你使用了Tokio，建议你还是使用tokio:io，而不是标准库std:io，这个看业务选择。

Tokio 还包括了一个事件循环（event loop），它负责调度和执行异步任务。这个事件循环可以处理成千上万的并发连接，所以Tokio非常适合用于构建高性能的网络应用，如 web 服务器、消息队列、实时通信系统等。

简单点说就是高性能中间件服务，例如Nginx这种服务基建。

同时，Tokio 还提供了许多实用的功能，如定时器、同步原语、TCP 和 UDP 网络协议等，以帮助开发者更轻松地构建复杂的网络应用。

三、基于Future实现

在Rust标准库中Future 是一个关键的概念，它是异步编程的基础。它代表了一个尚未完成的计算，可能在未来的某个时间点完成，并产生一个结果。

你可以完全类似的比作JavaScript中的Promise。

pub trait Future {type Output; // 关联类型，表示 Future 完成时产生的值类型fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output>;
}

这里的 Poll 是一个枚举，它表示 poll 方法的可能结果：

pub enum Poll<T> {Ready(T),    // 异步操作已经完成，并返回一个值Pending,     // 异步操作尚未完成
}

这个 Future 对象可以被传递给执行上下文（如 Tokio 的运行时），执行上下文会定期调用 poll 方法来检查操作是否完成。如果操作完成，poll 方法会返回 Poll::Ready(result)，其中 result 是操作的结果。如果操作尚未完成，poll 方法会返回 Poll::Pending，并且当操作有可能完成时，通知执行上下文重新调度 Future。

四、简单的示例

由于今天我们是初步认识Tokio，下面的例子都比较简单，但可以让你快速熟悉tokio的基本用法。

use tokio::time::Duration; // 注意这里使用的tokio提供的time，并非std#[tokio::main] //需要讲main函数设置为async并添加tokio::main注解async fn main() {println!("Hello before delay");tokio::time::sleep(Duration::from_secs(2)).await;println!("Hello after delay");}

五、TCP 客户端和服务端

// 服务端
use tokio::net::TcpListener;#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {let listener = TcpListener::bind("127.0.0.1:8080").await?;loop {let (mut socket, _) = listener.accept().await?;tokio::spawn(async move {let mut buf = [0; 1024];// 只要你使用tokio，那边这个io也是异步读取的！！！match socket.read(&mut buf).await {Ok(n) if n == 0 => return,Ok(n) => {println!("Received data: {}", String::from_utf8_lossy(&buf[..n]));}Err(e) => {eprintln!("Failed to read from socket; error = {}", e);}}});}
}// 客户端
use tokio::net::TcpStream;#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {let mut stream = TcpStream::connect("127.0.0.1:8080").await?;stream.write_all(b"hello world").await?;Ok(())
}

六、使用 select! 宏处理多个异步操作

这个特效有点像Golang的并非处理，只是Rust 的异步编程模型基于 async/await 语法和 futures，而 Go 使用了基于协程（goroutines）和通道（channels）的模型。

//golangselect {case msg1 := <-ch1:// 处理 msg1case msg2 := <-ch2:// 处理 msg2case <-time.After(time.Second):// 超时处理default:// 默认情况}

//rusttokio::select! {Some(msg1) = receiver.recv() => {// 处理 msg1}Some(msg2) = receiver2.recv() => {// 处理 msg2}_ = tokio::time::sleep(Duration::from_secs(1)) => {// 超时处理}}