From 3bec1e9d4c322b74aeced9f6a6bc9f859597f6ab Mon Sep 17 00:00:00 2001
From: r0r-5chach <r0r-5chach.xyz@proton.me>
Date: Thu, 6 Jun 2024 19:25:03 +0100
Subject: [PATCH] refactor

---
 src/lib.rs  | 9 ++++++++-
 src/main.rs | 6 +-----
 2 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 1de6955..ea956cb 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -2,7 +2,14 @@ mod account;
 mod db;
 mod model;
 
-pub fn router() -> axum::Router {
+pub async fn run() {
+    let app = router();
+    let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
+
+    axum::serve(listener, app).await.unwrap();
+}
+
+fn router() -> axum::Router {
     axum::Router::new()
         .nest("/account", account::router())
         .nest("/predict", model::router())
diff --git a/src/main.rs b/src/main.rs
index 9e88f72..272213f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,8 +1,4 @@
 #[tokio::main]
 async fn main() {
-    let app = dermy_server::api::api_router();
-    let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
-
-    axum::serve(listener, app).await.unwrap();
-
+    dermy_server::run().await;
 }