columnar/
arc.rs

1//! Implementations of traits for `Arc<T>`
2use std::sync::Arc;
3
4use crate::{Len, Borrow, HeapSize, AsBytes, FromBytes};
5
6impl<T: Borrow> Borrow for Arc<T> {
7    type Ref<'a> = T::Ref<'a> where T: 'a;
8    type Borrowed<'a> = T::Borrowed<'a>;
9    #[inline(always)] fn borrow<'a>(&'a self) -> Self::Borrowed<'a> { self.as_ref().borrow() }
10    #[inline(always)] fn reborrow<'b, 'a: 'b>(item: Self::Borrowed<'a>) -> Self::Borrowed<'b> where Self: 'a { T::reborrow(item) }
11    #[inline(always)] fn reborrow_ref<'b, 'a: 'b>(item: Self::Ref<'a>) -> Self::Ref<'b> where Self: 'a { T::reborrow_ref(item) }
12}
13impl<T: Len> Len for Arc<T> {
14    #[inline(always)] fn len(&self) -> usize { self.as_ref().len() }
15}
16impl<T: HeapSize> HeapSize for Arc<T> {
17    fn heap_size(&self) -> (usize, usize) {
18        let (l, c) = self.as_ref().heap_size();
19        (l + std::mem::size_of::<Arc<T>>(), c + std::mem::size_of::<Arc<T>>())
20    }
21}
22impl<'a, T: AsBytes<'a>> AsBytes<'a> for Arc<T> {
23    #[inline(always)] fn as_bytes(&self) -> impl Iterator<Item=(u64, &'a [u8])> { self.as_ref().as_bytes() }
24}
25impl<'a, T: FromBytes<'a>> FromBytes<'a> for Arc<T> {
26    #[inline(always)] fn from_bytes(bytes: &mut impl Iterator<Item=&'a [u8]>) -> Self { Arc::new(T::from_bytes(bytes)) }
27}
28
29#[cfg(test)]
30mod tests {
31    use std::sync::Arc;
32    use crate::{Borrow, Len, HeapSize, AsBytes, FromBytes};
33
34    #[test]
35    fn test_borrow() {
36        let x = Arc::new(vec![1, 2, 3]);
37        let y: &[i32] = x.borrow();
38        assert_eq!(y, &[1, 2, 3]);
39    }
40
41    #[test]
42    fn test_len() {
43        let x = Arc::new(vec![1, 2, 3]);
44        assert_eq!(x.len(), 3);
45    }
46
47    #[test]
48    fn test_heap_size() {
49        let x = Arc::new(vec![1, 2, 3]);
50        let (l, c) = x.heap_size();
51        assert!(l > 0);
52        assert!(c > 0);
53    }
54
55    #[test]
56    fn test_as_from_bytes() {
57        let x = Arc::new(vec![1u8, 2, 3, 4, 5]);
58        let bytes: Vec<_> = x.borrow().as_bytes().map(|(_, b)| b).collect();
59        let y: Arc<&[u8]> = FromBytes::from_bytes(&mut bytes.into_iter());
60        assert_eq!(*x, *y);
61    }
62
63    #[test]
64    fn test_borrow_tuple() {
65        let x = (vec![4,5,6,7,], Arc::new(vec![1, 2, 3]));
66        let y: (&[i32], &[i32]) = x.borrow();
67        assert_eq!(y, ([4,5,6,7].as_ref(), [1, 2, 3].as_ref()));
68    }
69}