proto_rs 0.11.26

Rust-first gRPC macros collection for .proto/protobufs managment and more
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

use std::sync::Mutex;

use proto_rs::DecodeError;
use proto_rs::ProtoDecode;
use proto_rs::ProtoEncode;
use proto_rs::ProtoShadowDecode;
use proto_rs::ProtoShadowEncode;
use proto_rs::encoding::DecodeContext;
use proto_rs::proto_message;

#[proto_message(sun = IdOwned)]
#[allow(dead_code)]
struct IdShadow {
    #[proto(tag = 1)]
    id: u64,
}

#[derive(Debug)]
struct IdOwned {
    id: Mutex<u64>,
}

impl PartialEq for IdOwned {
    fn eq(&self, other: &Self) -> bool {
        *self.id.lock().unwrap() == *other.id.lock().unwrap()
    }
}

impl ProtoShadowDecode<IdOwned> for IdShadow {
    fn to_sun(self) -> Result<IdOwned, DecodeError> {
        Ok(IdOwned { id: Mutex::new(self.id) })
    }
}

impl<'a> ProtoShadowEncode<'a, IdOwned> for IdShadow {
    fn from_sun(value: &'a IdOwned) -> Self {
        let guard = value.id.lock().unwrap();
        IdShadow { id: *guard }
    }
}

#[test]
fn encode_decode_reference_sun_top_level() {
    let id = IdOwned { id: Mutex::new(42) };
    let bytes = IdOwned::encode_to_vec(&id);
    let decoded = <IdOwned as ProtoDecode>::decode(bytes.as_slice(), DecodeContext::default()).expect("decode owned id");

    assert_eq!(decoded, id);
}