graphwalker-restful 1.0.2

REST and WebSocket API for GraphWalker model-based testing
Documentation 
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

use std::sync::mpsc;

use axum::extract::{Path, State};
use axum::http::StatusCode;
use axum::response::Json;
use serde_json::{json, Value};
use tokio::sync::oneshot;

use crate::actor::Command;

#[derive(Clone)]
pub struct RestState {
    pub machine_tx: mpsc::Sender<Command>,
    pub default_seed: Option<u64>,
}

async fn send_command(
    tx: &mpsc::Sender<Command>,
    build: impl FnOnce(oneshot::Sender<Result<Value, String>>) -> Command,
) -> (StatusCode, Json<Value>) {
    let (reply_tx, reply_rx) = oneshot::channel();
    if tx.send(build(reply_tx)).is_err() {
        return (
            StatusCode::INTERNAL_SERVER_ERROR,
            Json(json!({"result": "nok", "error": "Machine thread unavailable"})),
        );
    }
    match reply_rx.await {
        Ok(Ok(val)) => (StatusCode::OK, Json(val)),
        Ok(Err(msg)) => (StatusCode::OK, Json(json!({"result": "nok", "error": msg}))),
        Err(_) => (
            StatusCode::INTERNAL_SERVER_ERROR,
            Json(json!({"result": "nok", "error": "Machine thread dropped"})),
        ),
    }
}

pub async fn load(State(state): State<RestState>, body: String) -> (StatusCode, Json<Value>) {
    send_command(&state.machine_tx, |reply| Command::Load {
        json_body: body,
        seed: state.default_seed,
        global_data: None,
        reply,
    })
    .await
}

pub async fn has_next(State(state): State<RestState>) -> (StatusCode, Json<Value>) {
    send_command(&state.machine_tx, |reply| Command::HasNext { reply }).await
}

pub async fn get_next(State(state): State<RestState>) -> (StatusCode, Json<Value>) {
    send_command(&state.machine_tx, |reply| Command::GetNext {
        verbose: false,
        reply,
    })
    .await
}

pub async fn get_data(State(state): State<RestState>) -> (StatusCode, Json<Value>) {
    send_command(&state.machine_tx, |reply| Command::GetData { reply }).await
}

pub async fn set_data(
    State(state): State<RestState>,
    Path(script): Path<String>,
) -> (StatusCode, Json<Value>) {
    send_command(&state.machine_tx, |reply| Command::SetData { script, reply }).await
}

pub async fn restart(State(state): State<RestState>) -> (StatusCode, Json<Value>) {
    send_command(&state.machine_tx, |reply| Command::Restart { reply }).await
}

pub async fn get_statistics(State(state): State<RestState>) -> (StatusCode, Json<Value>) {
    send_command(&state.machine_tx, |reply| Command::GetStatistics { reply }).await
}