From d866fa37a6c30ad443f31f915c88e3a32927a0fb Mon Sep 17 00:00:00 2001
From: Zynh Ludwig <zynh0722@gmail.com>
Date: Wed, 28 Aug 2024 19:10:05 -0700
Subject: [PATCH] refactoring main

---
 src/main.rs | 26 +++++++++++++++-----------
 1 file changed, 15 insertions(+), 11 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index d562410..b501ef8 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -85,17 +85,7 @@ async fn main() -> io::Result<()> {
         .layer(TraceLayer::new_for_http())
         .layer(middleware::from_fn(log_source));
 
-    // Server creation
-    let addr = SocketAddr::from(([0, 0, 0, 0], 3000));
-    let listener = tokio::net::TcpListener::bind(&addr).await.unwrap();
-    axum::serve(
-        listener,
-        app.into_make_service_with_connect_info::<SocketAddr>(),
-    )
-    .await
-    .unwrap();
-
-    tracing::debug!("listening on http://{}/", addr);
+    serve(app).await;
 
     Ok(())
 }
@@ -321,3 +311,17 @@ async fn download(
 
     Ok(Redirect::to("/404.html").into_response())
 }
+
+async fn serve(app: Router) {
+    // // Server creation
+    let addr = SocketAddr::from(([0, 0, 0, 0], 3000));
+    let listener = tokio::net::TcpListener::bind(&addr).await.unwrap();
+    axum::serve(
+        listener,
+        app.into_make_service_with_connect_info::<SocketAddr>(),
+    )
+    .await
+    .unwrap();
+
+    tracing::debug!("listening on http://{}/", addr);
+}