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

//! helpers related to collection types

use std::collections::BTreeSet;

use crate::{NullableOffsetMarker, OffsetMarker};

/// A helper trait for array-like fields, where we need to know
/// the length in order to populate another field.
pub trait HasLen {
    fn len(&self) -> usize;

    fn checked_len<T: TryFrom<usize>>(&self) -> Result<T, <T as TryFrom<usize>>::Error> {
        self.len().try_into()
    }

    fn is_empty(&self) -> bool {
        self.len() == 0
    }
}

impl<T> HasLen for [T] {
    fn len(&self) -> usize {
        self.len()
    }
}

impl<T> HasLen for BTreeSet<T> {
    fn len(&self) -> usize {
        self.len()
    }
}

impl<T> HasLen for Vec<T> {
    fn len(&self) -> usize {
        self.len()
    }
}

impl<T: HasLen> HasLen for Option<T> {
    fn len(&self) -> usize {
        match &self {
            Some(t) => t.len(),
            None => 0,
        }
    }
}

impl<T: HasLen, const N: usize> HasLen for OffsetMarker<T, N> {
    fn len(&self) -> usize {
        T::len(self)
    }
}

impl<T: HasLen, const N: usize> HasLen for NullableOffsetMarker<T, N> {
    fn len(&self) -> usize {
        self.as_ref().map(HasLen::len).unwrap_or(0)
    }
}