cowbytes 0.1.1

Like Cow<[u8]>, but with reference-counted bytes::Bytes instead of an owned Vec<u8>
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

//! [`Buf`] for [`CowBytes`].

use bytes::{Buf, Bytes};

use crate::CowBytes;

/// Reads the bytes as a [`Buf`], so a [`CowBytes`] can be handed to buffer-generic code without
/// first copying a borrow into a [`Bytes`].
impl Buf for CowBytes<'_> {
    #[inline]
    fn remaining(&self) -> usize {
        self.len()
    }

    #[inline]
    fn chunk(&self) -> &[u8] {
        self.as_slice()
    }

    #[inline]
    fn advance(&mut self, cnt: usize) {
        match self {
            Self::Borrowed(bytes) => *bytes = &bytes[cnt..],
            Self::Shared(bytes) => bytes.advance(cnt),
        }
    }

    #[inline]
    fn copy_to_bytes(&mut self, len: usize) -> Bytes {
        match self {
            // Only a borrow has to be copied; the default implementation would copy either.
            Self::Borrowed(bytes) => {
                let (head, tail) = bytes.split_at(len);
                *bytes = tail;
                Bytes::copy_from_slice(head)
            }
            Self::Shared(bytes) => bytes.copy_to_bytes(len),
        }
    }
}