neon/compute_tools/src/http/routes/check_writability.rs

use std::sync::Arc;

use axum::{extract::State, response::Response};
use compute_api::responses::ComputeStatus;
use http::StatusCode;

use crate::{checker::check_writability, compute::ComputeNode, http::JsonResponse};

/// Check that the compute is currently running.
pub(in crate::http) async fn is_writable(State(compute): State<Arc<ComputeNode>>) -> Response {
    let status = compute.get_status();
    if status != ComputeStatus::Running {
        return JsonResponse::invalid_status(status);
    }

    match check_writability(&compute).await {
        Ok(_) => JsonResponse::success(StatusCode::OK, true),
        Err(e) => JsonResponse::error(StatusCode::INTERNAL_SERVER_ERROR, e),
    }
}