1
0
Fork 0
mirror of https://github.com/actix/actix-web.git synced 2024-12-21 23:56:35 +00:00

use response instead of result for asyn c handlers

This commit is contained in:
Nikolay Kim 2019-11-21 15:01:34 +06:00
parent 55698f2524
commit 53c5151692
5 changed files with 43 additions and 59 deletions

View file

@ -8,9 +8,9 @@ fn index(req: HttpRequest, name: web::Path<String>) -> String {
format!("Hello: {}!\r\n", name) format!("Hello: {}!\r\n", name)
} }
async fn index_async(req: HttpRequest) -> Result<&'static str, Error> { async fn index_async(req: HttpRequest) -> &'static str {
println!("REQ: {:?}", req); println!("REQ: {:?}", req);
Ok("Hello world!\r\n") "Hello world!\r\n"
} }
#[get("/")] #[get("/")]
@ -26,7 +26,7 @@ fn main() -> std::io::Result<()> {
App::new() App::new()
.wrap(middleware::DefaultHeaders::new().header("X-Version", "0.2")) .wrap(middleware::DefaultHeaders::new().header("X-Version", "0.2"))
.wrap(middleware::Compress::default()) .wrap(middleware::Compress::default())
// .wrap(middleware::Logger::default()) .wrap(middleware::Logger::default())
.service(index) .service(index)
.service(no_params) .service(no_params)
.service( .service(

View file

@ -119,21 +119,19 @@ impl<T: Responder> Future for HandlerServiceResponse<T> {
} }
/// Async handler converter factory /// Async handler converter factory
pub trait AsyncFactory<T, R, O, E>: Clone + 'static pub trait AsyncFactory<T, R, O>: Clone + 'static
where where
R: Future<Output = Result<O, E>>, R: Future<Output = O>,
O: Responder, O: Responder,
E: Into<Error>,
{ {
fn call(&self, param: T) -> R; fn call(&self, param: T) -> R;
} }
impl<F, R, O, E> AsyncFactory<(), R, O, E> for F impl<F, R, O> AsyncFactory<(), R, O> for F
where where
F: Fn() -> R + Clone + 'static, F: Fn() -> R + Clone + 'static,
R: Future<Output = Result<O, E>>, R: Future<Output = O>,
O: Responder, O: Responder,
E: Into<Error>,
{ {
fn call(&self, _: ()) -> R { fn call(&self, _: ()) -> R {
(self)() (self)()
@ -141,23 +139,21 @@ where
} }
#[doc(hidden)] #[doc(hidden)]
pub struct AsyncHandler<F, T, R, O, E> pub struct AsyncHandler<F, T, R, O>
where where
F: AsyncFactory<T, R, O, E>, F: AsyncFactory<T, R, O>,
R: Future<Output = Result<O, E>>, R: Future<Output = O>,
O: Responder, O: Responder,
E: Into<Error>,
{ {
hnd: F, hnd: F,
_t: PhantomData<(T, R, O, E)>, _t: PhantomData<(T, R, O)>,
} }
impl<F, T, R, O, E> AsyncHandler<F, T, R, O, E> impl<F, T, R, O> AsyncHandler<F, T, R, O>
where where
F: AsyncFactory<T, R, O, E>, F: AsyncFactory<T, R, O>,
R: Future<Output = Result<O, E>>, R: Future<Output = O>,
O: Responder, O: Responder,
E: Into<Error>,
{ {
pub fn new(hnd: F) -> Self { pub fn new(hnd: F) -> Self {
AsyncHandler { AsyncHandler {
@ -167,12 +163,11 @@ where
} }
} }
impl<F, T, R, O, E> Clone for AsyncHandler<F, T, R, O, E> impl<F, T, R, O> Clone for AsyncHandler<F, T, R, O>
where where
F: AsyncFactory<T, R, O, E>, F: AsyncFactory<T, R, O>,
R: Future<Output = Result<O, E>>, R: Future<Output = O>,
O: Responder, O: Responder,
E: Into<Error>,
{ {
fn clone(&self) -> Self { fn clone(&self) -> Self {
AsyncHandler { AsyncHandler {
@ -182,17 +177,16 @@ where
} }
} }
impl<F, T, R, O, E> Service for AsyncHandler<F, T, R, O, E> impl<F, T, R, O> Service for AsyncHandler<F, T, R, O>
where where
F: AsyncFactory<T, R, O, E>, F: AsyncFactory<T, R, O>,
R: Future<Output = Result<O, E>>, R: Future<Output = O>,
O: Responder, O: Responder,
E: Into<Error>,
{ {
type Request = (T, HttpRequest); type Request = (T, HttpRequest);
type Response = ServiceResponse; type Response = ServiceResponse;
type Error = Infallible; type Error = Infallible;
type Future = AsyncHandlerServiceResponse<R, O, E>; type Future = AsyncHandlerServiceResponse<R, O>;
fn poll_ready(&mut self, _: &mut Context) -> Poll<Result<(), Self::Error>> { fn poll_ready(&mut self, _: &mut Context) -> Poll<Result<(), Self::Error>> {
Poll::Ready(Ok(())) Poll::Ready(Ok(()))
@ -209,11 +203,10 @@ where
#[doc(hidden)] #[doc(hidden)]
#[pin_project] #[pin_project]
pub struct AsyncHandlerServiceResponse<T, R, E> pub struct AsyncHandlerServiceResponse<T, R>
where where
T: Future<Output = Result<R, E>>, T: Future<Output = R>,
R: Responder, R: Responder,
E: Into<Error>,
{ {
#[pin] #[pin]
fut: T, fut: T,
@ -222,11 +215,10 @@ where
req: Option<HttpRequest>, req: Option<HttpRequest>,
} }
impl<T, R, E> Future for AsyncHandlerServiceResponse<T, R, E> impl<T, R> Future for AsyncHandlerServiceResponse<T, R>
where where
T: Future<Output = Result<R, E>>, T: Future<Output = R>,
R: Responder, R: Responder,
E: Into<Error>,
{ {
type Output = Result<ServiceResponse, Infallible>; type Output = Result<ServiceResponse, Infallible>;
@ -247,16 +239,12 @@ where
} }
match this.fut.poll(cx) { match this.fut.poll(cx) {
Poll::Ready(Ok(res)) => { Poll::Ready(res) => {
let fut = res.respond_to(this.req.as_ref().unwrap()); let fut = res.respond_to(this.req.as_ref().unwrap());
self.as_mut().project().fut2.set(Some(fut)); self.as_mut().project().fut2.set(Some(fut));
self.poll(cx) self.poll(cx)
} }
Poll::Pending => Poll::Pending, Poll::Pending => Poll::Pending,
Poll::Ready(Err(e)) => {
let res: Response = e.into().into();
Poll::Ready(Ok(ServiceResponse::new(this.req.take().unwrap(), res)))
}
} }
} }
} }
@ -387,11 +375,10 @@ macro_rules! factory_tuple ({ $(($n:tt, $T:ident)),+} => {
} }
} }
impl<Func, $($T,)+ Res, O, E1> AsyncFactory<($($T,)+), Res, O, E1> for Func impl<Func, $($T,)+ Res, O> AsyncFactory<($($T,)+), Res, O> for Func
where Func: Fn($($T,)+) -> Res + Clone + 'static, where Func: Fn($($T,)+) -> Res + Clone + 'static,
Res: Future<Output = Result<O, E1>>, Res: Future<Output = O>,
O: Responder, O: Responder,
E1: Into<Error>,
{ {
fn call(&self, param: ($($T,)+)) -> Res { fn call(&self, param: ($($T,)+)) -> Res {
(self)($(param.$n,)+) (self)($(param.$n,)+)

View file

@ -264,13 +264,12 @@ where
/// App::new().service(web::resource("/").route(web::route().to_async(index))); /// App::new().service(web::resource("/").route(web::route().to_async(index)));
/// ``` /// ```
#[allow(clippy::wrong_self_convention)] #[allow(clippy::wrong_self_convention)]
pub fn to_async<F, I, R, O, E>(mut self, handler: F) -> Self pub fn to_async<F, I, R, U>(mut self, handler: F) -> Self
where where
F: AsyncFactory<I, R, O, E>, F: AsyncFactory<I, R, U>,
I: FromRequest + 'static, I: FromRequest + 'static,
R: Future<Output = Result<O, E>> + 'static, R: Future<Output = U> + 'static,
O: Responder + 'static, U: Responder + 'static,
E: Into<Error> + 'static,
{ {
self.routes.push(Route::new().to_async(handler)); self.routes.push(Route::new().to_async(handler));
self self

View file

@ -261,13 +261,12 @@ impl Route {
/// } /// }
/// ``` /// ```
#[allow(clippy::wrong_self_convention)] #[allow(clippy::wrong_self_convention)]
pub fn to_async<F, T, R, O, E>(mut self, handler: F) -> Self pub fn to_async<F, T, R, U>(mut self, handler: F) -> Self
where where
F: AsyncFactory<T, R, O, E>, F: AsyncFactory<T, R, U>,
T: FromRequest + 'static, T: FromRequest + 'static,
R: Future<Output = Result<O, E>> + 'static, R: Future<Output = U> + 'static,
O: Responder + 'static, U: Responder + 'static,
E: Into<Error> + 'static,
{ {
self.service = Box::new(RouteNewService::new(Extract::new(AsyncHandler::new( self.service = Box::new(RouteNewService::new(Extract::new(AsyncHandler::new(
handler, handler,
@ -410,7 +409,7 @@ mod tests {
.route(web::post().to_async(|| { .route(web::post().to_async(|| {
async { async {
delay_for(Duration::from_millis(100)).await; delay_for(Duration::from_millis(100)).await;
Ok::<_, Error>(HttpResponse::Created()) HttpResponse::Created()
} }
})) }))
.route(web::delete().to_async(|| { .route(web::delete().to_async(|| {
@ -423,9 +422,9 @@ mod tests {
.service(web::resource("/json").route(web::get().to_async(|| { .service(web::resource("/json").route(web::get().to_async(|| {
async { async {
delay_for(Duration::from_millis(25)).await; delay_for(Duration::from_millis(25)).await;
Ok::<_, Error>(web::Json(MyObject { web::Json(MyObject {
name: "test".to_string(), name: "test".to_string(),
})) })
} }
}))), }))),
) )

View file

@ -265,13 +265,12 @@ where
/// web::to_async(index)) /// web::to_async(index))
/// ); /// );
/// ``` /// ```
pub fn to_async<F, I, R, O, E>(handler: F) -> Route pub fn to_async<F, I, R, U>(handler: F) -> Route
where where
F: AsyncFactory<I, R, O, E>, F: AsyncFactory<I, R, U>,
I: FromRequest + 'static, I: FromRequest + 'static,
R: Future<Output = Result<O, E>> + 'static, R: Future<Output = U> + 'static,
O: Responder + 'static, U: Responder + 'static,
E: Into<Error> + 'static,
{ {
Route::new().to_async(handler) Route::new().to_async(handler)
} }