Function main

examples/http-axum-streaming-otel/src/main.rs:78–106

()

#[tokio::main]
async fn main() -> Result<(), Error> {
    // Set up OpenTelemetry tracer provider that writes spans to stdout for
    // debugging purposes
    let exporter = opentelemetry_stdout::SpanExporter::default();
    let tracer_provider = trace::SdkTracerProvider::builder()
        .with_batch_exporter(exporter)
        .build();

    // Set up link between OpenTelemetry and tracing crate
    tracing_subscriber::registry()
        .with(tracing_opentelemetry::OpenTelemetryLayer::new(
            tracer_provider.tracer("my-streaming-app"),
        ))
        .init();

    let svc = Router::new().route("/", get(stream_words));

    // Initialize the Lambda runtime and add OpenTelemetry tracing
    let runtime = Runtime::new(StreamAdapter::from(svc)).layer(
        OtelLayer::new(|| {
            if let Err(err) = tracer_provider.force_flush() {
                eprintln!("Error flushing traces: {err:#?}");
            }
        })
        .with_trigger(OpenTelemetryFaasTrigger::Http),
    );

    runtime.run().await
}