hastic/server/src/api/metric.rs

use super::BadQuery;
use crate::api::API;

use hastic::services::metric_service;

use subbeat::metric::MetricResult;

use warp::filters::method::get;
use warp::Filter;
use warp::{Rejection, Reply};

use serde::Serialize;

use anyhow;

use std::collections::HashMap;

#[derive(Serialize)]
struct QueryResponse {
    message: String,
}

async fn get_query(
    p: HashMap<String, String>,
    ms: metric_service::MetricService,
) -> anyhow::Result<MetricResult> {
    if !p.contains_key("from") {
        return Err(anyhow::anyhow!("Missing attribute from"));
    }
    if !p.contains_key("to") {
        return Err(anyhow::anyhow!("Missing attribute to"));
    }
    if !p.contains_key("step") {
        return Err(anyhow::anyhow!("Missing attribute step"));
    }
    let from = p.get("from").unwrap().parse::<u64>()?;
    let to = p.get("to").unwrap().parse::<u64>()?;
    let step = p.get("step").unwrap().parse::<u64>()?;

    Ok(ms.query(from, to, step).await?)
}

async fn query(
    p: HashMap<String, String>,
    ms: metric_service::MetricService,
) -> Result<impl warp::Reply, warp::Rejection> {
    match get_query(p, ms).await {
        Ok(res) => Ok(API::json(&res)),
        // TODO: parse different error types
        Err(_e) => Err(warp::reject::custom(BadQuery)),
    }
}

pub fn get_route(
    metric_service: metric_service::MetricService,
) -> impl Filter<Extract = impl Reply, Error = Rejection> + Clone {
    return warp::path!("metric")
        .and(get())
        .and(warp::query::<HashMap<String, String>>())
        .and(warp::any().map(move || metric_service.clone()))
        .and_then(query);
}