我尝试使用“indicatif”包来显示几个子任务的进度条以及一个计算所有已完成任务的进度条。我的代码是异步的并使用 tokio。这是一个例子:
货物.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
是计算所有已完成任务的条形图。
问题是在所有工作已经完成之前,控制台上不会出现任何进度条,只向我显示它们的最终状态。我尝试遵循issue1和issue2中的建议,但有两个问题:
- 我尝试
let m = tokio::task::spawn_blocking(move || m.join().unwrap());
按照上面的代码插入,但它不起作用; total_pb
在所有任务完成之前不会完成,这意味着m.join()
不能在每个生成的任务之后立即调用作为问题 1 中建议的同步。
如果相关,我在 Windows 10 上。