def patch_conn(conn, traced_conn_cls=dbapi.TracedConnection):
""" Wrap will patch the instance so that it's queries are traced."""
# ensure we've patched extensions (this is idempotent) in
# case we're only tracing some connections.
_patch_extensions(_psycopg2_extensions)
c = traced_conn_cls(conn)
# fetch tags from the dsn
dsn = sql.parse_pg_dsn(conn.dsn)
tags = {
net.TARGET_HOST: dsn.get("host"),
net.TARGET_PORT: dsn.get("port"),
db.NAME: dsn.get("dbname"),
db.USER: dsn.get("user"),
"db.application" : dsn.get("application_name"),
}
Pin(
service="postgres",
app="postgres",
app_type="db",
tags=tags).onto(c)
return c
def patch_conn(conn, traced_conn_cls=Psycopg2TracedConnection):
""" Wrap will patch the instance so that it's queries are traced."""
# ensure we've patched extensions (this is idempotent) in
# case we're only tracing some connections.
_patch_extensions(_psycopg2_extensions)
c = traced_conn_cls(conn)
# fetch tags from the dsn
dsn = sql.parse_pg_dsn(conn.dsn)
tags = {
net.TARGET_HOST: dsn.get("host"),
net.TARGET_PORT: dsn.get("port"),
db.NAME: dsn.get("dbname"),
db.USER: dsn.get("user"),
"db.application": dsn.get("application_name"),
}
Pin(
service="postgres",
app="postgres",
app_type="db",
tags=tags).onto(c)
return c
def _set_tags_from_cursor(span, vendor, cursor):
""" attempt to set db connection tags by introspecting the cursor. """
if 'postgres' == vendor:
if hasattr(cursor, 'connection') and hasattr(cursor.connection, 'dsn'):
dsn = getattr(cursor.connection, 'dsn', None)
if dsn:
d = sqlx.parse_pg_dsn(dsn)
span.set_tag(sqlx.DB, d.get("dbname"))
span.set_tag(netx.TARGET_HOST, d.get("host"))
span.set_tag(netx.TARGET_PORT, d.get("port"))
def patch_conn(conn, traced_conn_cls=Psycopg2TracedConnection):
""" Wrap will patch the instance so that its queries are traced."""
# ensure we've patched extensions (this is idempotent) in
# case we're only tracing some connections.
_patch_extensions(_psycopg2_extensions)
c = traced_conn_cls(conn)
# fetch tags from the dsn
dsn = sql.parse_pg_dsn(conn.dsn)
tags = {
net.TARGET_HOST: dsn.get('host'),
net.TARGET_PORT: dsn.get('port'),
db.NAME: dsn.get('dbname'),
db.USER: dsn.get('user'),
'db.application': dsn.get('application_name'),
}
Pin(app='postgres', tags=tags).onto(c)
return c