Worked on cli

Cargo.toml

@@ -9,7 +9,9 @@ edition = "2021"
 actix = "0.13.1"
 actix-web = "4.4.0"
 anyhow = "1.0.75"
+cfg-if = "1.0.0"
 chrono = { version = "0.4.31", features = ["serde"] }
+clap = { version = "4.4.4", features = ["string", "env", "derive", "debug"] }
 httptest = "0.15.4"
 hyper = { version = "0.14.27", features = ["full"] }
 hyper-tls = "0.5.0"
@@ -23,4 +25,9 @@ tokio = { version = "1.32.0", features = ["full"] }
 uuid = { version = "1.4.1", features = ["serde", "v4"] }
 
 [profile.dev]
-opt-level = 3 
+opt-level = 3
+
+[features]
+default = ["web"]
+web = []
+cli = []

src/cli.rs

@@ -0,0 +1,22 @@
+use clap::Parser;
+
+/// Simple program to greet a person
+#[derive(Parser, Debug)]
+#[command(author, version, about, long_about = None)]
+struct Args {
+    /// Name of the person to greet
+    #[arg(short, long)]
+    name: String,
+
+    /// Number of times to greet
+    #[arg(short, long, default_value_t = 1)]
+    count: u8,
+}
+
+pub fn run() {
+    let args = Args::parse();
+
+    for _ in 0..args.count {
+        println!("Hello {}!", args.name)
+    }
+}

src/main.rs

@@ -1,36 +1,20 @@
-use actix_web::{web, App, HttpServer};
-use routes::{create_test, get_test};
-use simple_logger::SimpleLogger;
-
 mod benchmark;
+mod cli;
 mod database;
-mod routes;
 mod utils;
 
+mod routes;
+mod web;
+
 #[tokio::main(flavor = "multi_thread")]
 async fn main() -> std::io::Result<()> {
-    match SimpleLogger::new()
-        .with_level(log::LevelFilter::Info)
-        .init()
-    {
-        Err(err) => {
-            println!("failed to setup logging: {:?}", err);
+    cfg_if::cfg_if! {
+        if #[cfg(target_feature = "cli")] {
+            cli::run();
+        }else {
+            web::run_web_app().await;
         }
-        _ => {}
     }
 
-    let connection = match database::setup().await {
-        Ok(c) => c,
-        Err(err) => panic!("failed to setup database: {:?}", err),
-    };
-
-    HttpServer::new(move || {
-        App::new()
-            .service(get_test)
-            .service(create_test)
-            .app_data(web::Data::new(connection.clone()))
-    })
-    .bind(("0.0.0.0", 3000))?
-    .run()
-    .await
+    Ok(())
 }

src/web.rs

@@ -0,0 +1,35 @@
+use crate::database;
+use crate::routes::{create_test, get_test};
+use actix_web::{web, App, HttpServer};
+use simple_logger::SimpleLogger;
+
+pub async fn run_web_app() {
+    match SimpleLogger::new()
+        .with_level(log::LevelFilter::Info)
+        .init()
+    {
+        Err(err) => {
+            println!("failed to setup logging: {:?}", err);
+        }
+        _ => {}
+    }
+
+    let connection = match database::setup().await {
+        Ok(c) => c,
+        Err(err) => panic!("failed to setup database: {:?}", err),
+    };
+
+    let server = match HttpServer::new(move || {
+        App::new()
+            .service(get_test)
+            .service(create_test)
+            .app_data(web::Data::new(connection.clone()))
+    })
+    .bind(("0.0.0.0", 3000))
+    {
+        Ok(s) => s,
+        Err(err) => panic!("failed to create server: {:?}", err),
+    };
+
+    server.run().await.unwrap();
+}