1
0
Fork 0
mirror of https://github.com/actix/actix-web.git synced 2024-11-23 01:51:11 +00:00

introduce custom FromRequest traint for conversion into Reply

This commit is contained in:
Nikolay Kim 2017-12-02 16:37:21 -08:00
parent 187948ddd1
commit 61744b68a1
6 changed files with 82 additions and 39 deletions

View file

@ -3,7 +3,7 @@ use std::collections::HashMap;
use futures::Future; use futures::Future;
use error::Error; use error::Error;
use route::{RouteHandler, Reply, Handler, WrapHandler, AsyncHandler}; use route::{RouteHandler, Reply, Handler, FromRequest, WrapHandler, AsyncHandler};
use resource::Resource; use resource::Resource;
use recognizer::{RouteRecognizer, check_pattern}; use recognizer::{RouteRecognizer, check_pattern};
use httprequest::HttpRequest; use httprequest::HttpRequest;
@ -190,7 +190,7 @@ impl<S> ApplicationBuilder<S> where S: 'static {
pub fn handler<P, F, R>(&mut self, path: P, handler: F) -> &mut Self pub fn handler<P, F, R>(&mut self, path: P, handler: F) -> &mut Self
where P: Into<String>, where P: Into<String>,
F: Fn(HttpRequest<S>) -> R + 'static, F: Fn(HttpRequest<S>) -> R + 'static,
R: Into<Reply> + 'static R: FromRequest + 'static
{ {
self.parts.as_mut().expect("Use after finish") self.parts.as_mut().expect("Use after finish")
.handlers.insert(path.into(), Box::new(WrapHandler::new(handler))); .handlers.insert(path.into(), Box::new(WrapHandler::new(handler)));

View file

@ -10,7 +10,7 @@
// dev specific // dev specific
pub use pipeline::Pipeline; pub use pipeline::Pipeline;
pub use route::Handler; pub use route::{Handler, FromRequest};
pub use channel::{HttpChannel, HttpHandler}; pub use channel::{HttpChannel, HttpHandler};
pub use recognizer::{FromParam, RouteRecognizer}; pub use recognizer::{FromParam, RouteRecognizer};

View file

@ -3,8 +3,7 @@
use http::StatusCode; use http::StatusCode;
use body::Body; use body::Body;
use route::Reply; use route::{Reply, RouteHandler, FromRequest};
use route::RouteHandler;
use httprequest::HttpRequest; use httprequest::HttpRequest;
use httpresponse::{HttpResponse, HttpResponseBuilder}; use httpresponse::{HttpResponse, HttpResponseBuilder};
@ -74,6 +73,12 @@ impl<S> RouteHandler<S> for StaticResponse {
} }
} }
impl FromRequest for StaticResponse {
fn from_request(self, _: HttpRequest) -> Reply {
Reply::response(HttpResponse::new(self.0, Body::Empty))
}
}
impl From<StaticResponse> for HttpResponse { impl From<StaticResponse> for HttpResponse {
fn from(st: StaticResponse) -> Self { fn from(st: StaticResponse) -> Self {
st.response() st.response()

View file

@ -83,6 +83,11 @@ impl HttpRequest<()> {
impl<S> HttpRequest<S> { impl<S> HttpRequest<S> {
/// Construct new http request without state.
pub fn clone_without_state(&self) -> HttpRequest {
HttpRequest(Rc::clone(&self.0), Rc::new(()))
}
/// get mutable reference for inner message /// get mutable reference for inner message
fn as_mut(&mut self) -> &mut HttpMessage { fn as_mut(&mut self) -> &mut HttpMessage {
let r: &HttpMessage = self.0.as_ref(); let r: &HttpMessage = self.0.as_ref();

View file

@ -5,7 +5,7 @@ use http::Method;
use futures::Future; use futures::Future;
use error::Error; use error::Error;
use route::{Reply, Handler, RouteHandler, AsyncHandler, WrapHandler}; use route::{Reply, Handler, FromRequest, RouteHandler, AsyncHandler, WrapHandler};
use httprequest::HttpRequest; use httprequest::HttpRequest;
use httpresponse::HttpResponse; use httpresponse::HttpResponse;
use httpcodes::{HTTPNotFound, HTTPMethodNotAllowed}; use httpcodes::{HTTPNotFound, HTTPMethodNotAllowed};
@ -53,14 +53,14 @@ impl<S> Resource<S> where S: 'static {
} }
/// Set resource name /// Set resource name
pub fn set_name<T: Into<String>>(&mut self, name: T) { pub fn name<T: Into<String>>(&mut self, name: T) {
self.name = name.into(); self.name = name.into();
} }
/// Register handler for specified method. /// Register handler for specified method.
pub fn handler<F, R>(&mut self, method: Method, handler: F) pub fn handler<F, R>(&mut self, method: Method, handler: F)
where F: Fn(HttpRequest<S>) -> R + 'static, where F: Fn(HttpRequest<S>) -> R + 'static,
R: Into<Reply> + 'static, R: FromRequest + 'static,
{ {
self.routes.insert(method, Box::new(WrapHandler::new(handler))); self.routes.insert(method, Box::new(WrapHandler::new(handler)));
} }
@ -83,28 +83,28 @@ impl<S> Resource<S> where S: 'static {
/// Register handler for `GET` method. /// Register handler for `GET` method.
pub fn get<F, R>(&mut self, handler: F) pub fn get<F, R>(&mut self, handler: F)
where F: Fn(HttpRequest<S>) -> R + 'static, where F: Fn(HttpRequest<S>) -> R + 'static,
R: Into<Reply> + 'static, { R: FromRequest + 'static, {
self.routes.insert(Method::GET, Box::new(WrapHandler::new(handler))); self.routes.insert(Method::GET, Box::new(WrapHandler::new(handler)));
} }
/// Register handler for `POST` method. /// Register handler for `POST` method.
pub fn post<F, R>(&mut self, handler: F) pub fn post<F, R>(&mut self, handler: F)
where F: Fn(HttpRequest<S>) -> R + 'static, where F: Fn(HttpRequest<S>) -> R + 'static,
R: Into<Reply> + 'static, { R: FromRequest + 'static, {
self.routes.insert(Method::POST, Box::new(WrapHandler::new(handler))); self.routes.insert(Method::POST, Box::new(WrapHandler::new(handler)));
} }
/// Register handler for `PUT` method. /// Register handler for `PUT` method.
pub fn put<F, R>(&mut self, handler: F) pub fn put<F, R>(&mut self, handler: F)
where F: Fn(HttpRequest<S>) -> R + 'static, where F: Fn(HttpRequest<S>) -> R + 'static,
R: Into<Reply> + 'static, { R: FromRequest + 'static, {
self.routes.insert(Method::PUT, Box::new(WrapHandler::new(handler))); self.routes.insert(Method::PUT, Box::new(WrapHandler::new(handler)));
} }
/// Register handler for `DELETE` method. /// Register handler for `DELETE` method.
pub fn delete<F, R>(&mut self, handler: F) pub fn delete<F, R>(&mut self, handler: F)
where F: Fn(HttpRequest<S>) -> R + 'static, where F: Fn(HttpRequest<S>) -> R + 'static,
R: Into<Reply> + 'static, { R: FromRequest + 'static, {
self.routes.insert(Method::DELETE, Box::new(WrapHandler::new(handler))); self.routes.insert(Method::DELETE, Box::new(WrapHandler::new(handler)));
} }
} }

View file

@ -14,16 +14,20 @@ use httpresponse::HttpResponse;
pub trait Handler<S>: 'static { pub trait Handler<S>: 'static {
/// The type of value that handler will return. /// The type of value that handler will return.
type Result: Into<Reply>; type Result: FromRequest;
/// Handle request /// Handle request
fn handle(&self, req: HttpRequest<S>) -> Self::Result; fn handle(&self, req: HttpRequest<S>) -> Self::Result;
} }
pub trait FromRequest {
fn from_request(self, req: HttpRequest) -> Reply;
}
/// Handler<S> for Fn() /// Handler<S> for Fn()
impl<F, R, S> Handler<S> for F impl<F, R, S> Handler<S> for F
where F: Fn(HttpRequest<S>) -> R + 'static, where F: Fn(HttpRequest<S>) -> R + 'static,
R: Into<Reply> + 'static R: FromRequest + 'static
{ {
type Result = R; type Result = R;
@ -67,28 +71,41 @@ impl Reply {
} }
} }
#[cfg(not(actix_nightly))] impl FromRequest for Reply {
impl<T: Into<HttpResponse>> From<T> for Reply fn from_request(self, _: HttpRequest) -> Reply {
{ self
fn from(item: T) -> Self { }
Reply(ReplyItem::Message(item.into())) }
impl FromRequest for HttpResponse {
fn from_request(self, _: HttpRequest) -> Reply {
Reply(ReplyItem::Message(self))
} }
} }
#[cfg(actix_nightly)] #[cfg(actix_nightly)]
default impl<T: Into<HttpResponse>> From<T> for Reply default impl<T: FromRequest> FromRequest for T
{ {
fn from(item: T) -> Self { fn from_request(self, req: HttpRequest) -> Reply {
Reply(ReplyItem::Message(item.into())) self.from_request(req)
} }
} }
#[cfg(actix_nightly)] #[cfg(actix_nightly)]
default impl<T: Into<HttpResponse>, E: Into<Error>> From<StdResult<T, E>> for Reply { default impl<T: FromRequest, E: Into<Error>> FromRequest for StdResult<T, E> {
fn from(res: StdResult<T, E>) -> Self { fn from_request(self, req: HttpRequest) -> Reply {
match res { match self {
Ok(val) => val.into().into(), Ok(val) => val.from_request(req),
Err(err) => err.into().into(), Err(err) => Reply(ReplyItem::Message(err.into().into())),
}
}
}
impl<E: Into<Error>> FromRequest for StdResult<Reply, E> {
fn from_request(self, _: HttpRequest) -> Reply {
match self {
Ok(val) => val,
Err(err) => Reply(ReplyItem::Message(err.into().into())),
} }
} }
} }
@ -97,22 +114,37 @@ impl<E: Into<Error>> From<StdResult<Reply, E>> for Reply {
fn from(res: StdResult<Reply, E>) -> Self { fn from(res: StdResult<Reply, E>) -> Self {
match res { match res {
Ok(val) => val, Ok(val) => val,
Err(err) => err.into().into(), Err(err) => Reply(ReplyItem::Message(err.into().into())),
} }
} }
} }
impl<A: Actor<Context=HttpContext<A, S>>, S: 'static> From<HttpContext<A, S>> for Reply impl<E: Into<Error>> FromRequest for StdResult<HttpResponse, E> {
{ fn from_request(self, _: HttpRequest) -> Reply {
fn from(item: HttpContext<A, S>) -> Self { match self {
Reply(ReplyItem::Actor(Box::new(item))) Ok(val) => Reply(ReplyItem::Message(val)),
Err(err) => Reply(ReplyItem::Message(err.into().into())),
}
} }
} }
impl From<Box<Future<Item=HttpResponse, Error=Error>>> for Reply impl<A: Actor<Context=HttpContext<A, S>>, S: 'static> FromRequest for HttpContext<A, S>
{ {
fn from(item: Box<Future<Item=HttpResponse, Error=Error>>) -> Self { fn from_request(self, _: HttpRequest) -> Reply {
Reply(ReplyItem::Future(item)) Reply(ReplyItem::Actor(Box::new(self)))
}
}
impl<A: Actor<Context=HttpContext<A, S>>, S: 'static> From<HttpContext<A, S>> for Reply {
fn from(ctx: HttpContext<A, S>) -> Reply {
Reply(ReplyItem::Actor(Box::new(ctx)))
}
}
impl FromRequest for Box<Future<Item=HttpResponse, Error=Error>>
{
fn from_request(self, _: HttpRequest) -> Reply {
Reply(ReplyItem::Future(self))
} }
} }
@ -125,7 +157,7 @@ pub(crate) trait RouteHandler<S>: 'static {
pub(crate) pub(crate)
struct WrapHandler<S, H, R> struct WrapHandler<S, H, R>
where H: Handler<S, Result=R>, where H: Handler<S, Result=R>,
R: Into<Reply>, R: FromRequest,
S: 'static, S: 'static,
{ {
h: H, h: H,
@ -134,7 +166,7 @@ struct WrapHandler<S, H, R>
impl<S, H, R> WrapHandler<S, H, R> impl<S, H, R> WrapHandler<S, H, R>
where H: Handler<S, Result=R>, where H: Handler<S, Result=R>,
R: Into<Reply>, R: FromRequest,
S: 'static, S: 'static,
{ {
pub fn new(h: H) -> Self { pub fn new(h: H) -> Self {
@ -144,11 +176,12 @@ impl<S, H, R> WrapHandler<S, H, R>
impl<S, H, R> RouteHandler<S> for WrapHandler<S, H, R> impl<S, H, R> RouteHandler<S> for WrapHandler<S, H, R>
where H: Handler<S, Result=R>, where H: Handler<S, Result=R>,
R: Into<Reply> + 'static, R: FromRequest + 'static,
S: 'static, S: 'static,
{ {
fn handle(&self, req: HttpRequest<S>) -> Reply { fn handle(&self, req: HttpRequest<S>) -> Reply {
self.h.handle(req).into() let req2 = req.clone_without_state();
self.h.handle(req).from_request(req2)
} }
} }