Run fetchers in parallel in MultiFetcher.first_available

Signed-off-by: Jiahao XU <Jiahao_XU@outlook.com>

src/fetchers.rs

@@ -4,6 +4,7 @@ use std::sync::Arc;
 pub use gh_crate_meta::*;
 pub use log::debug;
 pub use quickinstall::*;
+use tokio::task::JoinHandle;
 
 use crate::{BinstallError, PkgFmt, PkgMeta};
 
@@ -53,22 +54,53 @@ impl MultiFetcher {
         self.fetchers.push(fetcher);
     }
 
-    pub async fn first_available(&self) -> Option<&dyn Fetcher> {
+    pub async fn first_available(&self) -> Option<Arc<dyn Fetcher>> {
-        for fetcher in &self.fetchers {
+        let handles: Vec<_> = self
-            let available = fetcher.check().await.unwrap_or_else(|err| {
+            .fetchers
-                debug!(
+            .iter()
-                    "Error while checking fetcher {}: {}",
+            .cloned()
-                    fetcher.source_name(),
+            .map(|fetcher| {
-                    err
+                let fetcher_cloned = fetcher.clone();
-                );
-                false
-            });
 
-            if available {
+                (
-                return Some(&**fetcher);
+                    AutoAbortJoinHandle(tokio::spawn(async move {
+                        fetcher.check().await.unwrap_or_else(|err| {
+                            debug!(
+                                "Error while checking fetcher {}: {}",
+                                fetcher.source_name(),
+                                err
+                            );
+                            false
+                        })
+                    })),
+                    fetcher_cloned,
+                )
+            })
+            .collect();
+
+        for (mut handle, fetcher) in handles {
+            match (&mut handle.0).await {
+                Ok(true) => return Some(fetcher),
+                Err(join_err) => {
+                    debug!(
+                        "Error while checking fetcher {}: {}",
+                        fetcher.source_name(),
+                        join_err
+                    );
+                }
+                _ => (),
             }
         }
 
         None
     }
 }
+
+#[derive(Debug)]
+struct AutoAbortJoinHandle(JoinHandle<bool>);
+
+impl Drop for AutoAbortJoinHandle {
+    fn drop(&mut self) {
+        self.0.abort();
+    }
+}

src/main.rs

@@ -234,7 +234,7 @@ async fn entry() -> Result<()> {
         Some(fetcher) => {
             install_from_package(
                 binaries,
-                fetcher,
+                &*fetcher,
                 install_path,
                 meta,
                 opts,