diff options
Diffstat (limited to 'synapse/storage/database.py')
| -rw-r--r-- | synapse/storage/database.py | 12 |
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( |