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 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82
use super::DEFAULT_MESSAGE_LEVEL;
use http::Request;
use tracing::Level;
use tracing::Span;
/// Trait used to tell [`Trace`] what to do when a request is received.
///
/// See the [module docs](../trace/index.html#on_request) for details on exactly when the
/// `on_request` callback is called.
///
/// [`Trace`]: super::Trace
pub trait OnRequest<B> {
/// Do the thing.
///
/// `span` is the `tracing` [`Span`], corresponding to this request, produced by the closure
/// passed to [`TraceLayer::make_span_with`]. It can be used to [record field values][record]
/// that weren't known when the span was created.
///
/// [`Span`]: https://docs.rs/tracing/latest/tracing/span/index.html
/// [record]: https://docs.rs/tracing/latest/tracing/span/struct.Span.html#method.record
/// [`TraceLayer::make_span_with`]: crate::trace::TraceLayer::make_span_with
fn on_request(&mut self, request: &Request<B>, span: &Span);
}
impl<B> OnRequest<B> for () {
#[inline]
fn on_request(&mut self, _: &Request<B>, _: &Span) {}
}
impl<B, F> OnRequest<B> for F
where
F: FnMut(&Request<B>, &Span),
{
fn on_request(&mut self, request: &Request<B>, span: &Span) {
self(request, span)
}
}
/// The default [`OnRequest`] implementation used by [`Trace`].
///
/// [`Trace`]: super::Trace
#[derive(Clone, Debug)]
pub struct DefaultOnRequest {
level: Level,
}
impl Default for DefaultOnRequest {
fn default() -> Self {
Self {
level: DEFAULT_MESSAGE_LEVEL,
}
}
}
impl DefaultOnRequest {
/// Create a new `DefaultOnRequest`.
pub fn new() -> Self {
Self::default()
}
/// Set the [`Level`] used for [tracing events].
///
/// Please note that while this will set the level for the tracing events
/// themselves, it might cause them to lack expected information, like
/// request method or path. You can address this using
/// [`DefaultMakeSpan::level`].
///
/// Defaults to [`Level::DEBUG`].
///
/// [tracing events]: https://docs.rs/tracing/latest/tracing/#events
/// [`DefaultMakeSpan::level`]: crate::trace::DefaultMakeSpan::level
pub fn level(mut self, level: Level) -> Self {
self.level = level;
self
}
}
impl<B> OnRequest<B> for DefaultOnRequest {
fn on_request(&mut self, _: &Request<B>, _: &Span) {
event_dynamic_lvl!(self.level, "started processing request");
}
}