aion-server 0.2.0

Deployable HTTP, gRPC, WebSocket, and worker endpoint for Aion workflows.
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

//! Engine-error-to-wire-error mapping for the shared handlers.

use aion_core::{WorkflowId, WorkflowStatus};
use aion_proto::WireError;

use crate::ServerError;

pub(super) fn map_start_error(error: aion::EngineError, workflow_type: &str) -> WireError {
    match error {
        aion::EngineError::WorkflowNotFound { .. } => WireError::not_found_with_type(
            "WorkflowTypeNotFound",
            format!("workflow type {workflow_type} is not registered"),
        ),
        other => ServerError::from(other).to_wire_error(),
    }
}

pub(super) fn map_workflow_operation_error(
    error: aion::EngineError,
    workflow_id: &WorkflowId,
) -> WireError {
    match error {
        aion::EngineError::WorkflowNotFound { .. } => workflow_not_found_error(workflow_id),
        other => ServerError::from(other).to_wire_error(),
    }
}

pub(super) fn workflow_not_found_error(workflow_id: &WorkflowId) -> WireError {
    WireError::not_found_with_type(
        "WorkflowNotFound",
        format!("workflow {workflow_id} not found"),
    )
}

pub(super) fn signal_terminal_error(workflow_id: &WorkflowId, status: WorkflowStatus) -> WireError {
    WireError::not_running_with_type(
        "WorkflowTerminal",
        format!("workflow {workflow_id} has already reached terminal state {status:?}"),
    )
}

pub(super) fn cancel_terminal_error(workflow_id: &WorkflowId, status: WorkflowStatus) -> WireError {
    WireError::not_running_with_type(
        "WorkflowTerminal",
        format!("workflow {workflow_id} has already completed with status {status:?}"),
    )
}

pub(super) fn log_server_error(
    operation: &'static str,
    namespace: Option<&str>,
    workflow_id: Option<&WorkflowId>,
    error: &ServerError,
) -> WireError {
    let fields = error.trace_fields();
    tracing::error!(
        operation,
        namespace,
        workflow_id = workflow_id.map(ToString::to_string).as_deref(),
        error_type = %fields.error_type,
        store_error_type = fields.store_error_type,
        reason = %fields.reason,
        "request handler failed"
    );
    error.to_wire_error()
}