gin-tonic-core 0.7.0

core functions for gin-tonic
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113

use std::hash::Hash;

use indexmap::IndexMap;

use crate::{
    Decode, Encode, Map, RawMessageView, Scalar, Tag, decoder::Decoder, encoder::SizeHint,
    error::ProtoError, wire_types::WIRE_TYPE_LENGTH_ENCODED,
};

pub struct KeyValuePairView<'p, RustKey, RustValue> {
    pub key: &'p RustKey,
    pub value: &'p RustValue,
}

impl<'p, RustKey, RustValue> KeyValuePairView<'p, RustKey, RustValue> {
    fn size_hint<ProtobufKey, ProtobufValue>(&self) -> usize
    where
        RustKey: Scalar<ProtobufKey>,
        RustValue: Scalar<ProtobufValue>,
    {
        let mut hint = SizeHint::default();
        self.encode(&mut hint);
        hint.size()
    }

    fn encode<ProtobufKey, ProtobufValue>(&self, encoder: &mut impl Encode)
    where
        RustKey: Scalar<ProtobufKey>,
        RustValue: Scalar<ProtobufValue>,
    {
        let tag = Tag::from_parts(1u32, <RustKey as Scalar<ProtobufKey>>::WIRE_TYPE);
        encoder.encode_tag(tag);
        <RustKey as Scalar<ProtobufKey>>::encode(self.key, encoder);

        let tag = Tag::from_parts(2u32, <RustValue as Scalar<ProtobufValue>>::WIRE_TYPE);
        encoder.encode_tag(tag);
        <RustValue as Scalar<ProtobufValue>>::encode(self.value, encoder);
    }
}

pub struct KeyValuePairOwned<RustKey, RustValue> {
    pub key: RustKey,
    pub value: RustValue,
}

impl<RustKey, RustValue> KeyValuePairOwned<RustKey, RustValue> {
    pub fn decode<ProtobufKey, ProtobufValue>(buffer: &[u8]) -> Result<Self, ProtoError>
    where
        RustKey: Scalar<ProtobufKey>,
        RustValue: Scalar<ProtobufValue>,
    {
        let raw_message = RawMessageView::try_from(buffer)?;

        let tag = Tag::from_parts(1u32, <RustKey as Scalar<ProtobufKey>>::WIRE_TYPE);
        let bytes = raw_message
            .tag_data(tag)
            .next_back()
            .ok_or(ProtoError::MissingField(1))?;
        let mut decoder = Decoder::new(bytes);
        let key = <RustKey as Scalar<ProtobufKey>>::decode(&mut decoder)?;

        let tag = Tag::from_parts(2u32, <RustValue as Scalar<ProtobufValue>>::WIRE_TYPE);
        let bytes = raw_message
            .tag_data(tag)
            .next_back()
            .ok_or(ProtoError::MissingField(2))?;
        let mut decoder = Decoder::new(bytes);
        let value = <RustValue as Scalar<ProtobufValue>>::decode(&mut decoder)?;

        Ok(Self { key, value })
    }
}

impl<RustKey, ProtobufKey, RustValue, ProtobufValue> Map<ProtobufKey, ProtobufValue>
    for IndexMap<RustKey, RustValue>
where
    RustKey: Scalar<ProtobufKey> + Hash + Eq,
    RustValue: Scalar<ProtobufValue>,
{
    fn encode(&self, field_number: u32, encoder: &mut impl Encode) {
        for (key, value) in self.iter() {
            encoder.encode_tag(Tag::from_parts(field_number, WIRE_TYPE_LENGTH_ENCODED));

            let pair = KeyValuePairView { key, value };

            encoder.encode_uint64(pair.size_hint() as u64);
            pair.encode(encoder);
        }
    }

    fn decode<'buf>(
        field_number: u32,
        raw_message: &'buf RawMessageView<'buf>,
    ) -> Result<Self, ProtoError>
    where
        Self: Sized,
    {
        let mut map = Self::new();

        for buffer in raw_message.tag_data(Tag::from_parts(field_number, WIRE_TYPE_LENGTH_ENCODED))
        {
            let mut decoder = Decoder::new(buffer);
            let _size = decoder.decode_uint64()?;

            let pair = KeyValuePairOwned::<RustKey, RustValue>::decode::<ProtobufKey, ProtobufValue>(
                decoder.buffer(),
            )?;
            map.insert(pair.key, pair.value);
        }

        Ok(map)
    }
}