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
|
#[cfg(feature = "opentelemetry")]
pub mod telemetry;
#[cfg(feature = "opentelemetry")]
pub use opentelemetry_sdk::trace::SdkTracerProvider;
#[cfg(feature = "tracing-loki")]
mod loki;
use tracing_subscriber::{
EnvFilter, Layer, Registry, layer::SubscriberExt, util::SubscriberInitExt,
};
/// Telemetry handle
#[derive(bon::Builder)]
#[builder(finish_fn(vis = "", name = build_internal))]
pub struct Tracing {
#[builder(field = vec![tracing_subscriber::fmt::layer().boxed()])]
layers: Vec<Box<dyn Layer<Registry> + Sync + Send>>,
#[cfg(feature = "tracing-loki")]
#[builder(setters(vis = "", name = loki_internal))]
pub loki_task: tracing_loki::BackgroundTask,
#[cfg(feature = "opentelemetry")]
#[builder(setters(vis = "", name = otel_internal))]
pub otel_provider: opentelemetry_sdk::trace::SdkTracerProvider,
}
// Define a custom finishing function as a method on the `UserBuilder`.
// The builder's state must implement the `IsComplete` trait.
// See details about it in the tip below this example.
impl<S: tracing_builder::IsComplete> TracingBuilder<S> {
pub fn build(self, config: &crate::Monitoring) -> Tracing {
// Delegate to `build_internal()` to get the instance of user.
let mut tracing = self.build_internal();
let layers = std::mem::take(&mut tracing.layers);
tracing_subscriber::registry()
.with(layers)
.with(
EnvFilter::try_from_default_env()
.unwrap_or_else(|_| config.log_level.to_string().into()),
)
.try_init()
.ok();
tracing
}
}
// #[cfg(test)]
// mod tests {
// use super::*;
//
// #[test]
// fn build() {
// let builder = Tracing::builder().build();
// let level = crate::Monitoring {
// log_level: "info".to_string(),
// #[cfg(feature = "opentelemetry")]
// opentelemetry_endpoint: "http://localhost:4317".into(),
// #[cfg(feature = "tracing-loki")]
// loki_endpoint: "http://localhost:3100".into(),
// };
// builder.init(&level);
// builder.loki_task
// }
// }
|
