opentelemetry_cassandra_instrumentation_quickstart_with_console_tracing.py

1from cassandra.cluster import Cluster
2from opentelemetry.instrumentation.cassandra import CassandraInstrumentor
3from opentelemetry import trace
4from opentelemetry.sdk.trace import TracerProvider
5from opentelemetry.sdk.trace.export import (
6    BatchSpanProcessor,
7    ConsoleSpanExporter,
8)
9
10# Sets up the OpenTelemetry SDK to output traces to the console
11trace.set_tracer_provider(TracerProvider())
12trace.get_tracer_provider().add_span_processor(
13    BatchSpanProcessor(ConsoleSpanExporter())
14)
15
16# Instrument the Cassandra driver
17CassandraInstrumentor().instrument()
18
19# Standard Cassandra driver usage
20cluster = Cluster(['127.0.0.1'])
21session = cluster.connect()
22
23# This query will be automatically traced
24session.execute("SELECT release_version FROM system.local")
25
26cluster.shutdown()