时间戳

首页 > 解决方案 > 如何在启动异步操作和在循环中等待其结果之间有一个宽限期？

asynchronous - 如何在启动异步操作和在循环中等待其结果之间有一个宽限期？

问题描述

我有一个循环：

let grace = 2usize;
for i in 0..100 {
    if i % 10 == 0 {
        expensive_function()
    } else {
        cheap_function()
    }
}

目标是当它命中时expensive_function()，它异步运行，并允许grace进一步的迭代次数，直到等待expensive_function()。

如果expensive_function()在迭代 10 处触发，则它可以运行迭代 11 和 12，然后需要等待expensive_function()迭代 10 上的运行完成才能继续。

我怎么能这样做？

在我的情况下expensive_function()是有效的：

fn expensive_function(&b) -> Vec<_> {
    return b.iter().map(|a| a.inner_expensive_function()).collect();
}

因此，我计划在这个函数中使用多线程。

标签： asynchronousrust

解决方案

当您开始进行昂贵的计算时，将生成的 future 以及等待结果的截止时间存储在一个变量中。在这里，我使用Option一个元组：

use std::{thread, time::Duration};
use tokio::task; // 0.2.21, features = ["full"]

#[tokio::main]
async fn main() {
    let grace_period = 2usize;
    let mut pending = None;

    for i in 0..50 {
        if i % 10 == 0 {
            assert!(pending.is_none(), "Already had pending work");

            let future = expensive_function(i);
            let deadline = i + grace_period;
            pending = Some((deadline, future));
        } else {
            cheap_function(i);
        }

        if let Some((deadline, future)) = pending.take() {
            if i == deadline {
                future.await.unwrap();
            } else {
                pending = Some((deadline, future));
            }
        }
    }
}

fn expensive_function(n: usize) -> task::JoinHandle<()> {
    task::spawn_blocking(move || {
        println!("expensive_function {} start", n);

        thread::sleep(Duration::from_millis(500));

        println!("expensive_function {} done", n);
    })
}

fn cheap_function(n: usize) {
    println!("cheap_function {}", n);
    thread::sleep(Duration::from_millis(1));
}

这会产生输出

cheap_function 1
expensive_function 0 start
cheap_function 2
expensive_function 0 done
cheap_function 3
cheap_function 4
cheap_function 5

由于您没有提供 and 的定义expensive_function，因此cheap_function我提供了适当的定义。

这里有一件棘手的事情是我需要sleep在cheap_function. 没有它，我的操作系统在轮询它之前永远不会安排昂贵的线程，从而有效地消除任何并行工作。在更大的程序中，操作系统可能会调度线程，因为更多的工作将由cheap_function. 您也可以使用thread::yield_now相同的效果。

也可以看看：

推荐阅读