rocket_response_timer 0.1.1

Rocket Response Time Fairing. Rocket min v. - 0.5.0-rc.1
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use rocket::{fairing::Fairing, Request};

pub struct ResponseTimer {}

impl ResponseTimer {
    pub fn new() -> Self {
        Self {}
    }
}

#[rocket::async_trait]
impl Fairing for ResponseTimer {
    fn info(&self) -> rocket::fairing::Info {
        rocket::fairing::Info {
            name: "ResponseTimer",
            kind: rocket::fairing::Kind::Request | rocket::fairing::Kind::Response,
        }
    }

    async fn on_request(&self, request: &mut Request<'_>, _: &mut rocket::Data<'_>) {
        request.local_cache(|| std::time::Instant::now());
    }

    async fn on_response<'r>(&self, request: &'r Request<'_>, response: &mut rocket::Response<'r>) {
        let start_time = request.local_cache(|| std::time::Instant::now());
        let end_time = start_time.elapsed();

        response.set_raw_header("X-Response-Time", format!("{:.2?}", end_time));
    }
}