Check if all spawn stop

Author: iewnfod

src/main.rs

@@ -4,7 +4,7 @@ mod ui;
 
 
 // main
-#[tokio::main(worker_threads = 3)]
+#[tokio::main]
 async fn main() {
     // 初始化一些东西
     let save_path = std::path::Path::new(ui::data::SAVE_DIR);
@@ -22,24 +22,40 @@ async fn main() {
 
     root.show();
 
+    let mut handlers: Vec<tokio::task::JoinHandle<()>> = vec![];
+
     // app.run().unwrap();
 
     while app.wait() {
         app.redraw();
         // 刷新状态栏
         buffer.status_bar.set_value(unsafe { &ui::STATUS_BAR_CONTENT });
+
         // 刷新文件系统
         buffer.refresh_file_system(ui::data::SAVE_DIR);
 
+        // 查看是否还有下载在运行
+        let mut handle_flag = true;
+        for handle in handlers.iter() {
+            if !handle.is_finished() {  // 如果没有停止，就是 false
+                handle_flag = false;
+            }
+        }
+        if handle_flag {  // 如果还是 true 的话，那就清楚 status bar
+            ui::change_status_bar_content(&std::string::String::new());
+        }
+
         if let Some(msg) = receiver.recv() {
             match msg {
                 Message::Start => {
                     let buffer_clone = buffer.clone();
                     println!("Start");
 
-                    tokio::spawn( async {
-                        ui::network::start(buffer_clone).await;
-                    });
+                    handlers.push(
+                        tokio::spawn( async {
+                            ui::network::start(buffer_clone).await;
+                        })
+                    );
                 }
             }
         }

src/ui.rs

@@ -121,7 +121,7 @@ pub fn add_widgets(root: &mut window::Window, sender: app::Sender<Message>) -> B
 
 
             let mut start_bt = button::Button::default()
-                .with_label("Start");
+                .with_label("Start Download");
             start_bt.set_color(Color::White);
             start_bt.emit(buffer.sender, Message::Start);