opentelemetry_psycopg_v3_instrumentation_quickstart.py

1import psycopg
2from opentelemetry.instrumentation.psycopg import PsycopgInstrumentor
3
4# The instrumentor can be used to instrument the psycopg library 
5# so that all connections created will be instrumented.
6PsycopgInstrumentor().instrument()
7
8# Now, any connection created with psycopg will be automatically instrumented
9conn = psycopg.connect(dbname="test", user="postgres", password="password", host="localhost", port="5432")
10cursor = conn.cursor()
11cursor.execute("SELECT 1;")
12result = cursor.fetchone()
13cursor.close()
14conn.close()
15
16# Alternatively, you can instrument a specific connection
17# conn = psycopg.connect(dbname="test", user="postgres", password="password", host="localhost", port="5432")
18# PsycopgInstrumentor().instrument_connection(conn)