Broke loop when test is done

emilpriver · Sep 17, 2023 · 27ee77e · 27ee77e
1 parent 359a345
commit 27ee77e
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 24 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,5 +1,5 @@
 [package]
-name = "benchmarker"
+name = "elton"
 version = "0.1.0"
 edition = "2021"
 

diff --git a/src/benchmark.rs b/src/benchmark.rs
@@ -6,11 +6,6 @@ use tokio::{
 
 use crate::routes;
 
-/**
-* TODO:
-- Report back to main thread the result of test
-*/
-
 #[derive(Debug)]
 pub struct Result {
     pub connection_id: u64,
@@ -99,9 +94,11 @@ pub async fn run_benchmark(test: routes::CreateTest) -> Vec<Result> {
     }
 
     while let Some(i) = rx.recv().await {
-        println!("{:?}", i);
-        // TODO: break this
         results.push(i);
+
+        if results.len() >= (test.connections * test.seconds) as usize {
+            break;
+        }
     }
 
     results

diff --git a/src/routes.rs b/src/routes.rs
@@ -46,7 +46,9 @@ pub async fn create_test(
     .unwrap();
 
     tokio::spawn(async move {
-        benchmark::run_benchmark(payload.clone()).await;
+        let result = benchmark::run_benchmark(payload.clone()).await;
+        //TODO: store result in db
+        println!("{:?}", result);
     });
 
     HttpResponse::Created().json(test)