80 lines
2.0 KiB
Rust
80 lines
2.0 KiB
Rust
use crate::error::{ResponseError, SResult};
|
|
use crate::helpers::tide::ContextExt;
|
|
use crate::models::token::ACL::*;
|
|
use crate::Data;
|
|
|
|
use heed::types::{Str, Unit};
|
|
use serde::Deserialize;
|
|
use tide::Context;
|
|
|
|
const UNHEALTHY_KEY: &str = "_is_unhealthy";
|
|
|
|
pub async fn get_health(ctx: Context<Data>) -> SResult<()> {
|
|
let db = &ctx.state().db;
|
|
let env = &db.env;
|
|
let reader = env.read_txn().map_err(ResponseError::internal)?;
|
|
|
|
let common_store = ctx.state().db.common_store();
|
|
|
|
if let Ok(Some(_)) = common_store.get::<Str, Unit>(&reader, UNHEALTHY_KEY) {
|
|
return Err(ResponseError::Maintenance);
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn set_healthy(ctx: Context<Data>) -> SResult<()> {
|
|
ctx.is_allowed(Admin)?;
|
|
|
|
let db = &ctx.state().db;
|
|
let env = &db.env;
|
|
let mut writer = env.write_txn().map_err(ResponseError::internal)?;
|
|
|
|
let common_store = ctx.state().db.common_store();
|
|
match common_store.delete::<Str>(&mut writer, UNHEALTHY_KEY) {
|
|
Ok(_) => (),
|
|
Err(e) => return Err(ResponseError::internal(e)),
|
|
}
|
|
|
|
if let Err(e) = writer.commit() {
|
|
return Err(ResponseError::internal(e));
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn set_unhealthy(ctx: Context<Data>) -> SResult<()> {
|
|
ctx.is_allowed(Admin)?;
|
|
|
|
let db = &ctx.state().db;
|
|
let env = &db.env;
|
|
let mut writer = env.write_txn().map_err(ResponseError::internal)?;
|
|
|
|
let common_store = ctx.state().db.common_store();
|
|
|
|
if let Err(e) = common_store.put::<Str, Unit>(&mut writer, UNHEALTHY_KEY, &()) {
|
|
return Err(ResponseError::internal(e));
|
|
}
|
|
|
|
if let Err(e) = writer.commit() {
|
|
return Err(ResponseError::internal(e));
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[derive(Deserialize, Clone)]
|
|
struct HealtBody {
|
|
health: bool,
|
|
}
|
|
|
|
pub async fn change_healthyness(mut ctx: Context<Data>) -> SResult<()> {
|
|
let body: HealtBody = ctx.body_json().await.map_err(ResponseError::bad_request)?;
|
|
|
|
if body.health {
|
|
set_healthy(ctx).await
|
|
} else {
|
|
set_unhealthy(ctx).await
|
|
}
|
|
}
|