Start: de_singleton_inner()

Fortune for transfer's current commit: Small curse − 小凶
basic
Avril 3 years ago
parent 1f5ead31c9
commit 03eaddb7eb
Signed by: flanchan
GPG Key ID: 284488987C31F630

@ -13,7 +13,7 @@ use tokio::io::{
}; };
use serde::{ use serde::{
Serialize, Serialize,
Deserialize de::DeserializeOwned
}; };
#[derive(Debug, Clone, PartialEq, Eq, Hash, Copy, PartialOrd, Ord)] #[derive(Debug, Clone, PartialEq, Eq, Hash, Copy, PartialOrd, Ord)]
@ -75,6 +75,19 @@ pub struct SendOpt
} }
ref_self!(SendOpt); ref_self!(SendOpt);
#[derive(Debug, Clone, PartialEq, Eq, Hash, Copy, Default)]
pub struct RecvOpt
{
pub comp: Option<CompressionKind>,
pub handle_encrypted: Option<EncryptionKind>,
pub format: SerialFormat,
/// Max size of the (decompressed) input buffer.
/// 0 for unlimited.
//TODO: Do we need this?
pub max_size: usize,
}
ref_self!(RecvOpt);
/// Default buffer size for encryption transform stream copying. /// Default buffer size for encryption transform stream copying.
pub const DEFAULT_BUFSIZE: usize = 4096; pub const DEFAULT_BUFSIZE: usize = 4096;
@ -104,7 +117,13 @@ where F: AsyncRead + Unpin + ?Sized,
Ok((written, read)) Ok((written, read))
} }
async fn ser_singleton_inner<T: Serialize, V: AsyncWrite + Unpin, F>(to: F, value: &T, how: impl AsRef<SendOpt>) -> Result<(V, usize), SendErrorKind> async fn de_singleton_inner<T: DeserializeOwned, S>(from: S, how: impl AsRef<RecvOpt>) -> Result<T, TransformErrorKind>
where S: AsRef<[u8]> //TODO: Should we use bytes::Buf or something instead?
{
todo!("Deserialise from `from` using `how`.")
}
async fn ser_singleton_inner<T: Serialize, V: AsyncWrite + Unpin, F>(to: F, value: &T, how: impl AsRef<SendOpt>) -> Result<(V, usize), TransformErrorKind>
where F: FnOnce(&Vec<u8>) -> V where F: FnOnce(&Vec<u8>) -> V
{ {
let how = how.as_ref(); let how = how.as_ref();
@ -153,13 +172,25 @@ where F: FnOnce(&Vec<u8>) -> V
/// Serialise a single object to a stream with the method described by `how`. /// Serialise a single object to a stream with the method described by `how`.
#[inline] pub async fn write_singleton<T: Serialize, S: ?Sized + AsyncWrite + Unpin>(to: &mut S, value: &T, how: &SendOpt) -> Result<usize, SendError> #[inline] pub async fn write_singleton<T: Serialize, S: ?Sized + AsyncWrite + Unpin>(to: &mut S, value: &T, how: &SendOpt) -> Result<usize, SendError>
{ {
ser_singleton_inner(|_| to, value, &how).await let (cont, v) = ser_singleton_inner(|n| Vec::with_capacity(n.len()), value, &how).await
.map_err(|k| SendError(Box::new((k, how.to_owned())))) .map_err(|k| SendError(Box::new((k, how.to_owned()))))?;
.map(|(_, n)| n)
let n = async move {
to.write_all(&(v as u64).to_be_bytes()[..]).await?;
to.write_all(&cont).await
.map(|_| std::mem::size_of::<u64>() + cont.len())
}
.await
.map_err(|k| SendError(Box::new((k.into(), how.to_owned()))))?;
Ok(n)
} }
/// Kind of error for a send (serialise) or receive (deserialise) operation
#[derive(Debug)] #[derive(Debug)]
pub enum SendErrorKind pub enum TransformErrorKind
{ {
/// Invalid serialised format /// Invalid serialised format
Format, Format,
@ -174,14 +205,19 @@ pub enum SendErrorKind
/// An error when sending / serialising an object. /// An error when sending / serialising an object.
#[derive(Debug)] #[derive(Debug)]
pub struct SendError(Box<(SendErrorKind, SendOpt)>); pub struct RecvError(Box<(TransformErrorKind, RecvOpt)>);
/// An error when sending / serialising an object.
#[derive(Debug)]
pub struct SendError(Box<(TransformErrorKind, SendOpt)>);
impl error::Error for SendError impl error::Error for SendError
{ {
fn source(&self) -> Option<&(dyn error::Error + 'static)> { fn source(&self) -> Option<&(dyn error::Error + 'static)> {
Some(match &self.0.0 Some(match &self.0.0
{ {
SendErrorKind::IO(io) => io, TransformErrorKind::IO(io) => io,
_ => return None, _ => return None,
}) })
} }
@ -193,15 +229,15 @@ impl fmt::Display for SendError
{ {
write!(f, "error when serialising object with params {:?}: ", self.0.1)?; write!(f, "error when serialising object with params {:?}: ", self.0.1)?;
match self.0.0 { match self.0.0 {
SendErrorKind::Format => write!(f, "failed to serialise object to data"), TransformErrorKind::Format => write!(f, "failed to serialise object to data"),
SendErrorKind::Compress => write!(f, "failed to compress data"), TransformErrorKind::Compress => write!(f, "failed to compress data"),
SendErrorKind::Encrypt => write!(f, "failed to encrypt data"), TransformErrorKind::Encrypt => write!(f, "failed to encrypt data"),
SendErrorKind::IO(_) => write!(f, "i/o failure"), TransformErrorKind::IO(_) => write!(f, "i/o failure"),
} }
} }
} }
impl From<io::Error> for SendErrorKind impl From<io::Error> for TransformErrorKind
{ {
fn from(from: io::Error) -> Self fn from(from: io::Error) -> Self
{ {
@ -210,7 +246,7 @@ impl From<io::Error> for SendErrorKind
} }
impl From<serde_cbor::Error> for SendErrorKind impl From<serde_cbor::Error> for TransformErrorKind
{ {
#[inline] fn from(_: serde_cbor::Error) -> Self #[inline] fn from(_: serde_cbor::Error) -> Self
{ {
@ -218,7 +254,7 @@ impl From<serde_cbor::Error> for SendErrorKind
} }
} }
impl From<serde_json::Error> for SendErrorKind impl From<serde_json::Error> for TransformErrorKind
{ {
#[inline] fn from(_: serde_json::Error) -> Self #[inline] fn from(_: serde_json::Error) -> Self
{ {

Loading…
Cancel
Save