corgi 0.0.3

🦀 Rust based RPC library on UDP
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

use bytes::Bytes;
use std::{any::TypeId, collections::HashMap, sync::Arc};

use futures::future::BoxFuture;

use crate::protocol::{codec::ProtobufCodec, types::RpcError};

#[derive(Debug, Clone)]
pub struct Param {
    pub name: &'static str,
    pub type_id: TypeId,
}

type Handler =
    dyn Fn(Vec<Bytes>, ProtobufCodec) -> BoxFuture<'static, Result<Bytes, RpcError>> + Send + Sync;

#[derive(Clone)]
pub struct RpcFunction {
    pub name: &'static str,
    pub params: Vec<Param>,
    pub return_type: Option<TypeId>,
    pub handler: Arc<Handler>,
}

#[derive(Default)]
pub struct Container {
    functions: HashMap<&'static str, &'static RpcFunction>,
}

impl Container {
    pub fn register(&mut self, function: &'static RpcFunction) {
        self.functions.entry(function.name).or_insert(function);
    }

    pub fn find(&self, name: &str) -> Option<&'static RpcFunction> {
        self.functions.get(name).copied()
    }
}