From 731a7aff4cd47093bf6f5da23c85209f51d9230d Mon Sep 17 00:00:00 2001
From: Reimar <mail@reim.ar>
Date: Tue, 13 Aug 2024 15:12:21 +0200
Subject: [PATCH] Add database connection to app data and run integrity check

---
 rust-backend/src/main.rs | 27 +++++++++++++++++++++++----
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/rust-backend/src/main.rs b/rust-backend/src/main.rs
index d058381..dc94b4b 100644
--- a/rust-backend/src/main.rs
+++ b/rust-backend/src/main.rs
@@ -1,13 +1,24 @@
-use actix_web::{get, Responder, HttpResponse, HttpServer, App};
+use actix_web::{get, Responder, HttpResponse, HttpServer, App, web};
+use std::sync::{Mutex, Arc};
 
 mod embedded {
     use refinery::embed_migrations;
     embed_migrations!("./migrations");
 }
 
+struct AppData {
+    database: Arc<Mutex<rusqlite::Connection>>,
+}
+
 #[get("/hc")]
-async fn healthcheck() -> impl Responder {
-    HttpResponse::Ok().body("OK")
+async fn healthcheck(data: web::Data<AppData>) -> impl Responder {
+    let db = data.database.lock().unwrap();
+
+    match db.pragma_query(None, "integrity_check", |_| Ok(())) {
+        Ok(_) => HttpResponse::Ok().body("OK"),
+        Err(_) => HttpResponse::InternalServerError().body("Error"),
+    }
+
 }
 
 #[actix_web::main]
@@ -22,14 +33,22 @@ async fn main() -> std::io::Result<()> {
 
     println!("Opening database: {}", database_path);
 
-    let mut conn = rusqlite::Connection::open(database_path).unwrap();
+    let mut conn = rusqlite::Connection::open(database_path.clone()).unwrap();
 
     embedded::migrations::runner().run(&mut conn).unwrap();
 
     println!("Starting web server at port {}", port);
 
     HttpServer::new(|| {
+        let database_path = std::env::var("RUST_BACKEND_DB")
+            .unwrap_or("database.sqlite3".to_string());
+
+        let conn = Arc::new(Mutex::new(rusqlite::Connection::open(database_path).unwrap()));
+
         App::new()
+            .app_data(web::Data::new(AppData {
+                database: conn,
+            }))
             .service(healthcheck)
     })
     .bind(("0.0.0.0", port))?