Deleting subscriptions in parallel.

dmackdev · Aug 7, 2023 · e8c5439 · e8c5439
1 parent 0c6c5c4
commit e8c5439
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/pubsubman_backend/Cargo.toml b/pubsubman_backend/Cargo.toml
@@ -7,6 +7,7 @@ edition = "2021"
 
 [dependencies]
 chrono = { workspace = true }
+futures = "0.3.28"
 futures-util = "0.3.28"
 google-cloud-googleapis = "0.10.0"
 google-cloud-pubsub = "0.18.0"

diff --git a/pubsubman_backend/src/lib.rs b/pubsubman_backend/src/lib.rs
@@ -83,15 +83,18 @@ impl Backend {
                 FrontendMessage::DeleteSubscriptions(sub_names) => {
                     let back_tx = self.back_tx.clone();
                     rt.spawn(async move {
-                        let client = create_client().await;
+                        let mut futures = vec![];
 
-                        // TODO: Do in parallel
                         for sub_name in sub_names.into_iter() {
-                            let subscription = client.subscription(&sub_name.0);
-                            let _ = subscription.delete(None).await;
-                            println!("DELETED {}", &sub_name.0);
+                            futures.push(async move {
+                                let client = create_client().await;
+                                let subscription = client.subscription(&sub_name.0);
+                                let _ = subscription.delete(None).await;
+                            });
                         }
 
+                        futures::future::join_all(futures).await;
+
                         back_tx
                             .send(BackendMessage::SubscriptionsDeleted)
                             .await