otel-bootstrap 0.4.0

One-call OpenTelemetry bootstrap — traces + metrics with OTLP export
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

// SPDX-License-Identifier: MIT
//! Example axum server wiring the `OrgContextSpanEnricher` layer.
//!
//! Demonstrates the tower layer ordering required by ADR platform/0015:
//! the OrgContextSpanEnricher sits *inside* the [`axum::Extension`] layer that
//! injects `OrganizationContext` and *outside* the handler. The outer
//! [`otel_bootstrap::axum_layer`] opens an OTel server span; the enricher
//! records the four `enduser.*` attributes on the active tracing span for the
//! current request.
//!
//! Run with:
//! ```text
//! OTEL_EXPORTER_OTLP_ENDPOINT=http://localhost:4317 \
//!   cargo run --example axum_org_context --features "grpc,axum,org-context"
//! ```

use api_bones::{OrgId, OrganizationContext, Principal, RequestId};
use axum::{Extension, Router, routing::get};
use std::error::Error;
use uuid::Uuid;

async fn hello() -> &'static str {
    tracing::info!("handling request");
    "hello"
}

#[tokio::main(flavor = "current_thread")]
async fn main() -> Result<(), Box<dyn Error>> {
    let handles = otel_bootstrap::init_telemetry("axum-org-context-example")?;

    let ctx = OrganizationContext::new(
        OrgId::generate(),
        Principal::human(Uuid::new_v4()),
        RequestId::new(),
    );

    let _app: Router = Router::new()
        .route("/", get(hello))
        .layer(otel_bootstrap::org_context_span_enricher_layer())
        .layer(Extension(ctx))
        .layer(otel_bootstrap::axum_layer());

    handles.shutdown()?;
    Ok(())
}