rpc-core 0.3.2

a tiny rpc library for rust
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

use bitflags::bitflags;
use log::error;
use serde_json::Error;

use crate::type_def::{CmdType, SeqType};

bitflags! {
pub struct MsgType: u8 {
    const Command = 1 << 0;
    const Response = 1 << 1;
    const NeedRsp = 1 << 2;
    const Ping = 1 << 3;
    const Pong = 1 << 4;
    const NoSuchCmd = 1 << 5;
}
}

pub struct MsgWrapper {
    pub seq: SeqType,
    pub type_: MsgType,
    pub cmd: CmdType,
    pub data: Vec<u8>,
    pub request_payload: Option<Vec<u8>>,
}

impl MsgWrapper {
    pub fn new() -> Self {
        Self {
            seq: 0,
            type_: MsgType::Command,
            cmd: "".to_string(),
            data: vec![],
            request_payload: None,
        }
    }

    pub fn dump(&self) -> String {
        format!(
            "seq:{}, type:{}, cmd:{}",
            self.seq,
            self.type_.bits(),
            self.cmd
        )
    }

    pub fn unpack_as<'a, T>(&'a self) -> Result<T, Error>
    where
        T: serde::Deserialize<'a>,
    {
        let r = serde_json::from_slice::<T>(self.data.as_slice());
        if r.is_err() {
            error!("deserialize error, msg info:{}", self.dump());
        }
        r
    }

    pub fn make_rsp<R>(seq: SeqType, rsp: R) -> MsgWrapper
    where
        R: serde::Serialize,
    {
        let mut msg = MsgWrapper::new();
        msg.type_ = MsgType::Response;
        msg.seq = seq;
        msg.data = serde_json::to_string(&rsp).unwrap().into_bytes().into();
        msg
    }
}