我正在尝试tokio的tokio::spawn
和tokio::task::spawn
,结果我不明白后者是如何运作的。
当我运行以下代码时:
#[tokio::main]
pub async fn main() {
// I'm spawning one block of functions
let h = tokio::task::spawn_blocking(move || {
block_one();
});
// and another block of functions
let h2 = tokio::spawn(async move {
block_two().await;
});
// then I collect the handles
h.await.unwrap();
h2.await.unwrap();
}
#[tokio::main] //needed as this block is not treated as syncronous by main
pub async fn block_one() {
let mut handles = vec![];
for i in 1..10 {
let h = tokio::spawn(async move {
println!("Starting func #{}", i);
i_take_random_time().await;
println!("Ending func #{}", i);
});
handles.push(h);
}
for h in handles {
h.await.unwrap();
}
}
pub async fn block_two() {
let mut handles = vec![];
for i in 10001..10010 {
let h = tokio::spawn(async move {
println!("Starting func #{}", i);
i_take_random_time().await;
println!("Ending func #{}", i);
});
handles.push(h);
}
for h in handles {
h.await.unwrap();
}
}
我的期望是,第一个函数块将完全运行--只有第二个块才会运行。这就是我对"spawn_blocking“的理解--它阻止进一步的执行,直到它里面的任何东西都完成为止。
实际上,我得到的是第二个函数块首先启动(全部10个函数块)--只有第一个块才开始。所以这和我所期望的完全相反。
更让人困惑的是,当我修改上面的代码使两个块都有spawn_blocking
时,所有20个函数一起启动,就好像这两个块都是一个大异步循环的一部分一样。这也不是我所期望的--我认为第一个街区会运行,在它完成之前就会阻塞,然后第二个会运行。
谁能帮我破译这是怎么回事?
这个回购中提供了用于再现上述2种场景的完整代码。
注意:这里有两个异步级别:块之间和块内。希望有助于避免任何混乱。
发布于 2021-06-16 11:31:33
听起来,您希望spawn_blocking
阻止其他东西运行,但它的目的正好相反。spawn_blocking
的目的是避免阻止其他事物的运行。
使用spawn_blocking
的主要位置是用于否则阻塞线程使用非异步操作(如std::net
)的操作。它通过将它们卸载到一个单独的线程池来做到这一点。这个名称来自于这样一个事实:您正在生成一个阻塞操作,以便它可以在其他地方运行。
要等待第一个块完成,您可以这样做:
#[tokio::main]
pub async fn main() {
// I'm spawning one block of functions
let h = tokio::task::spawn_blocking(move || {
block_one();
});
// wait for the first block
h.await.unwrap();
// then spawn another block of functions
let h2 = tokio::spawn(async move {
block_two().await;
});
h2.await.unwrap();
}
请注意,在#[tokio::main]
(或block_on
)直接在spawn_blocking
中使用很少是您想要的。只需使用tokio::spawn
生成一个普通的任务。
https://stackoverflow.com/questions/67989412
复制相似问题