mirror of
https://github.com/fafhrd91/actix-web
synced 2024-11-28 01:52:57 +01:00
unify route not found handling
This commit is contained in:
parent
b1ae7f95cc
commit
007b7ce62f
@ -526,3 +526,27 @@ predicates match. i.e:
|
|||||||
```rust,ignore
|
```rust,ignore
|
||||||
pred::All(vec![pred::Get(), pred::Header("content-type", "plain/text")])
|
pred::All(vec![pred::Get(), pred::Header("content-type", "plain/text")])
|
||||||
```
|
```
|
||||||
|
|
||||||
|
## Changing the default Not Found response
|
||||||
|
|
||||||
|
If path pattern can not be found in routing table or resource can not find matching
|
||||||
|
route default resource is used. Default response is *NOT FOUND* response.
|
||||||
|
To override *NOT FOUND* resource use `Application::default_resource()` method.
|
||||||
|
This method accepts *configuration function* same as normal resource registration
|
||||||
|
with `Application::resource()` method.
|
||||||
|
|
||||||
|
``rust
|
||||||
|
# extern crate actix_web;
|
||||||
|
# extern crate http;
|
||||||
|
use actix_web::*;
|
||||||
|
use actix_web::httpcodes::*;
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
Application::new()
|
||||||
|
.default_resource(|r|
|
||||||
|
r.method(Method::GET).f(|req| HTTPNotFound);
|
||||||
|
r.route().p(pred::Not(pred::Get()).f(|req| HTTPMethodNotAllowed);
|
||||||
|
})
|
||||||
|
.finish();
|
||||||
|
}
|
||||||
|
```
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
use std::rc::Rc;
|
use std::rc::Rc;
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
use handler::{Reply, RouteHandler};
|
use handler::Reply;
|
||||||
use router::{Router, Pattern};
|
use router::{Router, Pattern};
|
||||||
use resource::Resource;
|
use resource::Resource;
|
||||||
use httprequest::HttpRequest;
|
use httprequest::HttpRequest;
|
||||||
@ -27,9 +27,9 @@ impl<S: 'static> HttpApplication<S> {
|
|||||||
|
|
||||||
pub(crate) fn run(&self, mut req: HttpRequest<S>) -> Reply {
|
pub(crate) fn run(&self, mut req: HttpRequest<S>) -> Reply {
|
||||||
if let Some(h) = self.router.recognize(&mut req) {
|
if let Some(h) = self.router.recognize(&mut req) {
|
||||||
h.handle(req)
|
h.handle(req.clone(), Some(&self.default))
|
||||||
} else {
|
} else {
|
||||||
self.default.handle(req)
|
self.default.handle(req, None)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -196,7 +196,7 @@ impl<S> Application<S> where S: 'static {
|
|||||||
self
|
self
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Default resource is used if no match route could be found.
|
/// Default resource is used if no matched route could be found.
|
||||||
pub fn default_resource<F>(&mut self, f: F) -> &mut Self
|
pub fn default_resource<F>(&mut self, f: F) -> &mut Self
|
||||||
where F: FnOnce(&mut Resource<S>) + 'static
|
where F: FnOnce(&mut Resource<S>) + 'static
|
||||||
{
|
{
|
||||||
|
@ -1,12 +1,13 @@
|
|||||||
use std::marker::PhantomData;
|
use std::marker::PhantomData;
|
||||||
|
|
||||||
use http::Method;
|
use http::{Method, StatusCode};
|
||||||
|
|
||||||
use pred;
|
use pred;
|
||||||
|
use body::Body;
|
||||||
use route::Route;
|
use route::Route;
|
||||||
use handler::{Reply, Handler, FromRequest, RouteHandler};
|
use handler::{Reply, Handler, FromRequest};
|
||||||
use httpcodes::HTTPNotFound;
|
|
||||||
use httprequest::HttpRequest;
|
use httprequest::HttpRequest;
|
||||||
|
use httpresponse::HttpResponse;
|
||||||
|
|
||||||
/// *Resource* is an entry in route table which corresponds to requested URL.
|
/// *Resource* is an entry in route table which corresponds to requested URL.
|
||||||
///
|
///
|
||||||
@ -124,16 +125,19 @@ impl<S: 'static> Resource<S> {
|
|||||||
self.routes.push(Route::default());
|
self.routes.push(Route::default());
|
||||||
self.routes.last_mut().unwrap().f(handler)
|
self.routes.last_mut().unwrap().f(handler)
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
impl<S: 'static> RouteHandler<S> for Resource<S> {
|
pub(crate) fn handle(&self, mut req: HttpRequest<S>, default: Option<&Resource<S>>)
|
||||||
|
-> Reply
|
||||||
fn handle(&self, mut req: HttpRequest<S>) -> Reply {
|
{
|
||||||
for route in &self.routes {
|
for route in &self.routes {
|
||||||
if route.check(&mut req) {
|
if route.check(&mut req) {
|
||||||
return route.handle(req)
|
return route.handle(req)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Reply::response(HTTPNotFound)
|
if let Some(resource) = default {
|
||||||
|
resource.handle(req, None)
|
||||||
|
} else {
|
||||||
|
Reply::response(HttpResponse::new(StatusCode::NOT_FOUND, Body::Empty))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user