1 files changed, 10 insertions, 2 deletions
diff --git a/synapse/storage/database.py b/synapse/storage/database.py
index 1c1d76520f..f713efbf90 100644
--- a/synapse/storage/database.py
+++ b/synapse/storage/database.py
@@ -107,10 +107,18 @@ def make_pool(
with LoggingContext("db.on_new_connection"):
# HACK Patch the connection's commit function so that we can see
# how long it's taking from Jaeger.
- setattr(conn, "commit", trace(conn.commit, "db.conn.commit"))
+ class NastyConnectionWrapper:
+ def __init__(self, connection):
+ self._connection = connection
+ self.commit = trace(conn.commit, "db.conn.commit")
+
+ def __getattr__(self, item):
+ return self._connection.__getattr__(item)
engine.on_new_connection(
- LoggingDatabaseConnection(conn, engine, "on_new_connection")
+ LoggingDatabaseConnection(
+ NastyConnectionWrapper(conn), engine, "on_new_connection"
+ )
)
connection_pool = adbapi.ConnectionPool(
|