mirror of
https://github.com/actix/actix-web.git
synced 2024-12-21 23:56:35 +00:00
add fn_guard
This commit is contained in:
parent
457b75c995
commit
6fcbe4bcda
1 changed files with 26 additions and 7 deletions
33
src/guard.rs
33
src/guard.rs
|
@ -39,15 +39,35 @@ pub trait Guard {
|
|||
fn check(&self, request: &RequestHead) -> bool;
|
||||
}
|
||||
|
||||
#[doc(hidden)]
|
||||
pub struct FnGuard<F: Fn(&RequestHead) -> bool + 'static>(F);
|
||||
|
||||
impl<F> Guard for F
|
||||
/// Return guard that matches if all of the supplied guards.
|
||||
///
|
||||
/// ```rust
|
||||
/// use actix_web::{guard, web, App, HttpResponse};
|
||||
///
|
||||
/// fn main() {
|
||||
/// App::new().service(web::resource("/index.html").route(
|
||||
/// web::route()
|
||||
/// .guard(
|
||||
/// guard::fn_guard(|req| req.headers().contains_key("content-type")))
|
||||
/// .to(|| HttpResponse::MethodNotAllowed()))
|
||||
/// );
|
||||
/// }
|
||||
/// ```
|
||||
pub fn fn_guard<F>(f: F) -> impl Guard
|
||||
where
|
||||
F: Fn(&RequestHead) -> bool + 'static,
|
||||
F: Fn(&RequestHead) -> bool,
|
||||
{
|
||||
FnGuard(f)
|
||||
}
|
||||
|
||||
struct FnGuard<F: Fn(&RequestHead) -> bool>(F);
|
||||
|
||||
impl<F> Guard for FnGuard<F>
|
||||
where
|
||||
F: Fn(&RequestHead) -> bool,
|
||||
{
|
||||
fn check(&self, head: &RequestHead) -> bool {
|
||||
(*self)(head)
|
||||
(self.0)(head)
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -93,7 +113,6 @@ impl Guard for AnyGuard {
|
|||
/// Return guard that matches if all of the supplied guards.
|
||||
///
|
||||
/// ```rust
|
||||
/// # extern crate actix_web;
|
||||
/// use actix_web::{guard, web, App, HttpResponse};
|
||||
///
|
||||
/// fn main() {
|
||||
|
|
Loading…
Reference in a new issue