Rust Async 异步编程 简易教程
理解tokio核心(1): runtime
Rust 中的异步编程
Rust 中的 async 和线程
OS 线程:
- 适用于少量任务,有内存和 CPU 开销,且线程生成和线程间切换非常昂贵
- 线程池可以降低一些成本
- 允许重用同步代码,代码无需大概,无需特定编程模型
- 有些系统支持修改线程优先级
Async:
- 显著降低内存和 CPU 开销
- 同等条件下,支持比线程多几个数量级的任务(少数线程支撑大量任务)
- 可执行文件大(需要生成状态机,每个可执行文件捆绑一个异步运行时)
- async 把一段代码转化为一个实现了 Future trait 的状态机。虽然在同步方法中调用阻塞函数会阻塞整个线程,但阻塞的 Future 将放弃对线程的控制,从而允许其他 Future 来运行。
asnyc 并不是比线程好,只是不同而已!
异步执行环境
异步执行环境(Async Execution Context)是一种机制,用于管理和调度异步任务的执行。在 Rust 中,Tokio 提供了一种异步执行环境,它基于 Futures
模型,允许你编写和执行异步任务。
下面是异步执行环境的一些关键概念:
- 事件循环(Event Loop):异步执行环境通常由一个事件循环驱动。事件循环不断地检查异步任务的状态,执行就绪的任务,并处理事件。它负责调度任务的执行,并确保异步任务按照适当的顺序执行。
- 异步任务(Async Tasks):异步执行环境下的任务是基于 Future 的。
Future 表示一个尚未完成的异步操作
,并提供了一种方便的方式来等待异步操作完成并获取其结果。在 Tokio 中,你可以创建异步任务,然后将它们提交给 Tokio 运行时执行
。 - 任务调度(Task Scheduling):异步执行环境负责调度异步任务的执行。它会根据任务的就绪状态和优先级,决定何时执行任务,并在需要时将任务挂起和恢复。
- 非阻塞执行(Non-blocking Execution):异步执行环境通过非阻塞方式执行异步任务。这意味着当一个任务在等待某些操作完成时,事件循环可以继续处理其他任务和事件,而不必等待该任务完成。
- 并发性(Concurrency):异步执行环境可以管理多个任务的并发执行。通过使用异步编程模型,你可以编写高效的并发代码,同时处理多个任务,而不会受到线程阻塞的影响。
总之,异步执行环境提供了一个灵活的框架,用于管理和执行异步任务。它通过事件循环和 Future 模型,实现了高效的非阻塞并发执行
,让你可以编写高性能的异步应用程序。
理解 async
异步编程,诀窍就是当 CPU 等待外部事件或动作时,异步运行时会安排其他可继续执行的任务在 CPU 上执行。而当从磁盘或 I/O 子系统的系统终端到达的时候,异步运行时会知道识别这事,并安排原来的任务继续执行。
一般来说,I/O 受限(I/O Bound)的程序(程序执行的速度依赖于 I/O 子系统的速度)比 CPU 受限(CPU Bound)的任务(程序执行的速度依赖于CPU的速度)可能更适合异步任务的执行。
async、.await
关键字是 Rust 标准库里拥有异步编程的内置核心原语集的代表,就是语法糖。
Future
Future 是 Rust 中的一个重要概念,用于表示一个尚未完成的异步操作。Future 代表了一个异步计算的结果,可能会在未来的某个时间点产生值或错误。
Future
可以表示:
- 下一次网络数据包的到来
- 下一次鼠标的移动
- 或者仅仅是经过一段时间的时间点
Rust 的异步函数都会返回 Future,Future 基本上就是代表着延迟的计算。
以下是关于 Future 的一些重要概念和特点:
- 表示异步计算:Future 是 Rust 中表示异步计算的一种方式。它表示了一个尚未完成的异步操作,并提供了一种方法来等待操作完成并获取结果。
- 延迟计算:Future 具有延迟计算的特性,即它不会立即执行,而是在未来的某个时间点执行。Future 可以被等待(await),直到它产生值或错误为止。
- 状态:Future 有三种可能的状态:Pending(等待中)、Ready(准备就绪)、Error(错误)。当 Future 处于 Pending 状态时,表示异步操作尚未完成;当 Future 处于 Ready 状态时,表示异步操作已经完成,并且产生了一个值;当 Future 处于 Error 状态时,表示异步操作产生了一个错误。
- 组合和转换:Future 可以被组合和转换,以便实现复杂的异步操作。例如,可以使用 map、and_then、or_else 等方法来对 Future 进行转换和处理,以便处理异步操作的结果。
- 异步编程:Future 是异步编程的基础,它与 async/await 语法一起使用,使得异步编程变得更加简单和直观。通过 async/await 语法,可以方便地编写和组合异步操作,而不必显式地使用 Future 的 API。
总的来说,Future 是 Rust 中表示异步计算的一种抽象,它提供了一种方法来等待异步操作完成并获取结果。通过组合和转换 Future,可以实现复杂的异步操作。Future 与 async/await 语法一起使用,可以使得异步编程变得更加简单和直观。
通俗的理解 Future Trait
Future
代表着一种你可以检验其是否完成的操作
Future
可以通过调用 poll
函数来取得进展
poll
函数会驱动Future
尽可能接近完成- 如果
Future
完成了,就返回poll::Ready(result)
,其中result
就是最终的结果 - 如果
Future
还无法完成,就返回poll::Pending
,并当Future
准备好取得更多进展时调用一个waker
的wake()
函数
针对一个 Future
,你唯一能做的就是使用 poll
来敲它,直到一个值掉出来
创建 future
1)使用 async/await
关键字定义异步函数或异步块。这是最常见和推荐的方式。
async fn
返回的是 Future(本质就是第二种的写法),这个 Future 需要一个执行者来运行。
async fn example_async_function() -> i32 {// Some async operations here42
}async {// Some async operations here42
};
2)使用 Future trait
的 async fn 方法。方法(1)是此方法的语法糖。
use std::future::Future;
// 返回一个实现了 Future trait 的类型,其 Output 类型为 i32
fn example_async_function() -> impl Future<Output = i32> {
// 创建一个异步任务并将其返回,并不会执行异步任务中的代码async {// Some async operations here42}
}
这个异步函数的定义中使用了 async 关键字,但它并不会立即执行。相反,调用者需要将这个异步函数返回的异步任务执行在一个异步执行上下文中,通常是通过 await 表达式或者将异步任务提交给异步运行时来执行。
3)使用异步库提供的函数或宏来创建 Future。比如在 Tokio 中,可以使用 tokio::spawn()
来创建一个异步任务的 Future。
use tokio::task;
// task::spawn() 创建一个异步任务,其参数是一个异步闭包,都是Future
task::spawn(async {// Some async operations here42
});
4)使用 futures crate
中的方法创建 Future。比如使用 futures::future::ready()
创建一个立即返回某个值的 Future,或者使用 futures::future::pending()
创建一个永远不会完成的 Future。
use futures::future;future::ready(42);
future::pending();
5)使用 async_std
或 tokio
等异步运行时提供的方法创建 Future。
async_std::future::ready(42);
tokio::time::delay_for(Duration::from_secs(1)).then(|_| async { 42 });
future 的定义
pub trait Future {/// The type of value produced on completion.#[stable(feature = "futures_api", since = "1.36.0")]#[rustc_diagnostic_item = "FutureOutput"]type Output; // future执行成功返回的类型/// Attempt to resolve the future to a final value, registering/// the current task for wakeup if the value is not yet available.////// # Return value////// This function returns:////// - [`Poll::Pending`] if the future is not ready yet/// - [`Poll::Ready(val)`] with the result `val` of this future if it/// finished successfully.////// Once a future has finished, clients should not `poll` it again.////// When a future is not ready yet, `poll` returns `Poll::Pending` and/// stores a clone of the [`Waker`] copied from the current [`Context`]./// This [`Waker`] is then woken once the future can make progress./// For example, a future waiting for a socket to become/// readable would call `.clone()` on the [`Waker`] and store it./// When a signal arrives elsewhere indicating that the socket is readable,/// [`Waker::wake`] is called and the socket future's task is awoken./// Once a task has been woken up, it should attempt to `poll` the future/// again, which may or may not produce a final value.////// Note that on multiple calls to `poll`, only the [`Waker`] from the/// [`Context`] passed to the most recent call should be scheduled to/// receive a wakeup.////// # Runtime characteristics////// Futures alone are *inert*; they must be *actively* `poll`ed to make/// progress, meaning that each time the current task is woken up, it should/// actively re-`poll` pending futures that it still has an interest in.////// The `poll` function is not called repeatedly in a tight loop -- instead,/// it should only be called when the future indicates that it is ready to/// make progress (by calling `wake()`). If you're familiar with the/// `poll(2)` or `select(2)` syscalls on Unix it's worth noting that futures/// typically do *not* suffer the same problems of "all wakeups must poll/// all events"; they are more like `epoll(4)`.////// An implementation of `poll` should strive to return quickly, and should/// not block. Returning quickly prevents unnecessarily clogging up/// threads or event loops. If it is known ahead of time that a call to/// `poll` may end up taking awhile, the work should be offloaded to a/// thread pool (or something similar) to ensure that `poll` can return/// quickly.////// # Panics////// Once a future has completed (returned `Ready` from `poll`), calling its/// `poll` method again may panic, block forever, or cause other kinds of/// problems; the `Future` trait places no requirements on the effects of/// such a call. However, as the `poll` method is not marked `unsafe`,/// Rust's usual rules apply: calls must never cause undefined behavior/// (memory corruption, incorrect use of `unsafe` functions, or the like),/// regardless of the future's state.////// [`Poll::Ready(val)`]: Poll::Ready/// [`Waker`]: crate::task::Waker/// [`Waker::wake`]: crate::task::Waker::wake#[lang = "poll"]#[stable(feature = "futures_api", since = "1.36.0")]fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output>;// poll方法被异步运行时调用,用来检查异步任务是否已经完成// 返回的Poll是一个枚举,其中Poll::Pending表示还未完成,Poll::Ready(value)表示已经完成
}
谁执行 poll
方法?
是异步执行器来,它是异步运行时的一部分。异步执行器会管理一个 Future
的集合,并通过调用 Future
上的 poll
方法来驱动他们完成。所以函数或代码块在前面加上 async
关键字后,就相当于告诉异步执行器它会返回 Future
,这个 Future
需要被驱动直到完成。
利用 Tokio 库来理解 Future
自定义 Future
异步任务 h1 中的 Future
一直返回 PollPending
,这意味这个 Future
将永远无法执行。
use std::future::Future;
use std::pin::Pin;
// Context包含异步任务的上下文,可以用来唤醒当前的任务
use std::task::{Context, Poll};
use std::thread::sleep;
use std::time::Duration;struct ReadFileFuture {}impl Future for ReadFileFuture {type Output = String;// 这个Future会被异步运行时反复地poll,把这个Future给固定住(pin住),固定到内存的某一特定位置,对异步代码的安全性是必要的fn poll(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<Self::Output> {println!("tokio stop polling me", );Poll::Pending}
}#[tokio::main]
async fn main() {println!("hello before reading file!", );
// 这里的Future一直返回 `PollPending`,这意味这个Future将永远无法执行let h1 = tokio::spawn(async {let future1 = ReadFileFuture {};future1.await;});
// 异步任务h1被poll一次之后执行完毕,就不会再被poll和执行了let h2 = tokio::spawn(async {let file2_contents = read_from_file2().await;println!("{:?}", file2_contents);});let _ = tokio::join!(h1, h2);
}fn read_from_file2() -> impl Future<Output = String> {async {sleep(Duration::new(2, 0));println!("{:?}", "processing file 2");String::from("hello, there from file 2")}
}
Tokio 执行器如何知道何时再次 poll 第一个 Future?
Tokio 执行器肯定不会一直 poll,Tokio使用一个 Weaker 组件来处理这件事。当被异步执行器 poll
过的任务还没有准备好产生值的时候,这个任务就被注册到一个 Weaker。Weaker 会有一个处理程序(handle),它会被存储在任务关联的 Context 对象中。
Weaker 有一个 wake()
方法,可以用来告诉异步执行器关联的任务应该被唤醒了。当 wake()
方法被调用了,Tokio 执行器就会被通知是时候再次 poll
这个异步的任务了,具体方式是调用任务上的 poll()
函数。简单来说,就是通过 wake()
函数,执行器就确切的知道哪些 Future
已准备好进行 poll
的调用。
Tokio 库组件
Tokio 组件的简化工作流程
存储或传递 Future
通常,async
的函数在调用后会立即 .await
,这么做就没有什么问题,但是如果存储 Future或将其传递给其他任务或线程,就会出现生命周期的问题。
一种变通解决办法:把使用引用作为参数的 async fn
转为一个 'static future
。在 async
块里,将参数和 async fn
的调用捆绑到一起(延长参数的生命周期来匹配 future)。
fn bad() -> impl Future<Output = u8> {let x = 5;borrow_x(&x) // x的生命周期是不够长的,活不到这个Future未来使用的时候
}fn good() -> impl Future<Output = u8> {async { // 套一个async块,它返回的是一个 Future
// 这个Future 会记住里边的一些变量,以保证 x 的生命周期和 borrow_x(&x).await 的生命周期是匹配的let x = 5;borrow_x(&x).await}
}
Waker
使用 Waker 实现一个简单的计时器 Future
依赖
futures = "0.3.21"
tokio = { version = "1", features = ["full"] }
src\main.rs
use std::{future::Future,pin::Pin,sync::{Arc, Mutex},task::{Context, Poll, Waker},thread,time::Duration,
};pub struct TimerFuture {shared_state: Arc<Mutex<SharedState>>,
}// TimerFutre 是要运行在一个任务上,这个任务拥有一个 Waker,// 在Future和等待的线程共享状态
struct SharedState {// 睡眠时间是否已经都过完completed: bool,// TimerFutre 所运行的任务的 Waker 在设置 completed = true 之后,// 线程可以使用它来告诉 TimerFuture 的任务可以唤醒,看到 completed = true 并前进waker: Option<Waker>, // 线程可以用这个 waker 来告诉 TimerFuture 这个任务可以唤醒了
}// 为 TimerFuture 这个结构体添加 Future trait
impl Future for TimerFuture {type Output = ();// 查看 shared state,看下 timer 是否已经结束fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {let mut shared_state = self.shared_state.lock().unwrap();if shared_state.completed {Poll::Ready(())} else {// 设置 waker 以便当 timer 结束时线程可以唤醒当前任务// 保证 Future 可以再次被 poll,并看到 completed = true// 相比每次克隆 waker,如果只做一次显然更有诱惑力// 但是TimerFuture 可在执行者的任务间移动,这会导致过期的 waker 指向错误的任务,// 从而阻止了 TimerFuture 正确的唤醒// 注意:可以使用 Waker::will_wake 函数来检查这一点// 但为了简单起见,我们就省略了这一点shared_state.waker = Some(cx.waker().clone());Poll::Pending}}
}// 创建一个关联函数,用来创建一个新的 TimerFuture,并将在提供的时限过后完成
impl TimerFuture {pub fn new(duration: Duration) -> Self {let shared_state = Arc::new(Mutex::new(SharedState{completed: false,waker: None,}));// 生成新线程let thread_shared_state = shared_state.clone();thread::spawn(move || {thread::sleep(duration);let mut shared_state = thread_shared_state.lock().unwrap();// 发出信号:计时器已停止并唤醒 Future 被 poll 的最后一个任务// 如果存在的话shared_state.completed = true;if let Some(waker) = shared_state.waker.take() {waker.wake();}});TimerFuture {shared_state}}
}#[tokio::main] // 使用 tokio 的宏来启动异步 main 函数
async fn main() {println!("Timer starts now!");TimerFuture::new(Duration::from_secs(3)).await;println!("Timer ended!");
}
await 函数
在 Rust 中,await 关键字是用于等待异步操作完成的关键字。它只能在异步函数中使用,并且用于暂停当前异步函数的执行,直到一个异步操作完成并返回结果。
换句话说,在发生阻塞时,它让放弃当前线程的控制权成为可能,这就允许在等待操作完成的时候,允许其他代码取得进展。
当程序执行到 await 关键字时,以下是发生的情况:
如果 await 关键字之前的表达式产生了一个 Future,那么它会开始执行这个 Future,并将当前的异步函数挂起。在 await 关键字之后的代码会暂时被挂起,等待异步操作的结果。当异步操作完成并返回结果时,异步函数会恢复执行,并继续执行 await 关键字之后的代码。
await 关键字的返回值是异步操作的结果,它的类型通常是 Future 所产生的值类型。
await 关键字的使用使得异步代码能够以同步的方式编写,不需要显式地处理回调函数或者使用其他复杂的异步编程模式。这极大地提高了代码的可读性和可维护性。
在多线程执行者上进行 .await
当使用多线程 future
执行者时,future
就可以在线程间移动:
- 所以
async
体里面用的变量都必须能够在线程间移动 - 因为任何的
.await
都可能导致切换到一个新的线程
这意味着使用以下类型是不安全的:
Rc
,&RefCell
和任何其他没有实现Send trait
的类型,包括没实现Sync trait
的引用- 注意:调用
.await
时,只要这些类型不在作用域内,就可以使用它们。
在跨越一个 .await
期间,持有传统的、对 future
无感知的锁,也不是好主意:
- 可导致线程池锁定
- 为此,可使用
future::lock
里的Mutex
而不是std::sync
里的
异步任务是lazy的
异步任务在被创建后并不会立即执行,而是在被提交给执行环境并开始执行上下文中的事件循环时才会被执行。具体来说,异步任务的执行是由事件循环驱动的,当事件循环开始运行时,它会轮询所有待执行的异步任务,并根据任务的状态和条件来决定是否执行或者继续等待。
一般来说,异步任务会在以下几种情况下被执行:
调用 .await
方法:
当一个异步任务被等待时(例如使用 await 表达式),它会被唤醒并开始执行。
在执行到一个 await 表达式时,事件循环会暂停当前任务的执行,并切换到其他待执行的任务。
async fn example() {// 异步任务逻辑
}async fn main() {let result = example().await;
}
被提交给执行环境:
当一个异步任务被提交给执行环境(如 Tokio 运行时)时,它会被加入到执行队列中等待执行。
当执行环境开始运行事件循环时,它会轮询执行队列中的任务,并执行准备就绪的任务。
use tokio::task;async fn example() {// 异步任务逻辑
}fn main() {let rt = tokio::runtime::Runtime::new().unwrap();rt.spawn(example());
}
总之,异步任务的执行是由事件循环控制的,在事件循环开始运行并开始轮询执行队列时,待执行的异步任务才会被执行。
tokio创建的线程进行切换时,需要上下文切换吗
在 Tokio 中,当线程执行异步任务时,通常不需要进行上下文切换。这是因为 Tokio 的异步执行模型是非阻塞的,它使用事件驱动的方式执行任务,而不是依赖于线程的阻塞和切换。
在 Tokio 中,异步任务通过 Future 和异步函数来表示,并由事件循环调度执行。当一个异步任务需要等待某些事件发生时(例如等待 I/O 操作完成、等待定时器触发等),它会将自己挂起,并注册一个回调函数来处理事件发生时的通知。这样,当事件发生时,事件循环会唤醒相应的任务,并继续执行。
由于 Tokio 的异步执行模型是非阻塞的,并且通过事件循环驱动任务的执行,因此不需要在任务之间进行显式的上下文切换。相反,任务的执行是事件驱动的,只有在等待事件时才会挂起,而不会占用线程的执行时间。这样可以避免传统多线程并发模型中因上下文切换而导致的性能损失。
总之,在 Tokio 中,异步任务的执行是通过事件驱动的方式实现的,通常不需要进行显式的上下文切换。这是异步编程模型的一个优势,能够提供高效的并发执行和良好的性能。
在实践中,即使使用Tokio,仍然可能存在上下文切换。这是因为某些情况下,任务可能需要等待I/O操作的完成,这可能会导致线程被阻塞。当一个线程被阻塞时,Tokio可以在另一个线程上调度其他任务,以确保整体的执行效率。虽然在 Tokio 运行时上运行的任务可能会涉及到上下文切换,但相比于传统的线程池模型,它大大减少了上下文切换的频率和成本。
tokio::Runtime
Tokio Runtime Builer
tokio::runtime::Builder
构建器用于用于构建一个 Tokio 运行时,构建一个 Tokio 运行时意味着创建一个异步执行环境,以便异步任务能够被调度和执行。
use anyhow::Result;fn main() -> Result<()> {let runtime = tokio::runtime::Builder::new_multi_thread().enable_all().build().unwrap();runtime.block_on(async {println!("Hello");});Ok(())
}
Tokio 运行时支持多线程与单线程,分别使用 Builder::new_multi_thread()
、Builder::new_current_thread()
去设置
enable_all
同时启用了IO和时间驱动程序,可通过 enable_io
、enable_time
分别去启用它们
如果在不启用时间驱动程序的运行时中调用tokio::time::sleep将发生 panic
可以通过Builder::new_current_thread().worker_threads(num)
去指定使用的工作线程数,不调用 worker_threads()
线程个数默认设置为系统的 cpu 核心数量,可以通过环境变量 TOKIO_WORKER_THREADS
去覆盖这个默认值
tokio::main 宏
#[tokio::main]
async fn main() {println!("Hello world");
}
等于以下代码
fn main() {tokio::runtime::Builder::new_multi_thread().enable_all().build().unwrap().block_on(async {println!("Hello world");})
}
可通过 flavor
去指定使用单线程还是多线程,通过 worker_threads
去指定线程数
#[tokio::main(flavor = "current_thread")]#[tokio::main(worker_threads = 2)]
block_on(在同步上下文中执行异步代码)
Runtime::block_on
是 Tokio 运行时提供的一个方法,它允许在同步上下文中执行异步代码,并且会一直阻塞当前线程,直到传递给它的 Future 完成。
不能在 Tokio 异步上下文中调用,示例:
use anyhow::Result;fn main() -> Result<()> {let runtime = tokio::runtime::Builder::new_current_thread().enable_all().build()?;runtime.block_on(async {let temp_runtime = tokio::runtime::Builder::new_current_thread().build().unwrap();temp_runtime.block_on(async {}); // ERROR! 此方法不能在异步上下文中调用!!tokio::spawn(async{let temp_runtime = tokio::runtime::Builder::new_current_thread().build().unwrap();temp_runtime.block_on(async {}); // ERROR! 此方法不能在异步上下文中调用!!}).await;tokio::task::spawn_blocking(||{let temp_runtime = tokio::runtime::Builder::new_current_thread().build().unwrap();temp_runtime.block_on(async {}); // OK!});});Ok(())
}
tokio::spawn、Runtime::spawn、enter
tokio::spawn
(全局函数,可以在任何地方直接调用)生成一个新的异步任务,它将立即在后台运行,返回一个 JoinHandle
,通过 .await
返回的JoinHandle
来等待它执行完成。生成的任务可与其他任务并发执行。可能在当前线程上执行,也可能被发送到不同的线程执行。具体情况取决于当前的 Runtime 配置。
Runtime::spawn
则直接在 self 运行时上生成任务,不需要在上下文上调用
tokio::spawn
函数必须在 Tokio 运行时上下文中调用
错误示例:
use std::thread;
use std::time::Duration;use anyhow::Result;fn main() -> Result<()> {let _ = tokio::spawn(async {tokio::time::sleep(Duration::from_secs(1)).await;println!("Hello World");});thread::sleep(Duration::from_secs(3));Ok(())
}
创建运行时并在 block_on
内部 spawn
use std::time::Duration;
use anyhow::Result;fn main() -> Result<()> {let runtime = tokio::runtime::Builder::new_multi_thread().enable_all().build()?;runtime.block_on(async{ // block_on在内部调用了Runtime::enter方法,进入了上下文let _ = tokio::spawn(async {tokio::time::sleep(Duration::from_secs(1)).await;println!("Hello World");}).await;});Ok(())
}
也可以手动调用enter()
进入上下文,示例:
use std::thread;
use std::time::Duration;
use anyhow::Result;fn main() -> Result<()> {let runtime = tokio::runtime::Builder::new_multi_thread().enable_all().build()?;let enter_guard = runtime.enter();let _ = tokio::spawn(async {tokio::time::sleep(Duration::from_secs(1)).await;println!("Hello World");});thread::sleep(Duration::from_secs(3));Ok(())
}
如果有多个Tokio 运行时,就可以使用runtime.enter()
选择进入哪个运行时上下文了,tokio::spawn
会在你进入的运行时中运行
enter()
返回了一个 enter_guard
对象,当 enter_guard
销毁时,也就退出了这个上下文
use std::thread;
use std::time::Duration;
use anyhow::Result;fn main() -> Result<()> {let runtime = tokio::runtime::Builder::new_multi_thread().enable_all().build()?;let enter_guard = runtime.enter();drop(enter_guard); // 退出运行时上下文// 这里的tokio::spawn没有在运行时上下文中执行,因此报错let _ = tokio::spawn(async {tokio::time::sleep(Duration::from_secs(1)).await;println!("Hello World");});thread::sleep(Duration::from_secs(3));Ok(())
}
tokio 的两种线程
Tokio能够在少量线程上同时运行许多任务,通过在每个线程上重复交换当前运行的任务来实现。然而,这种交换只能在 .await
点发生时进行,所以长时间没有达到 .await
的代码将阻止其他任务的运行。为了解决这个问题,Tokio提供了两种类型的线程:核心线程和阻塞线程。
核心线程
核心线程是所有异步代码运行的地方,默认情况下,Tokio会为每个CPU核心生成一个核心线程。可以使用环境变量 TOKIO_WORKER_THREADS
来覆盖默认值。这些线程会在运行时创建时立刻创建。
阻塞线程
阻塞线程是根据需要生成的,专用于运行阻塞任务,并在一定时间内保持活动状态,这个时间可以通过 thread_keep_alive
进行配置,默认值为 10 秒。由于Tokio无法像处理异步代码那样交换阻塞任务,因此阻塞线程的数量上限非常大。,可以通过 Builder
的 max_blocking_threads
去设置它的数量,默认数量为 512
,这些线程是按需创建的,如果达到了最大数量,就会在队列里排队。
在阻塞线程中执行的代码不会像异步线程中的任务那样在 await
点换出,它会独占一个线程去执行,所以生成一个堵塞任务接收闭包
类型而不是 Future
类型。
在阻塞线程中发生阻塞,不会阻碍异步任务的运行,因为异步任务只会被调度到核心线程中交替运行。
Runtime::spawn_blocking
要生成一个阻塞任务,应该使用 spawn_blocking
函数,接收一个闭包,并在阻塞线程中运行它
#[tokio::main]
async fn main() {// This is running on a core thread.let blocking_task = tokio::task::spawn_blocking(|| {// This is running on a blocking thread.// Blocking here is ok.});// We can wait for the blocking task like this:// If the blocking task panics, the unwrap below will propagate the// panic.blocking_task.await.unwrap();
}
与spawn
一样返回JoinHandle
类型,可通过.await
它来等待它执行完成并获取到返回值
thread::spawn
创建的线程与堵阻塞线程一样都是独占的,但是spawn_blocking
返回tokio::task::JoinHandle
类型实现了Future
,你可以在一个异步任务中等待它,而thread::spawn
做不到这点,它返回的std::thread::JoinHandle
没有实现 Future
Runtime::block_in_place
block_in_place
接收一个闭包,并在当前线程阻塞运行,并返回闭包的返回值
为了防止当前线程的其他异步任务被堵塞,Tokio 会把它们移交到其他异步线程中
此函数不能在 Tokio 使用单线程运行时时使用,因为在这种情况下没有其他异步线程,无法移交
tokio::task
tokio::task::spawn 与 tokio::task::JoinHandle
通常情况下,如果只使用默认的 Tokio 运行时,tokio::task::spawn
和 tokio::spawn
的功能基本相同,tokio::task::spawn
可以选择某一 Tokio 运行时执行异步任务。
它接受一个 返回Future
的异步函数 作为参数,并返回一个 JoinHandle
use tokio::task;async fn async_task() {// 异步任务的逻辑println!("Running async task");
}#[tokio::main]
async fn main() {let handle = tokio::task::spawn(async_task());// 等待异步任务完成// JoinHandlehandle.await.unwrap();
}
tokio::task::spawn_blocking
用于将一个阻塞的操作转换为异步任务,并在 Tokio 的线程池中执行。这对于需要执行阻塞操作(如CPU密集型计算)的场景非常有用,以避免阻塞整个Tokio运行时。
use tokio::task;fn blocking_task() -> u32 {// 阻塞操作的逻辑42
}#[tokio::main]
async fn main() {let handle = tokio::task::spawn_blocking(blocking_task);let result = handle.await.unwrap();println!("Blocking task result: {}", result);
}
异步运行实例
use std::{thread::sleep, time::Duration};#[tokio::main]
async fn main() {let h1 = tokio::spawn(async {let _file1_contents = read_file1().await;});let h2 = tokio::spawn(async {let _file2_contents = read_file2().await;});let _ = tokio::join!(h1, h2);
}async fn read_file1() -> String {sleep(Duration::new(4, 0));println!("processing file 1");String::from("file 1")
}async fn read_file2() -> String {sleep(Duration::new(2, 0));println!("processing file 2");String::from("file 2")
}