扬州将建设网站,沈阳网站建设小志,WordPress论坛小程序,西安集团网站建设目录 tokio的spawn以及spawn_blocking的使用tokio::task::spawn方法解析tokio::task::spawn_blocking()方法解析 时间会遗忘一切 最后更新时间2024.04.29
tokio版本#xff1a;
tokio的spawn以及spawn_blocking的使用
tokio::task::spawn方法解析
tokio的实现原理以及源码… 目录 tokio的spawn以及spawn_blocking的使用tokio::task::spawn方法解析tokio::task::spawn_blocking()方法解析 时间会遗忘一切 最后更新时间2024.04.29
tokio版本
tokio的spawn以及spawn_blocking的使用
tokio::task::spawn方法解析
tokio的实现原理以及源码解析请移步我的另一篇博客 我们举一个实际的例子来说明tokio::spawn的使用。我们创建一个tokio::main指定工作线程数量为2方便大家理解如果不指定则会与CPU数量相同。因为在这个例子中我们一共有两个异步sleep所以创建两个工作线程方便大家理解。
use tokio;#[tokio::main(flavor multi_thread, worker_threads 2)]
async fn main() {let handle_1 tokio::task::spawn(async {std::thread::sleep(std::time::Duration::from_secs(10));println!(sleeping 10s);});let handle_2 tokio::task::spawn(async {std::thread::sleep(std::time::Duration::from_secs(1));println!(second spawn!);});tokio::join!(handle_1, handle_2);println!(hello world!);
}tokio::spawn方法的返回值是一个handle如果不调用tokio::join!方法tokio是不会将这两个handle放入工作线程中去运行的。当我们调用了tokio::join!后相当于同时调用了handl_1.await和handle_2.awaitmain函数主线程会阻塞等待这两个handle执行完成。所以最终的输出结果是这样的
# 等1s后打印
second spawn!
# 打印second spawn后等9s后打印
sleeping 10s
# 打印sleeping 10s后立即打印
hello world!tokio::spawn方法的返回值是一个handle如果对这个handle执行.await方法会阻塞当前调用这个spawn方法的线程只有在这个handle执行完成后才会继续执行后面的代码。如下例所示
use tokio;#[tokio::main(flavor multi_thread, worker_threads 2)]
async fn main() {let _ tokio::task::spawn(async {std::thread::sleep(std::time::Duration::from_secs(10));println!(sleeping 10s);}).await;let _ tokio::task::spawn(async {std::thread::sleep(std::time::Duration::from_secs(1));println!(second spawn!);}).await;println!(hello world!);
}在该例子中因为在主线程中使用tokio::task::spawn创建了第一个handle_1并调用该handle_1的.await方法此时主线程阻塞在这里等待handle_1执行完毕即sleep 10s后打印sleep 10s。随后使用tokio::task::spawn创建第二个handle_2并调用该handle_2的.await方法此时主线程阻塞在这里等待handle_2执行完毕即sleep 1s后打印second spawn!。最后执行主线程中的hello world打印。
上例的输出为
# 等10s后打印
sleeping 10s
# 打印sleeping 10s完成后等1s后打印
second spawn!
# 打印second spawn完成后立即打印
hello world!如果我们工作线程有两个但是我们有3个异步操作会发生什么呢见下例
use tokio;#[tokio::main(flavor multi_thread, worker_threads 2)]
async fn main() {let handle_1 tokio::task::spawn(async {std::thread::sleep(std::time::Duration::from_secs(10));println!(sleeping 10s);});let handle_2 tokio::task::spawn(async {std::thread::sleep(std::time::Duration::from_secs(1));println!(second spawn!);});let handle_3 tokio::task::spawn(async {std::thread::sleep(std::time::Duration::from_secs(5));println!(third spawn!);});tokio::join!(handle_1, handle_2, handle_3);println!(hello world!);
}我们有3个异步spawn分别睡眠10s、1s、5s。tokio::join按照写代码的顺序先join的10s的handle_1然后1s的handle_2然后5s的handle_3。由于只有两个工作线程所以10s的handle_1和1s的handle_2可以同时在两个工作线程中执行但是handle_3不行因为已经没有多余的工作线程可供handle_3去运行了所以handle_3只能先挂起并不执行。过了1s后当handle_2的工作线程把handle_2执行完此时这个工作线程空闲出来就可以去执行handle_3了所以在handle_2的second spawn打印完成以后handle_3开始执行所以最终的输出如下
# 等1s后打印
second spawn!
# 在second spawn打印后等5s打印
third spawn!
# 在third spawn打印后等4s打印
sleeping 10s
# 在sleeping 10s打印后立即打印hello world
hello world!tokio::task::spawn_blocking()方法解析
我们知道当我们定义tokio的时候可以定义工作线程的数量
#[tokio::main(flavor multi_thread, worker_threads 2)]但是我们可以看到在上述的例子中如果工作线程被阻塞了即使这个工作线程啥都不做他也会阻塞在那里这样CPU就开始摸鱼了但是我们是社会主义怎么能摸鱼呢所以我们不能让CPU有能摸鱼的机会那么这些阻塞的工作应该怎么办呢这里tokio给出了一个spawn_blocking的方法。 spawn_blocking方法中的内容不会在工作线程中运行而是创建了一个单独的线程用来执行写在spawn_blocking方法中的内容这样即使是写在spawn_blocking方法中的内容是阻塞的工作也仅仅是阻塞了新创建出来的这个线程不会导致用来进行异步操作的工作线程阻塞这样工作线程可以正常调度其他的各种tokio::spawn而不至于陷在那里。 请看下例
use tokio;#[tokio::main(flavor multi_thread, worker_threads 2)]
async fn main() {let handle_1 tokio::task::spawn(async {std::thread::sleep(std::time::Duration::from_secs(10));println!(sleeping 10s);});let handle_2 tokio::task::spawn(async {std::thread::sleep(std::time::Duration::from_secs(1));println!(second spawn!);});let handle_3 tokio::task::spawn_blocking(async {std::thread::sleep(std::time::Duration::from_secs(5));println!(third spawn!);});tokio::join!(handle_1, handle_2, handle_3);println!(hello world!);
}
