trapeze 0.7.1

A rust implementation of ttrpc
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

use prost::bytes::{Buf, BufMut};

use super::{DecodeError, EncodeError};

pub trait BufMutExt: BufMut {
    fn ensure_capacity(&self, required: usize) -> Result<(), EncodeError> {
        let capacity = self.remaining_mut();
        if capacity < required {
            return Err(EncodeError::InsuficientCapacity { required, capacity });
        }
        Ok(())
    }
}

impl<B: BufMut> BufMutExt for B {}

pub trait BufExt: Buf {
    fn ensure_empty(&self) -> Result<(), DecodeError> {
        let remaining = self.remaining();
        if remaining != 0 {
            return Err(DecodeError::RemainingBytes(remaining));
        }
        Ok(())
    }

    fn ensure_remaining(&self, required: usize) -> Result<(), DecodeError> {
        let remaining = self.remaining();
        if remaining < required {
            return Err(DecodeError::UnexpectedEof);
        }
        Ok(())
    }
}

impl<B: Buf> BufExt for B {}