npsd 0.2.0

Network payload serializer / deserializer framework
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use core::mem;

use super::{Error, AsyncMiddleware, AsyncPayload, AsyncIntoPayload, AsyncFromPayload};

impl<'a, C: Send + Sync, T: AsyncIntoPayload<C>> AsyncIntoPayload<C> for &'a [T] {
    async fn poll_into_payload<'m, M: AsyncMiddleware<'m>>(&self, ctx: &mut C, next: &mut M) -> Result<(), Error> {
        if mem::size_of::<T>() == 1 {
            next.poll_into_payload(&self.len(), ctx).await?;
            next.poll_write(self).await?;
        } else {
            next.poll_into_payload(&self.len(), ctx).await?;

            for elem in *self {
                next.poll_into_payload(elem, ctx).await?;
            }
        }

        Ok(())
    }
}

impl<'a, C: Send + Sync, T: AsyncFromPayload<'a, C> + 'a> AsyncFromPayload<'a, C> for &'a [T] {
    async fn poll_from_payload<M: AsyncMiddleware<'a>>(ctx: &mut C, next: &mut M) -> Result<Self, Error> {
        let len: usize = next.poll_from_payload(ctx).await?;

        if mem::size_of::<T>() == 1 {
            next.poll_read(len).await
        } else {
            let mut vec = Vec::with_capacity(len);

            for _ in 0..len {
                vec.push(next.poll_from_payload::<C, T>(ctx).await?);
            }

            next.poll_push_array(vec.into_boxed_slice()).await
        }
    }
}

impl<'a, C: Send + Sync, T: AsyncPayload<'a, C>> AsyncPayload<'a, C> for &'a [T] {}

impl<C: Send + Sync, T: AsyncIntoPayload<C>> AsyncIntoPayload<C> for &mut [T] {
    #[inline]
    async fn poll_into_payload<'m, M: AsyncMiddleware<'m>>(&self, ctx: &mut C, next: &mut M) -> Result<(), Error> {
        next.poll_into_payload::<C, &[T]>(&self.as_ref(), ctx).await
    }
}

impl<'a, C: Send + Sync, T: AsyncFromPayload<'a, C>> AsyncFromPayload<'a, C> for &'a mut [T] where T: Clone {
    async fn poll_from_payload<M: AsyncMiddleware<'a>>(ctx: &mut C, next: &mut M) -> Result<Self, Error> {
        if mem::size_of::<T>() == 1 {
            let nbytes: usize = next.poll_from_payload(ctx).await?;

            next.poll_read_mut(nbytes).await
        } else {
            let len: usize = next.poll_from_payload(ctx).await?;
            let mut vec = Vec::with_capacity(len);

            for _ in 0..len {
                vec.push(next.poll_from_payload::<C, T>(ctx).await?);
            }

            next.poll_push_array_mut(vec.into_boxed_slice()).await
        }
    }
}

impl<'a, C: Send + Sync, T: AsyncPayload<'a, C>> AsyncPayload<'a, C> for &'a mut [T] 
    where T: Clone {}

impl<C: Send + Sync, T: AsyncIntoPayload<C>, const N: usize> AsyncIntoPayload<C> for [T; N] {
    async fn poll_into_payload<'m, M: AsyncMiddleware<'m>>(&self, ctx: &mut C, next: &mut M) -> Result<(), Error> {
        if mem::size_of::<T>() == 1 {
            next.poll_write(self).await?;
        } else {
            for elem in self.into_iter() {
                next.poll_into_payload(elem, ctx).await?;
            }
        }
        
        Ok(())
    }
}

impl<'a, C: Send + Sync, T: AsyncFromPayload<'a, C> + 'a, const N: usize> AsyncFromPayload<'a, C> for [T; N] 
    where T: Copy
{
    async fn poll_from_payload<M: AsyncMiddleware<'a>>(ctx: &mut C, next: &mut M) -> Result<Self, Error> {
        if mem::size_of::<T>() == 1 {
            let bytes: &[T] = next.poll_read(N).await?;

            Ok(unsafe {
                *(bytes.as_ptr() as *const [T; N])
            })
        } else {
            let mut vec = Vec::with_capacity(N);

            for _ in 0..N {
                vec.push(next.poll_from_payload::<C, T>(ctx).await?);
            }

            let slice = next.poll_push_array(vec.into_boxed_slice()).await?;

            Ok(unsafe { *(slice.as_ptr() as *const [T; N]) })
        }
    }
}

impl<'a, C: Send + Sync, T: AsyncPayload<'a, C>, const N: usize> AsyncPayload<'a, C> for [T; N] where T: Copy {}