editoast: add healthcheck command

Signed-off-by: ElysaSrc <[email protected]>

editoast/src/client/mod.rs

@@ -75,6 +75,8 @@ pub enum Commands {
     STDCMSearchEnv(StdcmSearchEnvCommands),
     #[command(subcommand, about, long_about = "Roles related commands")]
     Roles(RolesCommand),
+    #[command(about, long_about = "Healthcheck")]
+    Healthcheck,
 }
 
 #[derive(Subcommand, Debug)]

editoast/src/main.rs

@@ -48,6 +48,7 @@ use tower_http::cors::{Any, CorsLayer};
 use tower_http::limit::RequestBodyLimitLayer;
 use tower_http::normalize_path::NormalizePathLayer;
 use tower_http::trace::TraceLayer;
+use views::check_health;
 use views::train_schedule::{TrainScheduleForm, TrainScheduleResult};
 
 use colored::*;
@@ -266,9 +267,22 @@ async fn run() -> Result<(), Box<dyn Error + Send + Sync>> {
                     .map_err(Into::into)
             }
         },
+        Commands::Healthcheck => healthcheck_cmd(db_pool.into(), redis_config).await,
     }
 }
 
+async fn healthcheck_cmd(
+    db_pool: Arc<DbConnectionPoolV2>,
+    redis_config: RedisConfig,
+) -> Result<(), Box<dyn Error + Send + Sync>> {
+    let redis = RedisClient::new(redis_config).unwrap();
+    check_health(db_pool, redis.into())
+        .await
+        .map_err(|e| CliError::new(1, format!("❌ healthcheck failed: {0}", e)))?;
+    println!("✅ Healthcheck passed");
+    Ok(())
+}
+
 async fn trains_export(
     args: ExportTimetableArgs,
     db_pool: Arc<DbConnectionPoolV2>,

editoast/src/views/mod.rs

@@ -257,7 +257,7 @@ async fn health(
     Ok("ok")
 }
 
-async fn check_health(
+pub async fn check_health(
     db_pool: Arc<DbConnectionPoolV2>,
     redis_client: Arc<RedisClient>,
 ) -> Result<()> {