diff --git a/src/server/main.rs b/src/server/main.rs index 1bb4e27..c057aa2 100644 --- a/src/server/main.rs +++ b/src/server/main.rs @@ -1,4 +1,4 @@ -use dioxus::fullstack::axum::Router; +use dioxus::{fullstack::axum::Router, server::axum::Extension}; use crate::App; use crate::server::{config, database, key_val_store}; @@ -20,7 +20,7 @@ async fn router_setup() -> Result { .map_err(|e| Error::message_here(e.to_string())) .err_context("Failed to load config")?; - let _db_pool = database::setup(config.database.connection_uri()) + let db_pool = database::setup(config.database.connection_uri()) .await .err_context("Failed database setup")?; @@ -28,6 +28,8 @@ async fn router_setup() -> Result { .await .err_context("Failed key-value store setup")?; + let router = dioxus::server::router(App).layer(Extension(db_pool)); + tracing::info!("Setup complete, returning Router..."); - Ok(dioxus::server::router(App)) + Ok(router) }