如何使用指示符在异步 Rust 中显示总计数器栏？

Question

我尝试使用“indicatif”包来显示多个子任务的进度条以及一个计算所有已完成任务的进度条。我的代码处于异步状态并使用 tokio。这是一个例子：

cargo.toml

[package]
name = "test-project"
version = "0.1.0"
edition = "2018"

# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

[dependencies]
indicatif = "0.15.0"
tokio = { version = "1", features = ["full"] }
futures = "0.3"

src/main.rs

use std::time::Duration;
use futures::{StreamExt, stream::futures_unordered::FuturesUnordered};
use indicatif::{MultiProgress, ProgressBar, ProgressStyle};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let m = MultiProgress::new();
    let sty = ProgressStyle::default_bar()
        .template("[{elapsed_precise}] {bar:40.cyan/blue} {pos:>7}/{len:7} {msg}")
        .progress_chars("##-");
    let total_pb = m.add(ProgressBar::new(3));
    total_pb.set_style(sty.clone());
    let mut futs = FuturesUnordered::new();
    let durations = [15u64, 8, 3];
    let mut pb_cnt = 0usize;
    for &duration in durations.iter() {
        let pb = m.insert(pb_cnt, ProgressBar::new(256));
        pb_cnt += 1;
        pb.set_style(sty.clone());
        let total_pb = total_pb.clone();
        let task = tokio::spawn(async move {
            for i in 0i32..256 {
                pb.set_message(&format!("item #{}", i + 1));
                pb.inc(1);
                tokio::time::sleep(Duration::from_millis(duration)).await;
            }
            pb.finish_with_message("done");
            total_pb.inc(1);
        });
        futs.push(task);
    }
    while let Some(result) = futs.next().await {
        result?;
    }
    total_pb.finish_with_message("done");
    m.join()?;
    // tokio::task::spawn_blocking(move || m.join().unwrap() ).await.unwrap(); 

    Ok(())
}

total_pb 是计算所有已完成任务的栏。

问题是 none 进度条出现在控制台上，直到所有工作都已完成，只显示它们的最终状态。我尝试遵循 issue1 and issue2 中的建议，但有 2 个问题：

我尝试按照上面的代码插入 let m = tokio::task::spawn_blocking(move || m.join().unwrap());，但它不起作用；
total_pb 在所有任务完成之前不会完成，这意味着 m.join() 不能像问题 1 中建议的那样在每个生成的任务之后立即调用作为同步。

我在 Windows 10，如果相关的话。

Answer 1

有不同的解决方案，您需要在活动阻塞线程上调用m.join()。你要么在你的主线程中完成它，那么你需要移动你的：

while let Some(result) = futs.next().await {
    result?;
}

在其他地方，比如生成任务。

或者您生成一个阻塞线程并在最后等待它（可能是最好的）。所以就这样做：

let handle_m = tokio::task::spawn_blocking(move || m.join().unwrap()); // add this line
while let Some(result) = futs.next().await {
    result?;
}
total_pb.finish_with_message("done");
handle_m.await?; // don't forget to await your handle

请注意，与您的尝试 #1 相反，您必须在等待结果之前生成阻塞线程...

有 2 个示例可以在以后帮助您 tokio.rs multi.rs。

如何使用指示符在异步 Rust 中显示总计数器栏？

How to show a total counter bar in async Rust using indicatif?

asynchronous

rust

tokio