orpc 0.1.2

Type-safe RPC framework for Rust, inspired by oRPC
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

use std::marker::PhantomData;
use std::sync::Arc;

use orpc_procedure::{
    DynInput, ErasedProcedure, ErasedSchema, ErrorMap, Meta, ProcedureStream, Route,
};

/// Type-safe Procedure. Retains full generic information at compile time.
///
/// Generic parameters:
/// - `TBaseCtx`: Initial context required to call this procedure (entry point)
/// - `TInput`: Deserialized input type
/// - `TOutput`: Handler return type
/// - `TError`: Error type
///
/// The `exec` closure captures `TInput`/`TOutput`/`TError` internally —
/// deserialization and serialization happen inside.
pub struct Procedure<TBaseCtx, TInput, TOutput, TError> {
    pub(crate) exec: Arc<dyn Fn(TBaseCtx, DynInput) -> ProcedureStream + Send + Sync>,
    pub(crate) input_schema: Option<Box<dyn ErasedSchema>>,
    pub(crate) output_schema: Option<Box<dyn ErasedSchema>>,
    pub(crate) error_map: ErrorMap,
    pub(crate) route: Route,
    pub(crate) meta: Meta,
    pub(crate) _phantom: PhantomData<fn(TInput, TOutput, TError)>,
}

/// Conversion to `ErasedProcedure`. Delegates to `into_erased()`, preserving all metadata.
impl<TBaseCtx, TInput, TOutput, TError> From<Procedure<TBaseCtx, TInput, TOutput, TError>>
    for ErasedProcedure<TBaseCtx>
where
    TBaseCtx: Send + Sync + 'static,
{
    fn from(proc: Procedure<TBaseCtx, TInput, TOutput, TError>) -> Self {
        proc.into_erased()
    }
}

impl<TBaseCtx, TInput, TOutput, TError> Procedure<TBaseCtx, TInput, TOutput, TError>
where
    TBaseCtx: Send + Sync + 'static,
{
    /// Convert to type-erased procedure, preserving all metadata.
    pub fn into_erased(self) -> ErasedProcedure<TBaseCtx> {
        let exec = self.exec;
        let mut erased =
            ErasedProcedure::new(move |ctx, input| exec(ctx, input), self.route, self.meta)
                .with_error_map(self.error_map);
        if let Some(schema) = self.input_schema {
            erased.input_schema = Some(schema);
        }
        if let Some(schema) = self.output_schema {
            erased.output_schema = Some(schema);
        }
        erased
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn procedure_type_params_are_phantom() {
        // Verify Procedure can be created with arbitrary type params
        let _proc: Procedure<(), String, Vec<u8>, std::io::Error> = Procedure {
            exec: Arc::new(|_ctx, _input| {
                ProcedureStream::error(orpc_procedure::ProcedureError::Unwind(Box::new("test")))
            }),
            input_schema: None,
            output_schema: None,
            error_map: ErrorMap::default(),
            route: Route::default(),
            meta: Meta::default(),
            _phantom: PhantomData,
        };
    }

    #[test]
    fn into_erased_preserves_route() {
        let proc: Procedure<(), (), (), std::io::Error> = Procedure {
            exec: Arc::new(|_ctx, _input| {
                ProcedureStream::error(orpc_procedure::ProcedureError::Unwind(Box::new("test")))
            }),
            input_schema: None,
            output_schema: None,
            error_map: ErrorMap::default(),
            route: Route::get("/test").tag("api"),
            meta: Meta::default(),
            _phantom: PhantomData,
        };

        let erased = proc.into_erased();
        assert_eq!(erased.route.path.as_deref(), Some("/test"));
        assert_eq!(erased.route.tags, vec!["api"]);
    }
}