meilisearch/meilisearch-http/src/routes/health.rs

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
    }
}