1
0
mirror of https://github.com/fafhrd91/actix-web synced 2024-11-24 16:32:59 +01:00
actix-web/src/app.rs

952 lines
29 KiB
Rust
Raw Normal View History

2019-03-02 07:51:32 +01:00
use std::cell::RefCell;
use std::marker::PhantomData;
use std::rc::Rc;
2019-03-03 01:24:14 +01:00
use actix_http::body::{Body, MessageBody};
2019-03-02 07:51:32 +01:00
use actix_http::{Extensions, PayloadStream, Request, Response};
use actix_router::{Path, ResourceDef, ResourceInfo, Router, Url};
2019-03-02 08:59:44 +01:00
use actix_service::boxed::{self, BoxedNewService, BoxedService};
2019-03-02 07:51:32 +01:00
use actix_service::{
fn_service, AndThenNewService, ApplyTransform, IntoNewService, IntoTransform,
NewService, Service, Transform,
2019-03-02 07:51:32 +01:00
};
use futures::future::{ok, Either, FutureResult};
use futures::{Async, Future, IntoFuture, Poll};
use crate::config::AppConfig;
2019-03-04 20:47:53 +01:00
use crate::guard::Guard;
2019-03-02 07:51:32 +01:00
use crate::resource::Resource;
use crate::route::Route;
use crate::service::{
HttpServiceFactory, ServiceFactory, ServiceFactoryWrapper, ServiceRequest,
ServiceResponse,
};
2019-03-02 07:51:32 +01:00
use crate::state::{State, StateFactory, StateFactoryResult};
2019-03-04 20:47:53 +01:00
type Guards = Vec<Box<Guard>>;
2019-03-02 08:59:44 +01:00
type HttpService<P> = BoxedService<ServiceRequest<P>, ServiceResponse, ()>;
type HttpNewService<P> = BoxedNewService<(), ServiceRequest<P>, ServiceResponse, (), ()>;
2019-03-02 07:51:32 +01:00
type BoxedResponse = Box<Future<Item = ServiceResponse, Error = ()>>;
2019-03-03 07:03:45 +01:00
/// Application builder - structure that follows the builder pattern
/// for building application instances.
2019-03-02 20:53:05 +01:00
pub struct App<P, T>
where
2019-03-05 19:08:08 +01:00
T: NewService<ServiceRequest, Response = ServiceRequest<P>>,
2019-03-02 20:53:05 +01:00
{
chain: T,
2019-03-02 07:51:32 +01:00
extensions: Extensions,
state: Vec<Box<StateFactory>>,
2019-03-02 20:53:05 +01:00
_t: PhantomData<(P,)>,
2019-03-02 07:51:32 +01:00
}
2019-03-02 20:53:05 +01:00
impl App<PayloadStream, AppChain> {
/// Create application builder with empty state. Application can
2019-03-02 07:51:32 +01:00
/// be configured with a builder-like pattern.
pub fn new() -> Self {
2019-03-02 20:53:05 +01:00
App {
chain: AppChain,
extensions: Extensions::new(),
state: Vec::new(),
_t: PhantomData,
}
2019-03-02 07:51:32 +01:00
}
}
2019-03-02 20:53:05 +01:00
impl<P, T> App<P, T>
where
P: 'static,
T: NewService<
2019-03-05 19:08:08 +01:00
ServiceRequest,
2019-03-02 20:53:05 +01:00
Response = ServiceRequest<P>,
Error = (),
InitError = (),
>,
{
2019-03-03 07:03:45 +01:00
/// Set application state. Applicatin state could be accessed
/// by using `State<T>` extractor where `T` is state type.
2019-03-02 07:51:32 +01:00
///
/// **Note**: http server accepts an application factory rather than
/// an application instance. Http server constructs an application
/// instance for each thread, thus application state must be constructed
/// multiple times. If you want to share state between different
/// threads, a shared object should be used, e.g. `Arc`. Application
/// state does not need to be `Send` or `Sync`.
2019-03-03 07:11:24 +01:00
///
/// ```rust
/// use std::cell::Cell;
/// use actix_web::{web, State, App};
///
/// struct MyState {
/// counter: Cell<usize>,
/// }
///
/// fn index(state: State<MyState>) {
/// state.counter.set(state.counter.get() + 1);
/// }
///
/// fn main() {
/// let app = App::new()
/// .state(MyState{ counter: Cell::new(0) })
/// .service(
/// web::resource("/index.html").route(
/// web::get().to(index)));
2019-03-03 07:11:24 +01:00
/// }
/// ```
2019-03-02 07:51:32 +01:00
pub fn state<S: 'static>(mut self, state: S) -> Self {
self.state.push(Box::new(State::new(state)));
self
}
2019-03-03 07:03:45 +01:00
/// Set application state factory. This function is
2019-03-02 07:51:32 +01:00
/// similar to `.state()` but it accepts state factory. State get
/// constructed asynchronously during application initialization.
pub fn state_factory<F, Out>(mut self, state: F) -> Self
2019-03-02 07:51:32 +01:00
where
F: Fn() -> Out + 'static,
Out: IntoFuture + 'static,
Out::Error: std::fmt::Debug,
{
self.state.push(Box::new(state));
2019-03-02 07:51:32 +01:00
self
}
2019-03-02 20:53:05 +01:00
/// Register a middleware.
pub fn middleware<M, B, F>(
2019-03-02 20:53:05 +01:00
self,
mw: F,
) -> AppRouter<
T,
P,
B,
2019-03-02 20:53:05 +01:00
impl NewService<
2019-03-05 19:08:08 +01:00
ServiceRequest<P>,
Response = ServiceResponse<B>,
2019-03-02 20:53:05 +01:00
Error = (),
InitError = (),
>,
>
where
2019-03-05 06:37:57 +01:00
M: Transform<
2019-03-03 17:24:09 +01:00
AppRouting<P>,
2019-03-05 19:08:08 +01:00
ServiceRequest<P>,
Response = ServiceResponse<B>,
2019-03-02 20:53:05 +01:00
Error = (),
InitError = (),
>,
2019-03-05 19:08:08 +01:00
F: IntoTransform<M, AppRouting<P>, ServiceRequest<P>>,
2019-03-02 20:53:05 +01:00
{
let fref = Rc::new(RefCell::new(None));
2019-03-05 06:37:57 +01:00
let endpoint = ApplyTransform::new(mw, AppEntry::new(fref.clone()));
2019-03-02 20:53:05 +01:00
AppRouter {
endpoint,
chain: self.chain,
state: self.state,
services: Vec::new(),
default: None,
factory_ref: fref,
extensions: self.extensions,
_t: PhantomData,
}
}
/// Register a request modifier. It can modify any request parameters
/// including payload stream.
pub fn chain<C, F, P1>(
self,
chain: C,
) -> App<
P1,
impl NewService<
2019-03-05 19:08:08 +01:00
ServiceRequest,
2019-03-02 20:53:05 +01:00
Response = ServiceRequest<P1>,
Error = (),
InitError = (),
>,
>
where
C: NewService<
2019-03-05 19:08:08 +01:00
ServiceRequest<P>,
2019-03-02 20:53:05 +01:00
Response = ServiceRequest<P1>,
Error = (),
InitError = (),
>,
2019-03-05 19:08:08 +01:00
F: IntoNewService<C, ServiceRequest<P>>,
2019-03-02 20:53:05 +01:00
{
let chain = self.chain.and_then(chain.into_new_service());
2019-03-02 07:51:32 +01:00
App {
2019-03-02 20:53:05 +01:00
chain,
state: self.state,
extensions: self.extensions,
_t: PhantomData,
}
}
2019-03-04 06:02:01 +01:00
/// Configure route for a specific path.
///
/// This is a simplified version of the `App::service()` method.
/// This method can not be could multiple times, in that case
/// multiple resources with one route would be registered for same resource path.
///
/// ```rust
/// use actix_web::{web, App, HttpResponse, extract::Path};
///
/// fn index(data: Path<(String, String)>) -> &'static str {
/// "Welcome!"
/// }
///
/// fn main() {
/// let app = App::new()
/// .route("/test1", web::get().to(index))
/// .route("/test2", web::post().to(|| HttpResponse::MethodNotAllowed()));
/// }
/// ```
pub fn route(
self,
path: &str,
mut route: Route<P>,
) -> AppRouter<T, P, Body, AppEntry<P>> {
self.service(
Resource::new(path)
.add_guards(route.take_guards())
.route(route),
)
}
/// Register http service.
pub fn service<F>(self, service: F) -> AppRouter<T, P, Body, AppEntry<P>>
where
F: HttpServiceFactory<P> + 'static,
{
let fref = Rc::new(RefCell::new(None));
AppRouter {
chain: self.chain,
default: None,
endpoint: AppEntry::new(fref.clone()),
factory_ref: fref,
extensions: self.extensions,
state: self.state,
services: vec![Box::new(ServiceFactoryWrapper::new(service))],
_t: PhantomData,
}
}
2019-03-04 06:02:01 +01:00
/// Set server host name.
///
/// Host name is used by application router aa a hostname for url
/// generation. Check [ConnectionInfo](./dev/struct.ConnectionInfo.
/// html#method.host) documentation for more information.
///
/// By default host name is set to a "localhost" value.
pub fn hostname(self, _val: &str) -> Self {
// self.host = val.to_owned();
self
}
2019-03-02 07:51:32 +01:00
}
2019-03-03 23:45:56 +01:00
/// Application router builder - Structure that follows the builder pattern
/// for building application instances.
2019-03-02 20:53:05 +01:00
pub struct AppRouter<C, P, B, T> {
chain: C,
endpoint: T,
services: Vec<Box<ServiceFactory<P>>>,
default: Option<Rc<HttpNewService<P>>>,
2019-03-03 17:24:09 +01:00
factory_ref: Rc<RefCell<Option<AppRoutingFactory<P>>>>,
2019-03-02 20:53:05 +01:00
extensions: Extensions,
state: Vec<Box<StateFactory>>,
_t: PhantomData<(P, B)>,
}
impl<C, P, B, T> AppRouter<C, P, B, T>
2019-03-02 07:51:32 +01:00
where
P: 'static,
B: MessageBody,
T: NewService<
2019-03-05 19:08:08 +01:00
ServiceRequest<P>,
2019-03-02 07:51:32 +01:00
Response = ServiceResponse<B>,
Error = (),
InitError = (),
>,
{
/// Configure route for a specific path.
2019-03-04 06:02:01 +01:00
///
/// This is a simplified version of the `App::service()` method.
/// This method can not be could multiple times, in that case
/// multiple resources with one route would be registered for same resource path.
2019-03-04 06:02:01 +01:00
///
/// ```rust
/// use actix_web::{web, App, HttpResponse, extract::Path};
2019-03-04 06:02:01 +01:00
///
/// fn index(data: Path<(String, String)>) -> &'static str {
/// "Welcome!"
2019-03-04 06:02:01 +01:00
/// }
2019-03-02 07:51:32 +01:00
///
/// fn main() {
2019-03-03 07:03:45 +01:00
/// let app = App::new()
/// .route("/test1", web::get().to(index))
/// .route("/test2", web::post().to(|| HttpResponse::MethodNotAllowed()));
2019-03-02 07:51:32 +01:00
/// }
/// ```
pub fn route(self, path: &str, mut route: Route<P>) -> Self {
self.service(
Resource::new(path)
.add_guards(route.take_guards())
.route(route),
)
2019-03-02 07:51:32 +01:00
}
/// Register http service.
2019-03-02 07:51:32 +01:00
///
/// Http service is any type that implements `HttpServiceFactory` trait.
///
/// Actix web provides several services implementations:
///
/// * *Resource* is an entry in route table which corresponds to requested URL.
/// * *Scope* is a set of resources with common root path.
/// * "StaticFiles" is a service for static files support
pub fn service<F>(mut self, factory: F) -> Self
2019-03-02 07:51:32 +01:00
where
F: HttpServiceFactory<P> + 'static,
2019-03-02 07:51:32 +01:00
{
self.services
.push(Box::new(ServiceFactoryWrapper::new(factory)));
2019-03-02 07:51:32 +01:00
self
}
/// Register a middleware.
pub fn middleware<M, B1, F>(
self,
mw: F,
2019-03-02 20:53:05 +01:00
) -> AppRouter<
C,
2019-03-02 07:51:32 +01:00
P,
B1,
impl NewService<
2019-03-05 19:08:08 +01:00
ServiceRequest<P>,
2019-03-02 07:51:32 +01:00
Response = ServiceResponse<B1>,
Error = (),
InitError = (),
>,
>
where
2019-03-05 06:37:57 +01:00
M: Transform<
2019-03-02 07:51:32 +01:00
T::Service,
2019-03-05 19:08:08 +01:00
ServiceRequest<P>,
2019-03-02 07:51:32 +01:00
Response = ServiceResponse<B1>,
Error = (),
InitError = (),
>,
B1: MessageBody,
2019-03-05 19:08:08 +01:00
F: IntoTransform<M, T::Service, ServiceRequest<P>>,
2019-03-02 07:51:32 +01:00
{
2019-03-05 06:37:57 +01:00
let endpoint = ApplyTransform::new(mw, self.endpoint);
2019-03-02 20:53:05 +01:00
AppRouter {
2019-03-02 07:51:32 +01:00
endpoint,
2019-03-02 20:53:05 +01:00
chain: self.chain,
2019-03-02 07:51:32 +01:00
state: self.state,
services: self.services,
default: self.default,
factory_ref: self.factory_ref,
2019-03-02 20:53:05 +01:00
extensions: self.extensions,
2019-03-02 07:51:32 +01:00
_t: PhantomData,
}
}
/// Default resource to be used if no matching route could be found.
///
/// Default resource works with resources only and does not work with
/// custom services.
pub fn default_resource<F, U>(mut self, f: F) -> Self
where
F: FnOnce(Resource<P>) -> Resource<P, U>,
U: NewService<
ServiceRequest<P>,
Response = ServiceResponse,
Error = (),
InitError = (),
> + 'static,
{
// create and configure default resource
self.default = Some(Rc::new(boxed::new_service(
f(Resource::new("")).into_new_service().map_init_err(|_| ()),
)));
self
}
2019-03-02 07:51:32 +01:00
/// Register an external resource.
///
/// External resources are useful for URL generation purposes only
/// and are never considered for matching at request time. Calls to
/// `HttpRequest::url_for()` will work as expected.
///
/// ```rust,ignore
/// # extern crate actix_web;
/// use actix_web::{App, HttpRequest, HttpResponse, Result};
///
/// fn index(req: &HttpRequest) -> Result<HttpResponse> {
/// let url = req.url_for("youtube", &["oHg5SJYRHA0"])?;
/// assert_eq!(url.as_str(), "https://youtube.com/watch/oHg5SJYRHA0");
/// Ok(HttpResponse::Ok().into())
/// }
///
/// fn main() {
/// let app = App::new()
/// .resource("/index.html", |r| r.get().f(index))
/// .external_resource("youtube", "https://youtube.com/watch/{video_id}")
/// .finish();
/// }
/// ```
pub fn external_resource<N, U>(self, _name: N, _url: U) -> Self
where
N: AsRef<str>,
U: AsRef<str>,
{
// self.parts
// .as_mut()
// .expect("Use after finish")
// .router
// .register_external(name.as_ref(), ResourceDef::external(url.as_ref()));
self
}
}
2019-03-02 20:53:05 +01:00
impl<C, T, P: 'static, B: MessageBody>
2019-03-05 19:08:08 +01:00
IntoNewService<AndThenNewService<AppInit<C, P>, T>, Request>
for AppRouter<C, P, B, T>
2019-03-02 07:51:32 +01:00
where
T: NewService<
2019-03-05 19:08:08 +01:00
ServiceRequest<P>,
2019-03-02 07:51:32 +01:00
Response = ServiceResponse<B>,
Error = (),
InitError = (),
>,
2019-03-02 20:53:05 +01:00
C: NewService<
2019-03-05 19:08:08 +01:00
ServiceRequest,
2019-03-02 20:53:05 +01:00
Response = ServiceRequest<P>,
Error = (),
InitError = (),
>,
2019-03-02 07:51:32 +01:00
{
2019-03-05 19:08:08 +01:00
fn into_new_service(self) -> AndThenNewService<AppInit<C, P>, T> {
2019-03-02 07:51:32 +01:00
// update resource default service
let default = self.default.unwrap_or_else(|| {
Rc::new(boxed::new_service(fn_service(|req: ServiceRequest<P>| {
Ok(req.into_response(Response::NotFound().finish()))
})))
});
let mut config = AppConfig::new(
"127.0.0.1:8080".parse().unwrap(),
"localhost:8080".to_owned(),
false,
default.clone(),
);
// register services
self.services
.into_iter()
.for_each(|mut srv| srv.register(&mut config));
2019-03-02 07:51:32 +01:00
// set factory
2019-03-03 17:24:09 +01:00
*self.factory_ref.borrow_mut() = Some(AppRoutingFactory {
2019-03-07 04:19:27 +01:00
default,
2019-03-04 20:47:53 +01:00
services: Rc::new(
config
.into_services()
2019-03-04 20:47:53 +01:00
.into_iter()
.map(|(rdef, srv, guards)| (rdef, srv, RefCell::new(guards)))
.collect(),
),
2019-03-02 07:51:32 +01:00
});
2019-03-02 20:53:05 +01:00
AppInit {
chain: self.chain,
2019-03-02 07:51:32 +01:00
state: self.state,
extensions: Rc::new(RefCell::new(Rc::new(self.extensions))),
}
.and_then(self.endpoint)
}
}
2019-03-03 17:24:09 +01:00
pub struct AppRoutingFactory<P> {
2019-03-04 20:47:53 +01:00
services: Rc<Vec<(ResourceDef, HttpNewService<P>, RefCell<Option<Guards>>)>>,
default: Rc<HttpNewService<P>>,
2019-03-02 07:51:32 +01:00
}
2019-03-05 19:08:08 +01:00
impl<P: 'static> NewService<ServiceRequest<P>> for AppRoutingFactory<P> {
2019-03-02 07:51:32 +01:00
type Response = ServiceResponse;
type Error = ();
type InitError = ();
2019-03-03 17:24:09 +01:00
type Service = AppRouting<P>;
type Future = AppRoutingFactoryResponse<P>;
2019-03-02 07:51:32 +01:00
fn new_service(&self, _: &()) -> Self::Future {
2019-03-03 17:24:09 +01:00
AppRoutingFactoryResponse {
2019-03-02 07:51:32 +01:00
fut: self
.services
.iter()
2019-03-04 20:47:53 +01:00
.map(|(path, service, guards)| {
2019-03-03 17:24:09 +01:00
CreateAppRoutingItem::Future(
2019-03-02 07:51:32 +01:00
Some(path.clone()),
2019-03-04 20:47:53 +01:00
guards.borrow_mut().take(),
2019-03-02 07:51:32 +01:00
service.new_service(&()),
)
})
.collect(),
2019-03-04 06:02:01 +01:00
default: None,
default_fut: Some(self.default.new_service(&())),
2019-03-02 07:51:32 +01:00
}
}
}
2019-03-02 08:59:44 +01:00
type HttpServiceFut<P> = Box<Future<Item = HttpService<P>, Error = ()>>;
2019-03-02 07:51:32 +01:00
/// Create app service
#[doc(hidden)]
2019-03-03 17:24:09 +01:00
pub struct AppRoutingFactoryResponse<P> {
fut: Vec<CreateAppRoutingItem<P>>,
2019-03-04 06:02:01 +01:00
default: Option<HttpService<P>>,
default_fut: Option<Box<Future<Item = HttpService<P>, Error = ()>>>,
2019-03-02 07:51:32 +01:00
}
2019-03-03 17:24:09 +01:00
enum CreateAppRoutingItem<P> {
2019-03-04 20:47:53 +01:00
Future(Option<ResourceDef>, Option<Guards>, HttpServiceFut<P>),
Service(ResourceDef, Option<Guards>, HttpService<P>),
2019-03-02 07:51:32 +01:00
}
2019-03-03 17:24:09 +01:00
impl<P> Future for AppRoutingFactoryResponse<P> {
type Item = AppRouting<P>;
2019-03-02 07:51:32 +01:00
type Error = ();
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
let mut done = true;
2019-03-04 06:02:01 +01:00
if let Some(ref mut fut) = self.default_fut {
match fut.poll()? {
Async::Ready(default) => self.default = Some(default),
Async::NotReady => done = false,
}
}
2019-03-02 07:51:32 +01:00
// poll http services
for item in &mut self.fut {
let res = match item {
2019-03-04 20:47:53 +01:00
CreateAppRoutingItem::Future(
ref mut path,
ref mut guards,
ref mut fut,
) => match fut.poll()? {
Async::Ready(service) => {
Some((path.take().unwrap(), guards.take(), service))
2019-03-02 07:51:32 +01:00
}
2019-03-04 20:47:53 +01:00
Async::NotReady => {
done = false;
None
}
},
CreateAppRoutingItem::Service(_, _, _) => continue,
2019-03-02 07:51:32 +01:00
};
2019-03-04 20:47:53 +01:00
if let Some((path, guards, service)) = res {
*item = CreateAppRoutingItem::Service(path, guards, service);
2019-03-02 07:51:32 +01:00
}
}
if done {
let router = self
.fut
.drain(..)
.fold(Router::build(), |mut router, item| {
match item {
2019-03-04 20:47:53 +01:00
CreateAppRoutingItem::Service(path, guards, service) => {
router.rdef(path, service);
router.set_user_data(guards);
2019-03-02 07:51:32 +01:00
}
2019-03-04 20:47:53 +01:00
CreateAppRoutingItem::Future(_, _, _) => unreachable!(),
2019-03-02 07:51:32 +01:00
}
router
});
2019-03-03 17:24:09 +01:00
Ok(Async::Ready(AppRouting {
2019-03-02 07:51:32 +01:00
ready: None,
2019-03-04 06:02:01 +01:00
router: router.finish(),
default: self.default.take(),
2019-03-02 07:51:32 +01:00
}))
} else {
Ok(Async::NotReady)
}
}
}
2019-03-03 17:24:09 +01:00
pub struct AppRouting<P> {
2019-03-04 20:47:53 +01:00
router: Router<HttpService<P>, Guards>,
2019-03-02 07:51:32 +01:00
ready: Option<(ServiceRequest<P>, ResourceInfo)>,
2019-03-04 06:02:01 +01:00
default: Option<HttpService<P>>,
2019-03-02 07:51:32 +01:00
}
2019-03-05 19:08:08 +01:00
impl<P> Service<ServiceRequest<P>> for AppRouting<P> {
2019-03-02 07:51:32 +01:00
type Response = ServiceResponse;
type Error = ();
type Future = Either<BoxedResponse, FutureResult<Self::Response, Self::Error>>;
fn poll_ready(&mut self) -> Poll<(), Self::Error> {
if self.ready.is_none() {
Ok(Async::Ready(()))
} else {
Ok(Async::NotReady)
}
}
fn call(&mut self, mut req: ServiceRequest<P>) -> Self::Future {
2019-03-04 20:47:53 +01:00
let res = self.router.recognize_mut_checked(&mut req, |req, guards| {
if let Some(ref guards) = guards {
for f in guards {
if !f.check(req.head()) {
return false;
}
}
}
true
});
if let Some((srv, _info)) = res {
2019-03-02 07:51:32 +01:00
Either::A(srv.call(req))
2019-03-04 06:02:01 +01:00
} else if let Some(ref mut default) = self.default {
Either::A(default.call(req))
2019-03-02 07:51:32 +01:00
} else {
let req = req.into_request();
Either::B(ok(ServiceResponse::new(req, Response::NotFound().finish())))
}
}
}
#[doc(hidden)]
2019-03-03 17:24:09 +01:00
/// Wrapper service for routing
2019-03-02 07:51:32 +01:00
pub struct AppEntry<P> {
2019-03-03 17:24:09 +01:00
factory: Rc<RefCell<Option<AppRoutingFactory<P>>>>,
2019-03-02 07:51:32 +01:00
}
impl<P> AppEntry<P> {
2019-03-03 17:24:09 +01:00
fn new(factory: Rc<RefCell<Option<AppRoutingFactory<P>>>>) -> Self {
2019-03-02 07:51:32 +01:00
AppEntry { factory }
}
}
2019-03-05 19:08:08 +01:00
impl<P: 'static> NewService<ServiceRequest<P>> for AppEntry<P> {
2019-03-02 07:51:32 +01:00
type Response = ServiceResponse;
type Error = ();
type InitError = ();
2019-03-03 17:24:09 +01:00
type Service = AppRouting<P>;
type Future = AppRoutingFactoryResponse<P>;
2019-03-02 07:51:32 +01:00
fn new_service(&self, _: &()) -> Self::Future {
self.factory.borrow_mut().as_mut().unwrap().new_service(&())
}
}
2019-03-02 20:53:05 +01:00
#[doc(hidden)]
pub struct AppChain;
2019-03-05 19:08:08 +01:00
impl NewService<ServiceRequest> for AppChain {
type Response = ServiceRequest;
2019-03-02 20:53:05 +01:00
type Error = ();
type InitError = ();
type Service = AppChain;
type Future = FutureResult<Self::Service, Self::InitError>;
fn new_service(&self, _: &()) -> Self::Future {
ok(AppChain)
}
}
2019-03-05 19:08:08 +01:00
impl Service<ServiceRequest> for AppChain {
type Response = ServiceRequest;
2019-03-02 20:53:05 +01:00
type Error = ();
type Future = FutureResult<Self::Response, Self::Error>;
2019-03-03 17:24:09 +01:00
#[inline]
2019-03-02 20:53:05 +01:00
fn poll_ready(&mut self) -> Poll<(), Self::Error> {
Ok(Async::Ready(()))
}
2019-03-03 17:24:09 +01:00
#[inline]
2019-03-05 19:08:08 +01:00
fn call(&mut self, req: ServiceRequest) -> Self::Future {
2019-03-02 20:53:05 +01:00
ok(req)
}
}
2019-03-03 17:24:09 +01:00
/// Service factory to convert `Request` to a `ServiceRequest<S>`.
/// It also executes state factories.
2019-03-02 20:53:05 +01:00
pub struct AppInit<C, P>
where
2019-03-05 19:08:08 +01:00
C: NewService<ServiceRequest, Response = ServiceRequest<P>>,
2019-03-02 20:53:05 +01:00
{
chain: C,
state: Vec<Box<StateFactory>>,
extensions: Rc<RefCell<Rc<Extensions>>>,
}
2019-03-05 19:08:08 +01:00
impl<C, P: 'static> NewService<Request> for AppInit<C, P>
2019-03-02 20:53:05 +01:00
where
2019-03-05 19:08:08 +01:00
C: NewService<ServiceRequest, Response = ServiceRequest<P>, InitError = ()>,
2019-03-02 20:53:05 +01:00
{
type Response = ServiceRequest<P>;
type Error = C::Error;
type InitError = C::InitError;
type Service = AppInitService<C::Service, P>;
type Future = AppInitResult<C, P>;
fn new_service(&self, _: &()) -> Self::Future {
AppInitResult {
chain: self.chain.new_service(&()),
state: self.state.iter().map(|s| s.construct()).collect(),
extensions: self.extensions.clone(),
}
}
}
#[doc(hidden)]
pub struct AppInitResult<C, P>
where
2019-03-05 19:08:08 +01:00
C: NewService<ServiceRequest, Response = ServiceRequest<P>, InitError = ()>,
2019-03-02 20:53:05 +01:00
{
chain: C::Future,
state: Vec<Box<StateFactoryResult>>,
extensions: Rc<RefCell<Rc<Extensions>>>,
}
impl<C, P> Future for AppInitResult<C, P>
where
2019-03-05 19:08:08 +01:00
C: NewService<ServiceRequest, Response = ServiceRequest<P>, InitError = ()>,
2019-03-02 20:53:05 +01:00
{
type Item = AppInitService<C::Service, P>;
type Error = C::InitError;
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
if let Some(extensions) = Rc::get_mut(&mut *self.extensions.borrow_mut()) {
let mut idx = 0;
while idx < self.state.len() {
if let Async::Ready(_) = self.state[idx].poll_result(extensions)? {
self.state.remove(idx);
} else {
idx += 1;
}
}
if !self.state.is_empty() {
return Ok(Async::NotReady);
}
} else {
log::warn!("Multiple copies of app extensions exists");
}
let chain = futures::try_ready!(self.chain.poll());
Ok(Async::Ready(AppInitService {
chain,
extensions: self.extensions.borrow().clone(),
}))
}
}
/// Service to convert `Request` to a `ServiceRequest<S>`
pub struct AppInitService<C, P>
where
2019-03-05 19:08:08 +01:00
C: Service<ServiceRequest, Response = ServiceRequest<P>>,
2019-03-02 20:53:05 +01:00
{
chain: C,
extensions: Rc<Extensions>,
}
2019-03-05 19:08:08 +01:00
impl<C, P> Service<Request> for AppInitService<C, P>
2019-03-02 20:53:05 +01:00
where
2019-03-05 19:08:08 +01:00
C: Service<ServiceRequest, Response = ServiceRequest<P>>,
2019-03-02 20:53:05 +01:00
{
type Response = ServiceRequest<P>;
type Error = C::Error;
type Future = C::Future;
fn poll_ready(&mut self) -> Poll<(), Self::Error> {
self.chain.poll_ready()
}
2019-03-05 19:08:08 +01:00
fn call(&mut self, req: Request) -> Self::Future {
2019-03-02 20:53:05 +01:00
let req = ServiceRequest::new(
Path::new(Url::new(req.uri().clone())),
req,
self.extensions.clone(),
);
self.chain.call(req)
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::http::{Method, StatusCode};
2019-03-07 04:19:27 +01:00
use crate::test::{block_on, init_service, TestRequest};
use crate::{web, HttpResponse, State};
#[test]
fn test_default_resource() {
let mut srv = init_service(
App::new().service(web::resource("/test").to(|| HttpResponse::Ok())),
2019-03-06 04:03:59 +01:00
);
let req = TestRequest::with_uri("/test").to_request();
let resp = block_on(srv.call(req)).unwrap();
assert_eq!(resp.status(), StatusCode::OK);
let req = TestRequest::with_uri("/blah").to_request();
let resp = block_on(srv.call(req)).unwrap();
assert_eq!(resp.status(), StatusCode::NOT_FOUND);
let mut srv = init_service(
2019-03-06 04:03:59 +01:00
App::new()
.service(web::resource("/test").to(|| HttpResponse::Ok()))
.service(
web::resource("/test2")
.default_resource(|r| r.to(|| HttpResponse::Created()))
.route(web::get().to(|| HttpResponse::Ok())),
)
2019-03-06 04:03:59 +01:00
.default_resource(|r| r.to(|| HttpResponse::MethodNotAllowed())),
);
let req = TestRequest::with_uri("/blah").to_request();
let resp = block_on(srv.call(req)).unwrap();
assert_eq!(resp.status(), StatusCode::METHOD_NOT_ALLOWED);
let req = TestRequest::with_uri("/test2").to_request();
let resp = block_on(srv.call(req)).unwrap();
assert_eq!(resp.status(), StatusCode::OK);
let req = TestRequest::with_uri("/test2")
.method(Method::POST)
.to_request();
let resp = block_on(srv.call(req)).unwrap();
assert_eq!(resp.status(), StatusCode::CREATED);
}
#[test]
fn test_state() {
let mut srv = init_service(
2019-03-06 04:03:59 +01:00
App::new()
.state(10usize)
.service(web::resource("/").to(|_: State<usize>| HttpResponse::Ok())),
2019-03-06 04:03:59 +01:00
);
let req = TestRequest::default().to_request();
let resp = block_on(srv.call(req)).unwrap();
assert_eq!(resp.status(), StatusCode::OK);
let mut srv = init_service(
2019-03-06 04:03:59 +01:00
App::new()
.state(10u32)
.service(web::resource("/").to(|_: State<usize>| HttpResponse::Ok())),
2019-03-06 04:03:59 +01:00
);
let req = TestRequest::default().to_request();
let resp = block_on(srv.call(req)).unwrap();
assert_eq!(resp.status(), StatusCode::INTERNAL_SERVER_ERROR);
}
#[test]
fn test_state_factory() {
let mut srv = init_service(
2019-03-06 04:03:59 +01:00
App::new()
.state_factory(|| Ok::<_, ()>(10usize))
.service(web::resource("/").to(|_: State<usize>| HttpResponse::Ok())),
2019-03-06 04:03:59 +01:00
);
let req = TestRequest::default().to_request();
let resp = block_on(srv.call(req)).unwrap();
assert_eq!(resp.status(), StatusCode::OK);
let mut srv = init_service(
2019-03-06 04:03:59 +01:00
App::new()
.state_factory(|| Ok::<_, ()>(10u32))
.service(web::resource("/").to(|_: State<usize>| HttpResponse::Ok())),
2019-03-06 04:03:59 +01:00
);
let req = TestRequest::default().to_request();
let resp = block_on(srv.call(req)).unwrap();
assert_eq!(resp.status(), StatusCode::INTERNAL_SERVER_ERROR);
}
// #[test]
// fn test_handler() {
// let app = App::new()
// .handler("/test", |_: &_| HttpResponse::Ok())
// .finish();
// let req = TestRequest::with_uri("/test").request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::OK);
// let req = TestRequest::with_uri("/test/").request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::OK);
// let req = TestRequest::with_uri("/test/app").request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::OK);
// let req = TestRequest::with_uri("/testapp").request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::NOT_FOUND);
// let req = TestRequest::with_uri("/blah").request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::NOT_FOUND);
// }
// #[test]
// fn test_handler2() {
// let app = App::new()
// .handler("test", |_: &_| HttpResponse::Ok())
// .finish();
// let req = TestRequest::with_uri("/test").request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::OK);
// let req = TestRequest::with_uri("/test/").request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::OK);
// let req = TestRequest::with_uri("/test/app").request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::OK);
// let req = TestRequest::with_uri("/testapp").request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::NOT_FOUND);
// let req = TestRequest::with_uri("/blah").request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::NOT_FOUND);
// }
// #[test]
// fn test_route() {
// let app = App::new()
// .route("/test", Method::GET, |_: HttpRequest| HttpResponse::Ok())
// .route("/test", Method::POST, |_: HttpRequest| {
// HttpResponse::Created()
// })
// .finish();
// let req = TestRequest::with_uri("/test").method(Method::GET).request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::OK);
// let req = TestRequest::with_uri("/test")
// .method(Method::POST)
// .request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::CREATED);
// let req = TestRequest::with_uri("/test")
// .method(Method::HEAD)
// .request();
// let resp = app.run(req);
// assert_eq!(resp.as_msg().status(), StatusCode::NOT_FOUND);
// }
}