Merge pull request #31 from Coduck-Team/28-rocket-axum-replacement

Replace framework 'rocket' with 'axum'

Author: utilForever

.env

@@ -1 +1 @@
-DATABASE_URL="postgres://postgres:password@localhost:5432/bibimbap"
+DATABASE_URL="postgres://postgres:password@localhost:5432/coduck"

Cargo.toml

@@ -1,5 +1,5 @@
 [package]
-name = "bibimbap-backend"
+name = "coduck-backend"
 version = "0.1.0"
 edition = "2021"
 
@@ -8,7 +8,11 @@ path = "src/lib.rs"
 
 [[bin]]
 path = "src/main.rs"
-name = "bibimbap-backend"
+name = "coduck-backend"
 
 [dependencies]
-rocket = "0.5.1"
+axum = "0.8.4"
+tokio = { version = "1.45.1", features = ["full"] }
+
+[dev-dependencies]
+reqwest = { version = "0.12.19", features = ["json", "rustls-tls"] }

README.md

@@ -1,2 +1,3 @@
-# Bibimbap
-A platform for creation of programming contest problems heavily inspired by Polygon
+# Coduck
+
+A platform for the creation of programming contest problems heavily inspired by Polygon

scripts/init_db.sh

@@ -18,7 +18,7 @@ fi
 
 DB_USER=${POSTGRES_USER:=postgres}
 DB_PASSWORD="${POSTGRES_PASSWORD:=password}"
-DB_NAME="${POSTGRES_DB:=bibimbap}"
+DB_NAME="${POSTGRES_DB:=coduck}"
 DB_PORT="${POSTGRES_PORT:=5432}"
 DB_HOST="${POSTGRES_HOST:=localhost}"
 

src/lib.rs

@@ -1,13 +1,9 @@
-#[macro_use]
-extern crate rocket;
+use axum::{routing::get, Router};
 
-use rocket::{Build, Rocket};
-
-#[get("/health_check")]
-fn health_check() {
-    ()
+async fn health_check() -> &'static str {
+    "OK"
 }
 
-pub fn run() -> Rocket<Build> {
-    rocket::build().mount("/", routes![health_check])
+pub fn build_router() -> Router {
+    Router::new().route("/health", get(health_check))
 }

src/main.rs

@@ -1,9 +1,12 @@
-use bibimbap_backend::run;
-
-#[rocket::main]
+#[tokio::main]
 async fn main() {
-    if let Err(e) = run().launch().await {
-        println!("Failed to launch Rocket: {e}");
-        drop(e);
-    }
+    let app = coduck_backend::build_router();
+
+    let listener = tokio::net::TcpListener::bind("127.0.0.1:8080")
+        .await
+        .unwrap();
+    println!("Listening on {}", listener.local_addr().unwrap());
+    axum::serve(listener, app)
+        .await
+        .expect("Server failed to start");
 }

tests/health_check.rs

@@ -1,20 +1,23 @@
-#[macro_use]
-extern crate rocket;
+#[tokio::test]
+async fn health_check_works() {
+    let listener = tokio::net::TcpListener::bind("127.0.0.1:0").await.unwrap();
+    let port = listener.local_addr().unwrap().port();
 
-use rocket::http::Status;
-use rocket::local::blocking::Client;
-use rocket::{Build, Rocket};
-use std::io::Read;
+    tokio::spawn(async move {
+        let app = coduck_backend::build_router();
+        axum::serve(listener, app)
+            .await
+            .expect("Server failed to start");
+    });
 
-fn spawn_app() -> Rocket<Build> {
-    bibimbap_backend::run()
-}
-
-#[test]
-fn health_check_works() {
-    let client = Client::tracked(spawn_app()).expect("Failed to create client");
-    let response = client.get(uri!("/health_check")).dispatch();
+    let client = reqwest::Client::new();
+    let response = client
+        .get(&format!("http://127.0.0.1:{port}/health"))
+        .send()
+        .await
+        .unwrap();
 
-    assert_eq!(response.status(), Status::Ok);
-    assert_eq!(response.bytes().count(), 0);
+    assert_eq!(response.status(), reqwest::StatusCode::OK);
+    let body = response.text().await.unwrap();
+    assert_eq!(body, "OK");
 }