page 58. launchbadge/sqlx#3589

Author: t11em

Cargo.toml

@@ -4,6 +4,7 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
+anyhow = "1.0.93"
 axum = { version = "0.7.9", features = ["macros"] }
 futures = "0.3.31"
 tokio = { version = "1.41.1", features = ["full"] }

src/main.rs

@@ -1,16 +1,79 @@
-use axum::{routing::get, Router};
+use anyhow::Result;
+use axum::{extract::State, http::StatusCode, routing::get, Router};
+use sqlx::{postgres::PgConnectOptions, PgPool};
 use std::net::{Ipv4Addr, SocketAddr};
 use tokio::net::TcpListener;
 
+struct DatabaseConfig {
+    pub host: String,
+    pub port: u16,
+    pub username: String,
+    pub password: String,
+    pub database: String,
+}
+
+impl From<DatabaseConfig> for DatabaseConfig {
+    fn from(cfg: DatabaseConfig) -> Self {
+        Self::new()
+            .host(&cfg.host)
+            .port(cfg.port)
+            .username(&cfg.username)
+            .password(&cfg.password)
+            .database(&cfg.database)
+    }
+}
+
+fn connect_database_with(cfg: DatabaseConfig) -> PgPool {
+    PgPool::connect_lazy_with(cfg.into())
+}
+
+async fn health_check() -> StatusCode {
+    StatusCode::OK
+}
+
 async fn hello_world() -> &'static str {
     "Hello, World!"
 }
 
+async fn health_check_db(state(db): State<PgPool>) -> StatusCode {
+    let connection_result = sqlx::query("SELECT 1").fetch_one(&db).await;
+    match connection_result {
+        Ok(_) => StatusCode::OK,
+        Err(_) => StatusCode::INTERNAL_SERVER_ERROR,
+    }
+}
+
 #[tokio::main]
-async fn main() {
-    let app = Router::new().route("/hello", get(hello_world));
+async fn main() -> Result<()> {
+    let database_cfg = DatabaseConfig {
+        host: "localhost".into(),
+        port: 5432,
+        username: "app".into(),
+        password: "passwd".into(),
+        database: "app".into(),
+    };
+    let conn_pool = connect_lazy_with(database_cfg);
+
+    let app = Router::new()
+        .route("/hello", get(hello_world))
+        .route("/health_check", get(health_check))
+        .route("/health_db", get(health_check_db))
+        .with_state(conn_pool);
+
     let addr = SocketAddr::new(Ipv4Addr::LOCALHOST.into(), 8080);
     let listener = TcpListener::bind(addr).await.unwrap();
     println!("Listening on {}", addr);
-    axum::serve(listener, app).await.unwrap();
+    Ok(axum::serve(listener, app).await?)
 }
+
+#[tokio::test]
+async fn health_check_words() {
+    let status_code = health_check().await;
+    assert_eq!(status_code, StatusCode::OK);
+}
+
+// #[sqlx::test]
+// async fn health_check_db_works(pool: sqlx::PgPool) {
+//     // let status_code = health_check_db(State(Pool)).await;
+//     // assert_eq!(status_code, StatusCode::OK);
+// }