opentelemetry_psycopg2_instrumentation_postgresql_query_tracing.py

1import psycopg2
2from opentelemetry.instrumentation.psycopg2 import Psycopg2Instrumentor
3
4# Optional: To see the spans in the console, set up a simple processor
5from opentelemetry import trace
6from opentelemetry.sdk.trace import TracerProvider
7from opentelemetry.sdk.trace.export import (
8    BatchSpanProcessor,
9    ConsoleSpanExporter,
10)
11
12trace.set_tracer_provider(TracerProvider())
13trace.get_tracer_provider().add_span_processor(
14    BatchSpanProcessor(ConsoleSpanExporter())
15)
16
17# Instrument psycopg2
18Psycopg2Instrumentor().instrument()
19
20# Now use psycopg2 as usual
21cnx = psycopg2.connect(database='DATABASE', user='USER', password='PASSWORD', host='HOST', port='PORT')
22cursor = cnx.cursor()
23cursor.execute("SELECT 1;")
24result = cursor.fetchone()
25cursor.close()
26cnx.close()