48 lines
1.3 KiB
Rust
48 lines
1.3 KiB
Rust
use actix_web::{web, HttpResponse};
|
|
use actix_web_macros::{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", wrap = "Authentication::Private")]
|
|
async fn get_health(data: web::Data<Data>) -> Result<HttpResponse, ResponseError> {
|
|
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<Data>) -> Result<HttpResponse, ResponseError> {
|
|
data.db.main_write(|w| data.db.set_healthy(w))?;
|
|
Ok(HttpResponse::Ok().finish())
|
|
}
|
|
|
|
async fn set_unhealthy(data: web::Data<Data>) -> Result<HttpResponse, ResponseError> {
|
|
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<Data>,
|
|
body: web::Json<HealthBody>,
|
|
) -> Result<HttpResponse, ResponseError> {
|
|
if body.health {
|
|
set_healthy(data).await
|
|
} else {
|
|
set_unhealthy(data).await
|
|
}
|
|
}
|