bbox-core 0.6.1

Common functionality for BBOX services
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

use crate::config::MetricsCfg;
use once_cell::sync::OnceCell;
use opentelemetry::{
    global,
    sdk::{
        export::metrics::aggregation,
        metrics::{controllers, processors, selectors},
        propagation::TraceContextPropagator,
    },
};
use opentelemetry_prometheus::PrometheusExporter;

fn init_tracer(config: &MetricsCfg) {
    if let Some(cfg) = &config.jaeger {
        global::set_text_map_propagator(TraceContextPropagator::new()); // default header: traceparent
        opentelemetry_jaeger::new_agent_pipeline()
            .with_endpoint(cfg.agent_endpoint.clone())
            .with_service_name("bbox")
            .install_batch(opentelemetry::runtime::Tokio)
            .expect("Failed to initialize tracer");
    }
}

pub(crate) fn init_metrics_exporter() -> Option<PrometheusExporter> {
    let metrics_cfg = MetricsCfg::from_config()?;

    init_tracer(&metrics_cfg);

    metrics_cfg.prometheus.as_ref()?;

    // Prometheus request metrics handler
    let controller = controllers::basic(
        processors::factory(
            selectors::simple::histogram([1.0, 2.0, 5.0, 10.0, 20.0, 50.0]),
            aggregation::cumulative_temporality_selector(),
        )
        .with_memory(true),
    )
    .build();
    let exporter = opentelemetry_prometheus::exporter(controller).init();
    // let metrics_handler = PrometheusMetricsHandler::new(exporter);

    // Run actix server, metrics are now available at http://localhost:8080/metrics
    // HttpServer::new(move || {
    //     App::new()
    //         .wrap(RequestTracing::new())
    //         .wrap(request_metrics.clone())
    //         .route("/metrics", web::get().to(metrics_handler.clone()))
    // })

    Some(exporter)
}

#[derive(Default)]
pub struct NoMetrics;

pub fn no_metrics() -> &'static NoMetrics {
    static METRICS: OnceCell<NoMetrics> = OnceCell::new();
    METRICS.get_or_init(NoMetrics::default)
}