feat: add option to skip queries in benchmark

Author: michaelmior

src/bench_utils.rs

@@ -26,6 +26,9 @@ pub struct Args {
 
     #[arg(long, num_args = 0..=1, default_missing_value = "profile.pb")]
     pub profile: Option<String>,
+
+    #[arg(long, default_value_t = false)]
+    pub skip_queries: bool,
 }
 
 #[derive(Clone)]

src/bin/bench_runner.rs

@@ -85,6 +85,11 @@ async fn main() {
         return;
     }
 
+    if args.skip_queries {
+        println!("Skipping queries...");
+        return;
+    }
+
     // 3. Load Queries
     let queries = Arc::new(load_queries());
     if queries.is_empty() {

src/bin/bench_runner_mongo.rs

@@ -79,6 +79,11 @@ async fn main() {
         return;
     }
 
+    if args.skip_queries {
+        println!("Skipping queries...");
+        return;
+    }
+
     let queries = Arc::new(load_queries());
     let ctx = Arc::new(db);
 

src/bin/bench_runner_server.rs

@@ -161,6 +161,11 @@ async fn main() {
         return;
     }
 
+    if args.skip_queries {
+        println!("Skipping queries...");
+        return;
+    }
+
     // 4. Establish connections for benchmark
     let mut clients = Vec::new();
     let mut _handles = Vec::new();