Connects to mongo, and tracing

axumtest/Cargo.toml

@@ -7,4 +7,9 @@ edition = "2021"
 
 [dependencies]
 axum = { version = "0.5.15", features = ["headers"] }
+clap = { version = "3.2.17", features = ["derive", "env"] }
+dotenv = "0.15.0"
+mongodb = "2.3.0"
 tokio = { version = "1.20.1", features = ["macros", "rt-multi-thread"] }
+tracing = "0.1.36"
+tracing-subscriber = "0.3.15"

axumtest/src/main.rs

@@ -1,25 +1,79 @@
 mod headers;
 
-use axum::routing::get;
-use axum::routing::Router;
-use axum::TypedHeader;
+use std::sync::Arc;
+
 use headers::cipwd::CiPwd;
 use headers::cirole::CiRole;
 use headers::ciusr::CiUsr;
 
+use axum::routing::get;
+use axum::routing::Router;
+use axum::TypedHeader;
+use clap::Parser;
+use dotenv::dotenv;
+use mongodb::options::ClientOptions;
+use mongodb::Client;
+use mongodb::Database;
+use tracing_subscriber;
+
+#[derive(Parser)]
+struct Params {
+    #[clap(env = "SRV_ADDR", default_value = "0.0.0.0:3000")]
+    addr: String,
+
+    #[clap(short, long, env = "MONGO_URI")]
+    mongo_addr: String,
+}
+
+struct State {
+    db: Database,
+    collections: Vec<String>,
+}
+
 #[tokio::main]
 async fn main() {
-    let app = Router::new().route("/", get(index));
-    axum::Server::bind(&"0.0.0.0:3000".parse().unwrap())
+    dotenv().ok();
+    tracing_subscriber::fmt::init();
+
+    let args = Params::parse();
+
+    tracing::debug!("Connecting to mongo...");
+    let mongo_options = ClientOptions::parse(&args.mongo_addr).await.unwrap();
+    let client = Client::with_options(mongo_options).unwrap();
+    let db = client.database("helyxTestDatabase");
+    let collections = db.list_collection_names(None).await.unwrap();
+    tracing::debug!("Collections = {:?}", &collections);
+
+    let state = Arc::new(State { db, collections });
+
+    let app = Router::new().route("/", get(index)).route(
+        "/collections",
+        get({
+            let shared_state = Arc::clone(&state);
+            move |usr, pwd, role| get_collections(usr, pwd, role, Arc::clone(&shared_state))
+        }),
+    );
+
+    tracing::info!(args.addr, "Server listening in");
+    axum::Server::bind(&args.addr.parse().unwrap())
         .serve(app.into_make_service())
         .await
         .unwrap();
 }
 
 async fn index(
+    TypedHeader(usr): TypedHeader<CiUsr>,
+    TypedHeader(_pwd): TypedHeader<CiPwd>,
+    TypedHeader(_role): TypedHeader<CiRole>,
+) -> String {
+    format!("Hellow {}", usr)
+}
+
+async fn get_collections(
     TypedHeader(usr): TypedHeader<CiUsr>,
     TypedHeader(pwd): TypedHeader<CiPwd>,
     TypedHeader(role): TypedHeader<CiRole>,
+    state: Arc<State>,
 ) -> String {
-    format!("Hellow {}", usr)
+    format!("Collections")
 }