Update handler name used by SQLAlchemy

This commit is contained in:
Brendan Reilly
2021-08-09 14:49:34 -04:00
parent d115c86ec6
commit 1883fa8567

View File

@@ -15,6 +15,7 @@ from prometheus_client import ( # noqa: F401
Histogram,
start_http_server,
)
import sqlalchemy
from sqlalchemy import event
@@ -85,9 +86,11 @@ def db_hook_event_listeners(target=None):
if not target:
target = db.engine
@event.listens_for(target, "dbapi_error", named=True)
def receive_dbapi_error(**kw):
db_dbapi_error_counter.inc()
if sqlalchemy.__version__[0] == "0":
# Deprecated since 0.9
@event.listens_for(target, "dbapi_error", named=True)
def receive_dbapi_error(**kw):
db_dbapi_error_counter.inc()
@event.listens_for(target, "engine_connect")
def receive_engine_connect(conn, branch):