hyper-trace-id 0.1.0

Hyper (axum, warp, poem, ...) middleware for adding trace ids to requests.
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
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

#[cfg(feature = "axum")]
mod extract;

mod layer;

pub use crate::layer::SetTraceIdLayer;
use std::fmt::{Display, Formatter};
use uuid::Uuid;

/// Make a type usable as trace id.
///
/// ```
/// use std::fmt::{Display, Formatter};
/// use hyper_trace_id::MakeTraceId;
/// use uuid::Uuid;
///
/// #[derive(Clone)]
/// struct MyTraceId {
///     id: String,
/// }
///
/// impl Display for MyTraceId {
///     fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
///         write!(f, "{}", self.id)
///     }
/// }
///
/// impl MakeTraceId for MyTraceId {
///     fn make_trace_id() -> Self {
///         Self {
///             id: Uuid::new_v4().to_string()
///         }
///     }
/// }
/// ```
pub trait MakeTraceId: Send + Sync + Display + Clone {
    fn make_trace_id() -> Self;
}

impl MakeTraceId for String {
    fn make_trace_id() -> Self {
        Uuid::new_v4().to_string()
    }
}

#[derive(Debug, Clone)]
pub struct TraceId<T>
where
    T: MakeTraceId,
{
    pub id: T,
}

impl<T> TraceId<T>
where
    T: MakeTraceId,
{
    pub(crate) fn new() -> Self {
        TraceId {
            id: T::make_trace_id(),
        }
    }
}

impl<T> Display for TraceId<T>
where
    T: MakeTraceId,
{
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.id)
    }
}