gestalt-sdk 0.0.1-alpha.12

Rust SDK scaffolding and generated protocol bindings for Gestalt executable providers
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

use tonic::Status;

use crate::Error;
use crate::error::INTERNAL_ERROR_MESSAGE;

pub(crate) fn rpc_error_message(operation: &str, error: &Error) -> String {
    if !error.expose_message() {
        eprintln!("internal error in Gestalt {operation}: {}", error.message());
        return INTERNAL_ERROR_MESSAGE.to_owned();
    }
    error.message().to_owned()
}

pub(crate) fn rpc_status(operation: &str, error: Error) -> Status {
    let message = rpc_error_message(operation, &error);
    match error.status() {
        Some(400) => Status::invalid_argument(message),
        Some(404) => Status::not_found(message),
        Some(501) => Status::unimplemented(message),
        _ => Status::unknown(format!("{operation}: {message}")),
    }
}

#[cfg(test)]
mod tests {
    use tonic::Code;

    use super::*;

    #[test]
    fn rpc_status_sanitizes_hidden_internal_errors() {
        let status = rpc_status("get cache entry", Error::hidden_internal("disk exploded"));
        assert_eq!(status.code(), Code::Unknown);
        assert_eq!(status.message(), "get cache entry: internal error");
    }

    #[test]
    fn rpc_status_preserves_explicit_errors() {
        let status = rpc_status("get cache entry", Error::bad_request("bad key"));
        assert_eq!(status.code(), Code::InvalidArgument);
        assert_eq!(status.message(), "bad key");
    }
}