opentelemetry_sqlalchemy_auto_instrumentation_for_database_query_tracing.py

1from sqlalchemy import create_engine, text
2from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
3
4# 1. Initialize the instrumentor
5# This will automatically instrument all engines created hereafter
6SQLAlchemyInstrumentor().instrument()
7
8# 2. Create a SQLAlchemy engine as usual
9engine = create_engine("sqlite:///:memory:")
10
11# 3. Use the engine. Spans will be generated automatically for these operations.
12with engine.connect() as conn:
13    conn.execute(text("SELECT 1"))
14
15# To instrument a specific engine instance instead of all engines:
16# SQLAlchemyInstrumentor().instrument(engine=engine)