scattered-collect 0.1.0

Link-time collections for Rust (distributed slices, registries)
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

//! A collection of sized items that collected into a slice in an arbitrary
//! order.

use link_section::TypedSection;

/// A collection of sized items that collected into a slice in an arbitrary
/// order.
pub struct ScatteredSlice<T: 'static> {
    section: &'static TypedSection<T>,
}

impl<T> ScatteredSlice<T> {
    #[doc(hidden)]
    #[allow(unsafe_code)]
    pub const unsafe fn new(section: &'static TypedSection<T>) -> Self {
        Self { section }
    }
}

impl<T> ::core::ops::Deref for ScatteredSlice<T> {
    type Target = [T];
    fn deref(&self) -> &Self::Target {
        self.section
    }
}

/// Declare a scattered slice.
#[macro_export]
macro_rules! __slice {
    (@gather $(#[$meta:meta])* $vis:vis static $name:ident: $collection:ident < $ty:ty >;) => {
        #[doc(hidden)]
        #[allow(unused)]
        $vis use $crate::__slice as $name;

        #[allow(unused)]
        #[allow(non_snake_case)]
        #[doc(hidden)]
        $vis mod $name {
            $crate::__support::link_section::declarative::section!(
                #[section(typed)]
                pub static $name: $crate::__support::link_section::TypedSection<$ty>;
            );
        }

        $(#[$meta])*
        $vis static $name: $collection<$ty> = {
            unsafe { $crate::slice::ScatteredSlice::new(
                self::$name::$name.const_deref()
            ) }
        };
    };

    (@scatter [$($meta:tt)*] $(#[$imeta:meta])* $vis:vis $type:ident $name:tt: $ty:ty = $expr:expr;) => {
        $crate::__support::link_section::declarative::in_section!(
            #[in_section($($meta)*::$($meta)*)]
            $(#[$imeta])*
            $vis $type $name: $ty = $expr;
        );
    };
}

#[cfg(all(test, not(miri)))]
mod tests {
    pub use super::ScatteredSlice;

    __slice!(@gather pub static TEST_SLICE: ScatteredSlice<u32>;);
    __slice!(@scatter [TEST_SLICE] const _: u32 = 1;);
    __slice!(@scatter [TEST_SLICE] const _: u32 = 3;);
    __slice!(@scatter [TEST_SLICE] const _: u32 = 2;);

    #[test]
    fn test_scattered_slice() {
        assert_eq!(TEST_SLICE.len(), 3);
        assert!(TEST_SLICE.contains(&1));
        assert!(TEST_SLICE.contains(&2));
        assert!(TEST_SLICE.contains(&3));
    }
}