that was a bad idea
This commit is contained in:
parent
804292379f
commit
4d837da7a7
|
@ -60,7 +60,7 @@ fn main() -> Result<()> {
|
||||||
.attach(cors)
|
.attach(cors)
|
||||||
.attach(MainDatabase::fairing())
|
.attach(MainDatabase::fairing())
|
||||||
.attach(SpaceHelmet::default())
|
.attach(SpaceHelmet::default())
|
||||||
.attach(TraceRequest {})
|
.attach(RequestId {})
|
||||||
.attach(paseto::ed25519_keypair())
|
.attach(paseto::ed25519_keypair())
|
||||||
.attach(DiscordWebhook::fairing())
|
.attach(DiscordWebhook::fairing())
|
||||||
.attach(Mastodon::fairing())
|
.attach(Mastodon::fairing())
|
||||||
|
|
|
@ -1,91 +1,34 @@
|
||||||
use rocket::fairing::{Fairing, Info, Kind};
|
use rocket::fairing::{Fairing, Info, Kind};
|
||||||
use rocket::http::{ContentType, Header, Method, Status};
|
use rocket::http::Header;
|
||||||
use rocket::{Data, Request, Response};
|
use rocket::{Data, Request, Response};
|
||||||
use rusty_ulid::generate_ulid_string;
|
use rusty_ulid::generate_ulid_string;
|
||||||
use std::mem;
|
|
||||||
use tracing::{span, span::Entered, Level, Span};
|
|
||||||
|
|
||||||
struct SpanWrapper<'a> {
|
pub struct RequestId;
|
||||||
span: Box<Span>,
|
|
||||||
entered: Entered<'a>,
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct TraceRequest;
|
impl Fairing for RequestId {
|
||||||
|
|
||||||
impl Fairing for TraceRequest {
|
|
||||||
fn info(&self) -> Info {
|
fn info(&self) -> Info {
|
||||||
Info {
|
Info {
|
||||||
name: "Tracing spans per request",
|
name: "Request ID",
|
||||||
kind: Kind::Request | Kind::Response,
|
kind: Kind::Request | Kind::Response,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn on_request(&self, request: &mut Request, _: &Data) {
|
fn on_request(&self, request: &mut Request, _: &Data) {
|
||||||
let reqid: String = match request.headers().get_one("X-Request-Id") {
|
match request.headers().get_one("X-Request-Id") {
|
||||||
Some(reqid) => reqid.to_string(),
|
Some(_) => {}
|
||||||
None => {
|
None => {
|
||||||
let reqid = generate_ulid_string();
|
let reqid = generate_ulid_string();
|
||||||
request.add_header(Header::new("X-Request-Id", reqid.clone()));
|
request.add_header(Header::new("X-Request-Id", reqid.clone()));
|
||||||
reqid
|
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
request.local_cache(|| {
|
|
||||||
let span = span!(
|
|
||||||
Level::INFO,
|
|
||||||
"request",
|
|
||||||
method = &request.method().to_string()[..],
|
|
||||||
uri = &request.uri().to_string()[..],
|
|
||||||
addr = &request
|
|
||||||
.client_ip()
|
|
||||||
.map(|ip| ip.to_string())
|
|
||||||
.unwrap_or("unknown".to_string())[..],
|
|
||||||
request_id = &reqid[..],
|
|
||||||
);
|
|
||||||
let span = Box::new(span);
|
|
||||||
let entered = span.enter();
|
|
||||||
let entered_lt = unsafe { mem::transmute::<_, Entered<'static>>(entered) };
|
|
||||||
|
|
||||||
Box::new(SpanWrapper {
|
|
||||||
entered: entered_lt,
|
|
||||||
span: span,
|
|
||||||
})
|
|
||||||
});
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn on_response(&self, request: &Request, response: &mut Response) {
|
fn on_response(&self, request: &Request, response: &mut Response) {
|
||||||
let span: &Box<SpanWrapper> = request.local_cache(|| {
|
|
||||||
Box::new({
|
|
||||||
let span = error_span!(
|
|
||||||
"somehow the span wasn't put into the request, neat",
|
|
||||||
method = &request.method().to_string()[..],
|
|
||||||
uri = &request.uri().to_string()[..],
|
|
||||||
address = &request
|
|
||||||
.real_ip()
|
|
||||||
.map(|ip| ip.to_string())
|
|
||||||
.unwrap_or("unknown".to_string())[..],
|
|
||||||
);
|
|
||||||
|
|
||||||
let span = Box::new(span);
|
|
||||||
let entered = span.enter();
|
|
||||||
let entered_lt = unsafe { mem::transmute::<_, Entered<'static>>(entered) };
|
|
||||||
|
|
||||||
SpanWrapper {
|
|
||||||
entered: entered_lt,
|
|
||||||
span: span,
|
|
||||||
}
|
|
||||||
})
|
|
||||||
});
|
|
||||||
match request.headers().get_one("X-Request-Id") {
|
match request.headers().get_one("X-Request-Id") {
|
||||||
Some(reqid) => {
|
Some(reqid) => {
|
||||||
response.set_header(Header::new("X-Request-Id", format!("{}", reqid)));
|
response.set_header(Header::new("X-Request-Id", format!("{}", reqid)));
|
||||||
}
|
}
|
||||||
None => {}
|
None => {}
|
||||||
};
|
};
|
||||||
info!(
|
|
||||||
status_code = &response.status().code,
|
|
||||||
status_reason = response.status().reason
|
|
||||||
);
|
|
||||||
drop(span);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue