Skip to content

Commit bf1c211

Browse files
Add more wrap_connect info
1 parent eaa05bc commit bf1c211

File tree

1 file changed

+11
-4
lines changed
  • instrumentation/opentelemetry-instrumentation-dbapi/src/opentelemetry/instrumentation/dbapi

1 file changed

+11
-4
lines changed

instrumentation/opentelemetry-instrumentation-dbapi/src/opentelemetry/instrumentation/dbapi/__init__.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,14 +37,20 @@
3737
import mysql.connector
3838
import pyodbc
3939
40-
from opentelemetry.instrumentation.dbapi import trace_integration
41-
40+
from opentelemetry.instrumentation.dbapi import (
41+
trace_integration,
42+
wrap_connect,
43+
)
4244
4345
# Ex: mysql.connector
4446
trace_integration(mysql.connector, "connect", "mysql")
4547
# Ex: pyodbc
4648
trace_integration(pyodbc, "Connection", "odbc")
4749
50+
# Or, directly call wrap_connect for more configurability
51+
wrap_connect(__name__, mysql.connector, "connect", "mysql")
52+
wrap_connect(__name__, pyodbc, "Connection", "odbc")
53+
4854
4955
Configuration
5056
-------------
@@ -65,11 +71,12 @@
6571
import mysql.connector
6672
import pyodbc
6773
68-
from opentelemetry.instrumentation.dbapi import trace_integration
74+
from opentelemetry.instrumentation.dbapi import wrap_connect
6975
7076
7177
# Ex: mysql.connector
72-
trace_integration(
78+
wrap_connect(
79+
__name__,
7380
mysql.connector,
7481
"connect",
7582
"mysql",

0 commit comments

Comments
 (0)