From 38b9b911e2f95945b2544a8479e3e7ccea07c3f7 Mon Sep 17 00:00:00 2001
From: Julio Biason <j.biason@engys.com>
Date: Tue, 23 Aug 2022 13:47:01 -0300
Subject: [PATCH] Try to load cases in parallel

---
 axumtest/src/cases/mod.rs | 33 +++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/axumtest/src/cases/mod.rs b/axumtest/src/cases/mod.rs
index 08a2b12..5e19bb1 100644
--- a/axumtest/src/cases/mod.rs
+++ b/axumtest/src/cases/mod.rs
@@ -11,6 +11,7 @@ use futures::StreamExt;
 use mongodb::bson::doc;
 use mongodb::bson::Document;
 use mongodb::options::FindOptions;
+use tokio::task::JoinHandle;
 
 use crate::auth;
 use crate::State;
@@ -34,21 +35,29 @@ pub fn router(state: Arc<State>, ci_usr: String, ci_pwd: String, ci_role: String
 }
 
 async fn all_cases_on_collection(Path(collection): Path<String>, state: Arc<State>) -> String {
-    let collection = state.db.collection::<Document>(&collection);
+    let collection = Arc::new(state.db.collection::<Document>(&collection));
     let options = FindOptions::builder()
         .projection(doc! { "caseID": 1 })
         .build();
-    let cursor = collection.find(None, options).await.unwrap();
-
-    let result = cursor
-        .map(|r| r.unwrap().get_str("caseID").unwrap().to_lowercase())
-        .collect::<Vec<String>>()
-        .await;
-    // while let Some(record) = cursor.try_next().await.unwrap() {
-    //     tracing::debug!("{:?}", record.get_str("caseID").unwrap());
-    // }
-
-    tracing::debug!("Cases: {:?}", result);
+    let mut cursor = collection.find(None, options).await.unwrap();
+    let mut calls = Vec::new();
+
+    while let Some(record) = cursor.try_next().await.unwrap() {
+        let name = record.get_str("caseID").unwrap();
+        let filter = doc! { "caseID": name };
+        let task_collection = Arc::clone(&collection);
+        let task = tokio::task::spawn(async move {
+            let record = task_collection.find_one(filter, None).await.unwrap();
+            record
+        });
+
+        calls.push(task);
+    }
+
+    for handle in calls {
+        let record = handle.await;
+        tracing::debug!("Case: {:?}", record);
+    }
 
     format!("Cases!")
 }