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
use crate::{ channel::{Channel, ForkHandle}, kind, kind::Future, ConstructResult, DeconstructResult, Kind, }; use futures::{SinkExt, StreamExt}; use alloc::sync::Arc; use std::sync::Mutex; use super::WrappedError; #[kind] impl<T> Kind for Arc<Mutex<T>> where T: Kind, { type ConstructItem = ForkHandle; type ConstructError = WrappedError<T::ConstructError>; type ConstructFuture = Future<ConstructResult<Self>>; type DeconstructItem = (); type DeconstructError = WrappedError<T::DeconstructError>; type DeconstructFuture = Future<DeconstructResult<Self>>; fn deconstruct<C: Channel<Self::DeconstructItem, Self::ConstructItem>>( self, mut channel: C, ) -> Self::DeconstructFuture { Box::pin(async move { Ok(channel .send( channel .fork::<T>( Arc::try_unwrap(self) .map_err(|_| panic!()) .unwrap() .into_inner() .unwrap(), ) .await?, ) .await .map_err(WrappedError::Send)?) }) } fn construct<C: Channel<Self::ConstructItem, Self::DeconstructItem>>( mut channel: C, ) -> Self::ConstructFuture { Box::pin(async move { let handle = channel.next().await.ok_or(WrappedError::Insufficient { got: 0, expected: 1, })?; Ok(Arc::new(Mutex::new(channel.get_fork(handle).await?))) }) } } #[kind] impl<T> Kind for Box<T> where T: Kind, { type ConstructItem = ForkHandle; type ConstructError = WrappedError<T::ConstructError>; type ConstructFuture = Future<ConstructResult<Self>>; type DeconstructItem = (); type DeconstructError = WrappedError<T::DeconstructError>; type DeconstructFuture = Future<DeconstructResult<Self>>; fn deconstruct<C: Channel<Self::DeconstructItem, Self::ConstructItem>>( self, mut channel: C, ) -> Self::DeconstructFuture { Box::pin(async move { Ok(channel .send(channel.fork::<T>(*self).await?) .await .map_err(WrappedError::Send)?) }) } fn construct<C: Channel<Self::ConstructItem, Self::DeconstructItem>>( mut channel: C, ) -> Self::ConstructFuture { Box::pin(async move { let handle = channel.next().await.ok_or(WrappedError::Insufficient { got: 0, expected: 1, })?; Ok(Box::new(channel.get_fork(handle).await?)) }) } }