opentelemetry-instrumentation-cassandra

from opentelemetry import trace from opentelemetry.sdk.trace import TracerProvider from opentelemetry.sdk.trace.export import ConsoleSpanExporter, SimpleSpanProcessor from opentelemetry.instrumentation.cassandra import CassandraInstrumentor from cassandra.cluster import Cluster, ConsistencyLevel import os # 1. Configure OpenTelemetry SDK # For production, use an OTLPSpanExporter or other suitable exporter provider = TracerProvider() processor = SimpleSpanProcessor(ConsoleSpanExporter()) provider.add_span_processor(processor) trace.set_tracer_provider(provider) # 2. Instrument the Cassandra driver CassandraInstrumentor().instrument() # 3. Use the Cassandra driver as usual # Replace with your Cassandra connection details cassandra_contact_points = os.environ.get('CASSANDRA_CONTACT_POINTS', '127.0.0.1').split(',') cluster = Cluster(cassandra_contact_points) session = cluster.connect() try: print("Executing Cassandra operations...") session.execute("CREATE KEYSPACE IF NOT EXISTS mykeyspace WITH REPLICATION = { 'class' : 'SimpleStrategy', 'replication_factor' : 1 }") session.execute("USE mykeyspace") session.execute("CREATE TABLE IF NOT EXISTS users (id uuid PRIMARY KEY, name text, age int)") # Insert data session.execute("INSERT INTO users (id, name, age) VALUES (uuid(), 'Jane Doe', 25)", consistency_level=ConsistencyLevel.QUORUM) print("Inserted 'Jane Doe'.") # Select data rows = session.execute("SELECT * FROM users WHERE name = 'Jane Doe'") for row in rows: print(f"Retrieved: {row.name}, {row.age}") # Update data session.execute("UPDATE users SET age = 26 WHERE name = 'Jane Doe'") print("Updated 'Jane Doe's age.") # Select all data again to see update rows = session.execute("SELECT * FROM users ALLOW FILTERING") print("All users:") for row in rows: print(f" {row.name}, {row.age}") finally: session.shutdown() cluster.shutdown() print("Cassandra session and cluster shut down.")