variable_len_reader/asynchronous/
read_raw.rs

1macro_rules! read_raw_future {
2    ($primitive: ty, $future: ident, $from: ident) => {
3        read_raw_future!(f cfg(feature = "async_raw"), $primitive, $future, $from);
4    };
5    (f $feature: meta, $primitive: ty, $future: ident, $from: ident) => {
6        #[$feature]
7        $crate::pin_project_lite::pin_project! {
8            #[cfg_attr(docsrs, doc($feature))]
9            #[derive(Debug)]
10            #[project(!Unpin)]
11            #[must_use = "futures do nothing unless you `.await` or poll them"]
12            pub struct $future<'a, R: ?Sized> {
13                #[pin]
14                reader: &'a mut R,
15                buf: OwnedReadBuf<[u8; ::core::mem::size_of::<$primitive>()]>,
16            }
17        }
18        #[$feature]
19        impl<'a, R: ?Sized> ReaderFuture for $future<'a, R> {
20            fn reset(self: Pin<&mut Self>) {
21                let me = self.project();
22                me.buf.reset();
23            }
24        }
25        #[$feature]
26        impl<'a, R: AsyncVariableReader + Unpin + ?Sized> Future for $future<'a, R> {
27            type Output = ::core::result::Result<$primitive, R::Error>;
28
29            fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
30                let mut me = self.project();
31                let mut ref_buf = me.buf.into();
32                let res = R::poll_read_more(Pin::new(&mut *me.reader), cx, &mut ref_buf);
33                let position = ref_buf.position();
34                me.buf.set_position(position);
35                ::core::task::ready!(res)?;
36                Poll::Ready(Ok(<$primitive>::$from(me.buf.clone().into_inner())))
37            }
38        }
39    };
40}
41macro_rules! read_raw_func {
42    ($primitive: ty, $func: ident, $future: ident) => {
43        read_raw_func!(f cfg(feature = "async_raw"), $primitive, $func, $future);
44    };
45    (f $feature: meta, $primitive: ty, $func: ident, $future: ident) => {
46        #[$feature]
47        #[cfg_attr(docsrs, doc($feature))]
48        #[inline]
49        fn $func(&mut self) -> $future<Self> where Self: Unpin {
50            $future { reader: self, buf: OwnedReadBuf::new([0; ::core::mem::size_of::<$primitive>()]) }
51        }
52    };
53}
54
55macro_rules! define_read_raw_future {
56    () => {
57        read_raw_future!(u8, ReadU8Raw, from_ne_bytes);
58        read_raw_future!(i8, ReadI8Raw, from_ne_bytes);
59
60        read_raw_future!(u16, ReadU16RawLe, from_le_bytes);
61        read_raw_future!(u16, ReadU16RawBe, from_be_bytes);
62        read_raw_future!(i16, ReadI16RawLe, from_le_bytes);
63        read_raw_future!(i16, ReadI16RawBe, from_be_bytes);
64
65        read_raw_future!(u32, ReadU32RawLe, from_le_bytes);
66        read_raw_future!(u32, ReadU32RawBe, from_be_bytes);
67        read_raw_future!(i32, ReadI32RawLe, from_le_bytes);
68        read_raw_future!(i32, ReadI32RawBe, from_be_bytes);
69
70        read_raw_future!(u64, ReadU64RawLe, from_le_bytes);
71        read_raw_future!(u64, ReadU64RawBe, from_be_bytes);
72        read_raw_future!(i64, ReadI64RawLe, from_le_bytes);
73        read_raw_future!(i64, ReadI64RawBe, from_be_bytes);
74
75        read_raw_future!(u128, ReadU128RawLe, from_le_bytes);
76        read_raw_future!(u128, ReadU128RawBe, from_be_bytes);
77        read_raw_future!(i128, ReadI128RawLe, from_le_bytes);
78        read_raw_future!(i128, ReadI128RawBe, from_be_bytes);
79
80        read_raw_future!(f32, ReadF32RawLe, from_le_bytes);
81        read_raw_future!(f32, ReadF32RawBe, from_be_bytes);
82        read_raw_future!(f64, ReadF64RawLe, from_le_bytes);
83        read_raw_future!(f64, ReadF64RawBe, from_be_bytes);
84    };
85}
86macro_rules! define_read_raw_func {
87    () => {
88        read_raw_func!(u8, read_u8_raw, ReadU8Raw);
89        read_raw_func!(i8, read_i8_raw, ReadI8Raw);
90
91        read_raw_func!(u16, read_u16_raw_le, ReadU16RawLe);
92        read_raw_func!(u16, read_u16_raw_be, ReadU16RawBe);
93        read_raw_func!(i16, read_i16_raw_le, ReadI16RawLe);
94        read_raw_func!(i16, read_i16_raw_be, ReadI16RawBe);
95
96        read_raw_func!(u32, read_u32_raw_le, ReadU32RawLe);
97        read_raw_func!(u32, read_u32_raw_be, ReadU32RawBe);
98        read_raw_func!(i32, read_i32_raw_le, ReadI32RawLe);
99        read_raw_func!(i32, read_i32_raw_be, ReadI32RawBe);
100
101        read_raw_func!(u64, read_u64_raw_le, ReadU64RawLe);
102        read_raw_func!(u64, read_u64_raw_be, ReadU64RawBe);
103        read_raw_func!(i64, read_i64_raw_le, ReadI64RawLe);
104        read_raw_func!(i64, read_i64_raw_be, ReadI64RawBe);
105
106        read_raw_func!(u128, read_u128_raw_le, ReadU128RawLe);
107        read_raw_func!(u128, read_u128_raw_be, ReadU128RawBe);
108        read_raw_func!(i128, read_i128_raw_le, ReadI128RawLe);
109        read_raw_func!(i128, read_i128_raw_be, ReadI128RawBe);
110
111        read_raw_func!(f32, read_f32_raw_le, ReadF32RawLe);
112        read_raw_func!(f32, read_f32_raw_be, ReadF32RawBe);
113        read_raw_func!(f64, read_f64_raw_le, ReadF64RawLe);
114        read_raw_func!(f64, read_f64_raw_be, ReadF64RawBe);
115    };
116}
117
118define_read_raw_future!();