aws_opentelemetry_distro_xray_tracing_quickstart_with_otlp.py

1from opentelemetry import trace
2from opentelemetry.sdk.trace import TracerProvider
3from opentelemetry.sdk.trace.export import BatchSpanProcessor
4from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
5from aws_otel_distro import AwsSpanMetricsProcessor
6from opentelemetry.sdk.resources import SERVICE_NAME, Resource
7from opentelemetry.sdk.extension.aws.trace import AwsXRayIdGenerator
8
9# 1. Setup Resource with Service Name
10resource = Resource.create(attributes={
11    SERVICE_NAME: "my-python-service"
12})
13
14# 2. Initialize TracerProvider with AWS X-Ray ID Generator
15provider = TracerProvider(
16    resource=resource,
17    id_generator=AwsXRayIdGenerator()
18)
19
20# 3. Setup OTLP Exporter (pointing to the ADOT Collector)
21otlp_exporter = OTLPSpanExporter(endpoint="localhost:4317", insecure=True)
22
23# 4. Add the AWS Span Metrics Processor (specific to ADOT for CloudWatch Metrics)
24# and the standard BatchSpanProcessor
25provider.add_span_processor(AwsSpanMetricsProcessor(otlp_exporter, resource))
26provider.add_span_processor(BatchSpanProcessor(otlp_exporter))
27
28# 5. Set the global Tracer Provider
29trace.set_tracer_provider(provider)
30
31# 6. Instrument your code
32tracer = trace.get_tracer(__name__)
33
34with tracer.start_as_current_span("parent-span"):
35    with tracer.start_as_current_span("child-span"):
36        print("Hello from AWS Distro for OpenTelemetry!")