首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >意外tokio::task::spawn_blocking行为

意外tokio::task::spawn_blocking行为
EN

Stack Overflow用户
提问于 2021-06-15 15:41:28
回答 1查看 2.6K关注 0票数 4

我正在尝试tokio的tokio::spawntokio::task::spawn,结果我不明白后者是如何运作的。

当我运行以下代码时:

代码语言:javascript
运行
复制
#[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种场景的完整代码。

  • 场景5=我描述的第一个案例
  • 场景6=我描述的第二个案例

注意:这里有两个异步级别:块之间和块内。希望有助于避免任何混乱。

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-06-16 11:31:33

听起来,您希望spawn_blocking阻止其他东西运行,但它的目的正好相反。spawn_blocking的目的是避免阻止其他事物的运行。

使用spawn_blocking的主要位置是用于否则阻塞线程使用非异步操作(如std::net )的操作。它通过将它们卸载到一个单独的线程池来做到这一点。这个名称来自于这样一个事实:您正在生成一个阻塞操作,以便它可以在其他地方运行。

要等待第一个块完成,您可以这样做:

代码语言:javascript
运行
复制
#[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生成一个普通的任务。

票数 8
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/67989412

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档