pi_logger 0.8.0

pi_logger
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

use crate::observability::config::MetricsConfig;
use crate::observability::remote::append_otlp_path;
use anyhow::{bail, Context as AnyhowContext, Result};
use opentelemetry::KeyValue;
use opentelemetry_otlp::{MetricExporter, Protocol, WithExportConfig};
use opentelemetry_sdk::{metrics::SdkMeterProvider, metrics::Temporality, Resource};

/// Builds the OpenTelemetry metrics provider.
pub fn build_metrics_provider(config: &MetricsConfig) -> Result<SdkMeterProvider> {
    if config.exporter != "otlp" {
        bail!("unsupported metrics exporter `{}`", config.exporter);
    }

    let endpoint = append_otlp_path(&config.endpoint, "/v1/metrics");
    let exporter = MetricExporter::builder()
        .with_http()
        .with_protocol(Protocol::HttpBinary)
        .with_endpoint(endpoint)
        .with_temporality(Temporality::LowMemory)
        .build()
        .context("failed to create OpenTelemetry metrics exporter")?;

    let provider = SdkMeterProvider::builder()
        .with_periodic_exporter(exporter)
        .with_resource(resource(config))
        .build();

    Ok(provider)
}

fn resource(config: &MetricsConfig) -> Resource {
    let mut builder = Resource::builder().with_service_name(config.service_name.clone());
    if let Some(version) = &config.service_version {
        builder = builder.with_attribute(KeyValue::new("service.version", version.clone()));
    }
    builder.build()
}