Move app to its own module

Tests/shared/test_app.rs

@@ -1,4 +1,4 @@
-use briefly::{configuration::get_configuration, run};
+use briefly::{app::run, configuration::get_configuration};
 use std::net::TcpListener;
 use std::sync::Arc;
 

src/app.rs

@@ -0,0 +1,45 @@
+use crate::{
+    request_tracing::RequestSpan,
+    routes::{health_check, redirect, shorten},
+};
+use axum::routing::{get, post};
+use axum::Router;
+use sqlx::{postgres::PgPoolOptions, Pool, Postgres};
+use std::net::TcpListener;
+use std::sync::Arc;
+use tower_http::trace::TraceLayer;
+
+pub struct AppState {
+    pub db: Pool<Postgres>,
+}
+
+pub async fn run(listener: TcpListener, db_connection: &str) {
+    let pool = connect_to_database(db_connection)
+        .await
+        .expect("Failed to conect to the database");
+    let app_state = Arc::new(AppState { db: pool.clone() });
+
+    let app = app(app_state);
+    axum::Server::from_tcp(listener)
+        .unwrap()
+        .serve(app.into_make_service())
+        .await
+        .unwrap()
+}
+
+pub fn app(app_state: Arc<AppState>) -> Router {
+    Router::new()
+        .route("/health_check", get(health_check))
+        .route("/shorten", post(shorten))
+        .route("/:extension", get(redirect))
+        .layer(TraceLayer::new_for_http().make_span_with(RequestSpan))
+        .with_state(app_state)
+}
+
+async fn connect_to_database(connection_string: &str) -> Option<Pool<Postgres>> {
+    PgPoolOptions::new()
+        .max_connections(10)
+        .connect(connection_string)
+        .await
+        .ok()
+}

src/lib.rs

@@ -1,51 +1,8 @@
+pub mod app;
 pub mod app_error;
 pub mod configuration;
 pub mod key_generator;
 pub mod model;
 pub mod request_tracing;
 pub mod routes;
 pub mod schema;
-
-use axum::routing::{get, post};
-use axum::Router;
-use request_tracing::RequestSpan;
-use routes::{health_check, redirect, shorten};
-use sqlx::{postgres::PgPoolOptions, Pool, Postgres};
-use std::net::TcpListener;
-use std::sync::Arc;
-use tower_http::trace::TraceLayer;
-
-pub async fn run(listener: TcpListener, db_connection: &str) {
-    let pool = connect_to_database(db_connection)
-        .await
-        .expect("Failed to conect to the database");
-    let app_state = Arc::new(AppState { db: pool.clone() });
-
-    let app = app(app_state);
-    axum::Server::from_tcp(listener)
-        .unwrap()
-        .serve(app.into_make_service())
-        .await
-        .unwrap()
-}
-
-pub struct AppState {
-    db: Pool<Postgres>,
-}
-
-pub fn app(app_state: Arc<AppState>) -> Router {
-    Router::new()
-        .route("/health_check", get(health_check))
-        .route("/shorten", post(shorten))
-        .route("/:extension", get(redirect))
-        .layer(TraceLayer::new_for_http().make_span_with(RequestSpan))
-        .with_state(app_state)
-}
-
-async fn connect_to_database(connection_string: &str) -> Option<Pool<Postgres>> {
-    PgPoolOptions::new()
-        .max_connections(10)
-        .connect(connection_string)
-        .await
-        .ok()
-}

src/main.rs

@@ -1,4 +1,4 @@
-use briefly::{configuration::get_configuration, run};
+use briefly::{app::run, configuration::get_configuration};
 use std::net::TcpListener;
 use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
 

src/routes/redirect.rs

@@ -1,4 +1,4 @@
-use crate::{app_error::AppError, model::UrlRequestModel, AppState};
+use crate::{app::AppState, app_error::AppError, model::UrlRequestModel};
 use axum::{
     extract::{Path, State},
     response::Redirect,

src/routes/shorten.rs

@@ -1,6 +1,6 @@
 use crate::{
-    app_error::AppError, key_generator::generate, model::UrlRequestModel,
-    schema::CreateShortUrlSchema, AppState,
+    app::AppState, app_error::AppError, key_generator::generate, model::UrlRequestModel,
+    schema::CreateShortUrlSchema,
 };
 use anyhow::anyhow;
 use axum::{extract::State, Json};