mirror of
https://github.com/fafhrd91/actix-web
synced 2024-11-28 01:52:57 +01:00
implement ResponseError trait for BlockingError
This commit is contained in:
parent
db566a634c
commit
db39a604ae
@ -1,10 +1,15 @@
|
|||||||
//! Thread pool for blocking operations
|
//! Thread pool for blocking operations
|
||||||
|
|
||||||
|
use std::fmt;
|
||||||
|
|
||||||
|
use derive_more::Display;
|
||||||
use futures::sync::oneshot;
|
use futures::sync::oneshot;
|
||||||
use futures::{Async, Future, Poll};
|
use futures::{Async, Future, Poll};
|
||||||
use parking_lot::Mutex;
|
use parking_lot::Mutex;
|
||||||
use threadpool::ThreadPool;
|
use threadpool::ThreadPool;
|
||||||
|
|
||||||
|
use crate::ResponseError;
|
||||||
|
|
||||||
/// Env variable for default cpu pool size
|
/// Env variable for default cpu pool size
|
||||||
const ENV_CPU_POOL_VAR: &str = "ACTIX_CPU_POOL";
|
const ENV_CPU_POOL_VAR: &str = "ACTIX_CPU_POOL";
|
||||||
|
|
||||||
@ -36,18 +41,23 @@ thread_local! {
|
|||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
pub enum BlockingError<E> {
|
#[derive(Debug, Display)]
|
||||||
|
pub enum BlockingError<E: fmt::Debug> {
|
||||||
|
#[display(fmt = "{:?}", _0)]
|
||||||
Error(E),
|
Error(E),
|
||||||
|
#[display(fmt = "Thread pool is gone")]
|
||||||
Canceled,
|
Canceled,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl<E: fmt::Debug> ResponseError for BlockingError<E> {}
|
||||||
|
|
||||||
/// Execute blocking function on a thread pool, returns future that resolves
|
/// Execute blocking function on a thread pool, returns future that resolves
|
||||||
/// to result of the function execution.
|
/// to result of the function execution.
|
||||||
pub fn run<F, I, E>(f: F) -> CpuFuture<I, E>
|
pub fn run<F, I, E>(f: F) -> CpuFuture<I, E>
|
||||||
where
|
where
|
||||||
F: FnOnce() -> Result<I, E> + Send + 'static,
|
F: FnOnce() -> Result<I, E> + Send + 'static,
|
||||||
I: Send + 'static,
|
I: Send + 'static,
|
||||||
E: Send + 'static,
|
E: Send + fmt::Debug + 'static,
|
||||||
{
|
{
|
||||||
let (tx, rx) = oneshot::channel();
|
let (tx, rx) = oneshot::channel();
|
||||||
POOL.with(|pool| {
|
POOL.with(|pool| {
|
||||||
@ -63,7 +73,7 @@ pub struct CpuFuture<I, E> {
|
|||||||
rx: oneshot::Receiver<Result<I, E>>,
|
rx: oneshot::Receiver<Result<I, E>>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<I, E> Future for CpuFuture<I, E> {
|
impl<I, E: fmt::Debug> Future for CpuFuture<I, E> {
|
||||||
type Item = I;
|
type Item = I;
|
||||||
type Error = BlockingError<E>;
|
type Error = BlockingError<E>;
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user