msrd0/deprecated-gotham-restful
1
0
Fork 0
mirror of https://gitlab.com/msrd0/gotham-restful.git synced 2025-04-20 06:54:46 +00:00
Code Issues Releases Activity Actions

expose async to handlers

Browse source
This commit is contained in:
Dominic 2020-04-15 20:55:25 +02:00
parent d7282786b1
commit 427c836f52
Signed by: msrd0
GPG key ID: DCC8C247452E98F9
5 changed files with 71 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

1
gotham_restful/src/lib.rs
View file

@ -115,6 +115,7 @@ pub use gotham_restful_derive::*;
#[doc(hidden)]
pub mod export
{
pub use futures_util::future::FutureExt;
pub use gotham::{
hyper::body::Bytes,
state::{FromState, State}

24
gotham_restful/src/resource.rs
View file

@ -5,7 +5,11 @@ use gotham::{
};
use hyper::Body;
use serde::de::DeserializeOwned;
use std::panic::RefUnwindSafe;
use std::{
future::Future,
panic::RefUnwindSafe,
pin::Pin
};
/// This trait must be implemented by every RESTful Resource. It will
/// allow you to register the different methods for this Resource.
@ -21,7 +25,7 @@ pub trait Resource
pub trait ResourceMethod
{
type Res : ResourceResult;
type Res : ResourceResult + Send + 'static;
#[cfg(feature = "openapi")]
fn operation_id() -> Option<String>
@ -38,7 +42,7 @@ pub trait ResourceMethod
/// Handle a GET request on the Resource root.
pub trait ResourceReadAll : ResourceMethod
{
fn read_all(state : &mut State) -> Self::Res;
fn read_all(state : &mut State) -> Pin<Box<dyn Future<Output = Self::Res> + Send>>;
}
/// Handle a GET request on the Resource with an id.
@ -46,7 +50,7 @@ pub trait ResourceRead : ResourceMethod
{
type ID : DeserializeOwned + Clone + RefUnwindSafe + Send + Sync + 'static;
fn read(state : &mut State, id : Self::ID) -> Self::Res;
fn read(state : &mut State, id : Self::ID) -> Pin<Box<dyn Future<Output = Self::Res> + Send>>;
}
/// Handle a GET request on the Resource with additional search parameters.
@ -54,7 +58,7 @@ pub trait ResourceSearch : ResourceMethod
{
type Query : ResourceType + QueryStringExtractor<Body> + Sync;
fn search(state : &mut State, query : Self::Query) -> Self::Res;
fn search(state : &mut State, query : Self::Query) -> Pin<Box<dyn Future<Output = Self::Res> + Send>>;
}
/// Handle a POST request on the Resource root.
@ -62,7 +66,7 @@ pub trait ResourceCreate : ResourceMethod
{
type Body : RequestBody;
fn create(state : &mut State, body : Self::Body) -> Self::Res;
fn create(state : &mut State, body : Self::Body) -> Pin<Box<dyn Future<Output = Self::Res> + Send>>;
}
/// Handle a PUT request on the Resource root.
@ -70,7 +74,7 @@ pub trait ResourceUpdateAll : ResourceMethod
{
type Body : RequestBody;
fn update_all(state : &mut State, body : Self::Body) -> Self::Res;
fn update_all(state : &mut State, body : Self::Body) -> Pin<Box<dyn Future<Output = Self::Res> + Send>>;
}
/// Handle a PUT request on the Resource with an id.
@ -79,13 +83,13 @@ pub trait ResourceUpdate : ResourceMethod
type Body : RequestBody;
type ID : DeserializeOwned + Clone + RefUnwindSafe + Send + Sync + 'static;
fn update(state : &mut State, id : Self::ID, body : Self::Body) -> Self::Res;
fn update(state : &mut State, id : Self::ID, body : Self::Body) -> Pin<Box<dyn Future<Output = Self::Res> + Send>>;
}
/// Handle a DELETE request on the Resource root.
pub trait ResourceDeleteAll : ResourceMethod
{
fn delete_all(state : &mut State) -> Self::Res;
fn delete_all(state : &mut State) -> Pin<Box<dyn Future<Output = Self::Res> + Send>>;
}
/// Handle a DELETE request on the Resource with an id.
@ -93,5 +97,5 @@ pub trait ResourceDelete : ResourceMethod
{
type ID : DeserializeOwned + Clone + RefUnwindSafe + Send + Sync + 'static;
fn delete(state : &mut State, id : Self::ID) -> Self::Res;
fn delete(state : &mut State, id : Self::ID) -> Pin<Box<dyn Future<Output = Self::Res> + Send>>;
}

2
gotham_restful/src/result.rs
View file

@ -80,7 +80,7 @@ impl Response
/// A trait provided to convert a resource's result to json.
pub trait ResourceResult : Send
pub trait ResourceResult
{
type Err : Error + Send + 'static;

49
gotham_restful/src/routing.rs
View file

@ -9,7 +9,7 @@ use crate::OpenapiRouter;
use futures_util::{future, future::FutureExt};
use gotham::{
handler::{HandlerError, HandlerFuture, IntoHandlerError, IntoHandlerFuture},
handler::{HandlerError, HandlerFuture, IntoHandlerError},
helpers::http::response::{create_empty_response, create_response},
pipeline::chain::PipelineHandleChain,
router::{
@ -32,6 +32,7 @@ use gotham::hyper::{
};
use mime::{Mime, APPLICATION_JSON};
use std::{
future::Future,
panic::RefUnwindSafe,
pin::Pin
};
@ -105,27 +106,26 @@ fn response_from(res : Response, state : &State) -> hyper::Response<Body>
r
}
fn to_handler_future<F, R>(mut state : State, get_result : F) -> Pin<Box<HandlerFuture>>
async fn to_handler_future<F, R>(mut state : State, get_result : F) -> Result<(State, gotham::hyper::Response<Body>), (State, HandlerError)>
where
F : FnOnce(&mut State) -> R,
F : FnOnce(&mut State) -> Pin<Box<dyn Future<Output = R> + Send>>,
R : ResourceResult
{
get_result(&mut state).into_response()
.then(|res|
match res {
Ok(res) => {
let r = response_from(res, &state);
(state, r).into_handler_future()
},
Err(e) => future::err((state, e.into_handler_error())).boxed()
}
).boxed()
let res = get_result(&mut state).await;
let res = res.into_response().await;
match res {
Ok(res) => {
let r = response_from(res, &state);
Ok((state, r))
},
Err(e) => Err((state, e.into_handler_error()))
}
}
async fn body_to_res<B, F, R>(mut state : State, get_result : F) -> (State, Result<gotham::hyper::Response<Body>, HandlerError>)
where
B : RequestBody,
F : FnOnce(&mut State, B) -> R,
F : FnOnce(&mut State, B) -> Pin<Box<dyn Future<Output = R> + Send>>,
R : ResourceResult
{
let body = to_bytes(Body::take_from(&mut state)).await;
@ -160,8 +160,11 @@ where
};
get_result(&mut state, body)
};
let res = match res.into_response().await {
let res = res.await;
let res = res.into_response().await;
let res = match res {
Ok(res) => {
let r = response_from(res, &state);
Ok(r)
@ -174,8 +177,8 @@ where
fn handle_with_body<B, F, R>(state : State, get_result : F) -> Pin<Box<HandlerFuture>>
where
B : RequestBody + 'static,
F : FnOnce(&mut State, B) -> R + Send + 'static,
R : ResourceResult + 'static
F : FnOnce(&mut State, B) -> Pin<Box<dyn Future<Output = R> + Send>> + Send + 'static,
R : ResourceResult + Send + 'static
{
body_to_res(state, get_result)
.then(|(state, res)| match res {
@ -187,7 +190,7 @@ where
fn read_all_handler<Handler : ResourceReadAll>(state : State) -> Pin<Box<HandlerFuture>>
{
to_handler_future(state, |state| Handler::read_all(state))
to_handler_future(state, |state| Handler::read_all(state)).boxed()
}
fn read_handler<Handler : ResourceRead>(state : State) -> Pin<Box<HandlerFuture>>
@ -196,13 +199,13 @@ fn read_handler<Handler : ResourceRead>(state : State) -> Pin<Box<HandlerFuture>
let path : &PathExtractor<Handler::ID> = PathExtractor::borrow_from(&state);
path.id.clone()
};
to_handler_future(state, |state| Handler::read(state, id))
to_handler_future(state, |state| Handler::read(state, id)).boxed()
}
fn search_handler<Handler : ResourceSearch>(mut state : State) -> Pin<Box<HandlerFuture>>
{
let query = Handler::Query::take_from(&mut state);
to_handler_future(state, |state| Handler::search(state, query))
to_handler_future(state, |state| Handler::search(state, query)).boxed()
}
fn create_handler<Handler : ResourceCreate>(state : State) -> Pin<Box<HandlerFuture>>
@ -235,7 +238,7 @@ where
fn delete_all_handler<Handler : ResourceDeleteAll>(state : State) -> Pin<Box<HandlerFuture>>
{
to_handler_future(state, |state| Handler::delete_all(state))
to_handler_future(state, |state| Handler::delete_all(state)).boxed()
}
fn delete_handler<Handler : ResourceDelete>(state : State) -> Pin<Box<HandlerFuture>>
@ -244,7 +247,7 @@ fn delete_handler<Handler : ResourceDelete>(state : State) -> Pin<Box<HandlerFut
let path : &PathExtractor<Handler::ID> = PathExtractor::borrow_from(&state);
path.id.clone()
};
to_handler_future(state, |state| Handler::delete(state, id))
to_handler_future(state, |state| Handler::delete(state, id)).boxed()
}
#[derive(Clone)]