src/varint.rs


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

/// Encode u64 as varint.
/// Panics if buffer length is less than 10.
#[inline]
pub fn encode_varint64(mut value: u64, buf: &mut [u8]) -> usize {
    assert!(buf.len() >= 10);

    unsafe {
        let mut i = 0;
        while (value & !0x7F) > 0 {
            *buf.get_unchecked_mut(i) = ((value & 0x7F) | 0x80) as u8;
            value >>= 7;
            i += 1;
        }
        *buf.get_unchecked_mut(i) = value as u8;
        i + 1
    }
}

/// Encode u32 value as varint.
/// Panics if buffer length is less than 5.
#[inline]
pub fn encode_varint32(mut value: u32, buf: &mut [u8]) -> usize {
    assert!(buf.len() >= 5);

    unsafe {
        let mut i = 0;
        while (value & !0x7F) > 0 {
            *buf.get_unchecked_mut(i) = ((value & 0x7F) | 0x80) as u8;
            value >>= 7;
            i += 1;
        }
        *buf.get_unchecked_mut(i) = value as u8;
        i + 1
    }
}