59 lines
1.4 KiB
Rust
59 lines
1.4 KiB
Rust
//! 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),
|
|
},
|
|
}
|
|
}
|
|
}
|