You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
tungstenite-rs/src/util.rs

58 lines
1.4 KiB

//! Helper traits to ease non-blocking handling.
use std::{
io::{Error as IoError, ErrorKind as IoErrorKind},
result::Result as StdResult,
};
use crate::error::Error;
/// Non-blocking IO handling.
pub trait NonBlockingError: Sized {
/// Convert WouldBlock to None and don't touch other errors.
fn into_non_blocking(self) -> Option<Self>;
}
impl NonBlockingError for IoError {
fn into_non_blocking(self) -> Option<Self> {
match self.kind() {
IoErrorKind::WouldBlock => None,
_ => Some(self),
}
}
}
impl NonBlockingError for Error {
fn into_non_blocking(self) -> Option<Self> {
match self {
Error::Io(e) => e.into_non_blocking().map(|e| e.into()),
x => Some(x),
}
}
}
/// Non-blocking IO wrapper.
///
/// This trait is implemented for `Result<T, E: NonBlockingError>`.
pub trait NonBlockingResult {
/// Type of the converted result: `Result<Option<T>, E>`
type Result;
/// Perform the non-block conversion.
fn no_block(self) -> Self::Result;
}
impl<T, E> NonBlockingResult for StdResult<T, E>
where
E: NonBlockingError,
{
type Result = StdResult<Option<T>, E>;
fn no_block(self) -> Self::Result {
match self {
Ok(x) => Ok(Some(x)),
Err(e) => match e.into_non_blocking() {
Some(e) => Err(e),
None => Ok(None),
},
}
}
}