Broadcast asynchronously

plume-common/src/activity_pub/mod.rs

@@ -1,12 +1,13 @@
 use activitypub::{Activity, Link, Object};
 use array_tool::vec::Uniq;
-use reqwest::{blocking::ClientBuilder, header::HeaderValue, Url};
+use reqwest::{header::HeaderValue, ClientBuilder, Url};
 use rocket::{
     http::Status,
     request::{FromRequest, Request},
     response::{Responder, Response},
     Outcome,
 };
+use tokio::{runtime, sync::mpsc};
 use tracing::{debug, warn};
 
 use self::sign::Signable;
@@ -139,6 +140,12 @@ where
     .connect_timeout(std::time::Duration::from_secs(5))
     .build()
     .expect("Can't build client");
+    let rt = runtime::Builder::new_current_thread()
+        .enable_all()
+        .build()
+        .expect("Error while initializing tokio runtime for federation");
+    rt.block_on(async {
+        let (tx, mut rx) = mpsc::channel(32);
         for inbox in boxes {
             let body = signed.to_string();
             let mut headers = request::headers();
@@ -159,26 +166,37 @@ where
             }
             headers.insert("Host", host_header_value.unwrap());
             headers.insert("Digest", request::Digest::digest(&body));
-        let _ = client
-            .post(&inbox)
-            .headers(headers.clone())
-            .header(
+            headers.insert(
                 "Signature",
                 request::signature(sender, &headers, ("post", url.path(), url.query()))
                     .expect("activity_pub::broadcast: request signature error"),
-            )
+            );
+            let client = client.clone();
+            let tx = tx.clone();
+            rt.spawn(async move {
+                let _ = tx.send(
+                    client
+                        .post(&inbox)
+                        .headers(headers.clone())
                         .body(body)
                         .send()
+                        .await,
+                );
+            });
+        }
+        while let Some(request) = rx.recv().await {
+            let _ = request
                 .map(move |r| {
                     if r.status().is_success() {
-                    debug!("Successfully sent activity to inbox ({})", &inbox);
+                        debug!("Successfully sent activity to inbox ({})", &r.url());
                     } else {
-                    warn!("Error while sending to inbox ({} {:?})", &inbox, &r)
+                        warn!("Error while sending to inbox ({} {:?})", &r.url(), &r)
                     }
                     debug!("Response: \"{:?}\"\n", r);
                 })
                 .map_err(|e| warn!("Error while sending to inbox ({:?})", e));
         }
+    });
 }
 
 #[derive(Shrinkwrap, Clone, Serialize, Deserialize)]