use actix_web::{web, HttpResponse}; use actix_web::{get, put}; use serde::Deserialize; use crate::error::{Error, ResponseError}; use crate::helpers::Authentication; use crate::Data; pub fn services(cfg: &mut web::ServiceConfig) { cfg.service(get_health).service(change_healthyness); } #[get("/health")] async fn get_health(data: web::Data) -> Result { let reader = data.db.main_read_txn()?; if let Ok(Some(_)) = data.db.get_health(&reader) { return Err(Error::Maintenance.into()); } Ok(HttpResponse::Ok().finish()) } async fn set_healthy(data: web::Data) -> Result { data.db.main_write(|w| data.db.set_healthy(w))?; Ok(HttpResponse::Ok().finish()) } async fn set_unhealthy(data: web::Data) -> Result { data.db.main_write(|w| data.db.set_unhealthy(w))?; Ok(HttpResponse::Ok().finish()) } #[derive(Deserialize, Clone)] struct HealthBody { health: bool, } #[put("/health", wrap = "Authentication::Private")] async fn change_healthyness( data: web::Data, body: web::Json, ) -> Result { if body.health { set_healthy(data).await } else { set_unhealthy(data).await } }