krpc-core 0.2.0

RPC framework for service registration and discovery through API exposure, compatible with Dubbo3 protocol, intertunable with Java projects
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
98

use bytes::BufMut;
use prost::Message;

#[allow(clippy::derive_partial_eq_without_eq)]
#[derive(Clone, PartialEq, ::prost::Message)]
pub struct TripleRequestWrapper {
    /// hessian4
    /// json
    #[prost(string, tag = "1")]
    pub serialize_type: ::prost::alloc::string::String,
    #[prost(bytes = "vec", repeated, tag = "2")]
    pub args: ::prost::alloc::vec::Vec<::prost::alloc::vec::Vec<u8>>,
    #[prost(string, repeated, tag = "3")]
    pub arg_types: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
}
#[allow(clippy::derive_partial_eq_without_eq)]
#[derive(Clone, PartialEq, ::prost::Message)]
pub struct TripleResponseWrapper {
    #[prost(string, tag = "1")]
    pub serialize_type: ::prost::alloc::string::String,
    #[prost(bytes = "vec", tag = "2")]
    pub data: ::prost::alloc::vec::Vec<u8>,
    #[prost(string, tag = "3")]
    pub r#type: ::prost::alloc::string::String,
}
#[allow(clippy::derive_partial_eq_without_eq)]
#[derive(Clone, PartialEq, ::prost::Message)]
pub struct TripleExceptionWrapper {
    #[prost(string, tag = "1")]
    pub language: ::prost::alloc::string::String,
    #[prost(string, tag = "2")]
    pub serialization: ::prost::alloc::string::String,
    #[prost(string, tag = "3")]
    pub class_name: ::prost::alloc::string::String,
    #[prost(bytes = "vec", tag = "4")]
    pub data: ::prost::alloc::vec::Vec<u8>,
}


impl TripleRequestWrapper {
    pub fn get_buf(strs: Vec<String>) -> Vec<u8> {        
        let mut trip = TripleRequestWrapper::default();
        trip.serialize_type = "fastjson".to_string();
        trip.args = vec![];
        for str in strs {
            trip.args.push(str.as_bytes().to_vec());
        }
        return get_buf(trip.encode_to_vec());
    }
    pub fn get_req(self)-> Vec<String> {
        let mut res = vec![];
        for str in self.args {
            res.push(String::from_utf8(str).unwrap());
        }
        return res;
    }
}

impl TripleResponseWrapper {
    pub fn get_buf(strs: String) -> Vec<u8> {
        let mut trip = TripleResponseWrapper::default();
        trip.serialize_type = "fastjson".to_string();
        trip.data = strs.as_bytes().to_vec();
        return get_buf(trip.encode_to_vec());
    }
    pub fn is_empty_body(&self) -> bool {
        return self.data.starts_with("null".as_bytes());
    } 
}

impl TripleExceptionWrapper {
    pub fn get_buf(strs: String) -> Vec<u8> {
        let mut trip = TripleExceptionWrapper::default();
        trip.serialization = "fastjson".to_string();
        trip.data = strs.as_bytes().to_vec();
        return get_buf(trip.encode_to_vec());
    }
    pub fn get_err_info(&self) -> String {
        return serde_json::from_slice(&self.data[..]).unwrap_or("rpc error".to_owned());
    }
}


fn get_buf(data : Vec<u8>) -> Vec<u8>{
    let mut len = data.len();
    let mut u8_array = [0 as u8;4];
    for idx in 0..4 {
        u8_array[idx] = len as u8 | 0;
        len >>= 8;
    }
    let mut buf = bytes::BytesMut::default();
    buf.put_u8(0);
    for item in u8_array.iter().rev() {
        buf.put_u8(*item);
    }
    buf.put_slice(&data);
    return buf.to_vec();
}