aldrin-core 0.13.0

Shared core components of Aldrin, a message bus for service-oriented RPC and interprocess communication.
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

use crate::buf_ext::BufMutExt;
use crate::{SerializeError, ValueKind};
use bytes::{BufMut, BytesMut};

#[derive(Debug)]
pub struct Bytes1Serializer<'a> {
    buf: &'a mut BytesMut,
    num_elems: u32,
}

impl<'a> Bytes1Serializer<'a> {
    pub(super) fn new(buf: &'a mut BytesMut, num_elems: usize) -> Result<Self, SerializeError> {
        if num_elems <= u32::MAX as usize {
            buf.put_discriminant_u8(ValueKind::Bytes1);
            buf.put_varint_u32_le(num_elems as u32);

            Ok(Self {
                buf,
                num_elems: num_elems as u32,
            })
        } else {
            Err(SerializeError::Overflow)
        }
    }

    pub fn remaining_elements(&self) -> usize {
        self.num_elems as usize
    }

    pub fn requires_additional_elements(&self) -> bool {
        self.num_elems > 0
    }

    pub fn serialize(&mut self, bytes: &[u8]) -> Result<&mut Self, SerializeError> {
        if bytes.len() <= self.num_elems as usize {
            self.num_elems -= bytes.len() as u32;
            self.buf.put_slice(bytes);
            Ok(self)
        } else {
            Err(SerializeError::TooManyElements)
        }
    }

    pub fn finish(self) -> Result<(), SerializeError> {
        if self.num_elems == 0 {
            Ok(())
        } else {
            Err(SerializeError::TooFewElements)
        }
    }
}

#[derive(Debug)]
pub struct Bytes2Serializer<'a> {
    buf: &'a mut BytesMut,
}

impl<'a> Bytes2Serializer<'a> {
    pub(super) fn new(buf: &'a mut BytesMut) -> Result<Self, SerializeError> {
        buf.put_discriminant_u8(ValueKind::Bytes2);
        Ok(Self { buf })
    }

    pub fn serialize(&mut self, bytes: &[u8]) -> Result<&mut Self, SerializeError> {
        if !bytes.is_empty() {
            if bytes.len() <= u32::MAX as usize {
                self.buf.put_varint_u32_le(bytes.len() as u32);
                self.buf.put_slice(bytes);
                Ok(self)
            } else {
                Err(SerializeError::TooManyElements)
            }
        } else {
            Ok(self)
        }
    }

    pub fn finish(self) -> Result<(), SerializeError> {
        self.buf.put_varint_u32_le(0);
        Ok(())
    }
}