azure_sdk_opentelemetry_distributed_tracing_quickstart.py

1import os
2from opentelemetry import trace
3from opentelemetry.sdk.trace import TracerProvider
4from opentelemetry.sdk.trace.export import BatchSpanProcessor, ConsoleSpanExporter
5from azure.core.settings import settings
6from azure.core.tracing.ext.opentelemetry_span import OpenTelemetrySpan
7from azure.identity import DefaultAzureCredential
8from azure.storage.blob import BlobServiceClient
9
10# 1. Setup OpenTelemetry
11# Set the global tracer provider
12trace.set_tracer_provider(TracerProvider())
13tracer = trace.get_tracer(__name__)
14
15# Configure a span processor to output traces to the console
16span_processor = BatchSpanProcessor(ConsoleSpanExporter())
17trace.get_tracer_provider().add_span_processor(span_processor)
18
19# 2. Register OpenTelemetry with Azure Core
20# This tells Azure SDKs to use OpenTelemetry for tracing
21settings.tracing_implementation = OpenTelemetrySpan
22
23# 3. Use an Azure Client as usual
24# Traces will now be automatically captured for SDK operations
25endpoint = os.environ.get("AZURE_STORAGE_BLOB_ENDPOINT")
26credential = DefaultAzureCredential()
27
28with tracer.start_as_current_span("MyApplicationSpan"):
29    client = BlobServiceClient(endpoint, credential=credential)
30    
31    # This call will generate a child span automatically
32    properties = client.get_service_properties()
33    print(f"Account Kind: {properties['Logging']['Read']}")