35 lines
976 B
Rust
35 lines
976 B
Rust
use rocket::fairing::{Fairing, Info, Kind};
|
|
use rocket::http::Header;
|
|
use rocket::{Data, Request, Response};
|
|
use rusty_ulid::generate_ulid_string;
|
|
|
|
pub struct RequestId;
|
|
|
|
impl Fairing for RequestId {
|
|
fn info(&self) -> Info {
|
|
Info {
|
|
name: "Request ID",
|
|
kind: Kind::Request | Kind::Response,
|
|
}
|
|
}
|
|
|
|
fn on_request(&self, request: &mut Request, _: &Data) {
|
|
match request.headers().get_one("X-Request-Id") {
|
|
Some(_) => {}
|
|
None => {
|
|
let reqid = generate_ulid_string();
|
|
request.add_header(Header::new("X-Request-Id", reqid.clone()));
|
|
}
|
|
};
|
|
}
|
|
|
|
fn on_response(&self, request: &Request, response: &mut Response) {
|
|
match request.headers().get_one("X-Request-Id") {
|
|
Some(reqid) => {
|
|
response.set_header(Header::new("X-Request-Id", format!("{}", reqid)));
|
|
}
|
|
None => {}
|
|
};
|
|
}
|
|
}
|