aw-server 0.8.0

A reimplementation of aw-server in Rust.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

use rocket::http::Status;
use rocket::State;
use rocket_contrib::json::{Json, JsonValue};

use aw_models::Query;

use crate::endpoints::{HttpErrorJson, ServerState};

#[post("/", data = "<query_req>", format = "application/json")]
pub fn query(
    query_req: Json<Query>,
    state: State<ServerState>,
) -> Result<JsonValue, HttpErrorJson> {
    let query_code = query_req.0.query.join("\n");
    let intervals = &query_req.0.timeperiods;
    let mut results = Vec::new();
    let datastore = endpoints_get_lock!(state.datastore);
    for interval in intervals {
        let result = match aw_query::query(&query_code, &interval, &datastore) {
            Ok(data) => data,
            Err(e) => {
                warn!("Query failed: {:?}", e);
                return Err(HttpErrorJson::new(
                    Status::InternalServerError,
                    e.to_string(),
                ));
            }
        };
        results.push(result);
    }
    Ok(json!(results))
}