diff --git a/apps/profile/middleware.py b/apps/profile/middleware.py index f9f6a8481..5b583ce9c 100644 --- a/apps/profile/middleware.py +++ b/apps/profile/middleware.py @@ -106,6 +106,8 @@ class SQLLogToConsoleMiddleware: if connection.queries: time_elapsed = sum([float(q['time']) for q in connection.queries]) queries = connection.queries + if getattr(connection, 'queriesx', False): + queries.extend(connection.queriesx) for query in queries: if query.get('mongo'): query['sql'] = "~FM%s: %s" % (query['mongo']['collection'], query['mongo']['query']) diff --git a/utils/mongo_raw_log_middleware.py b/utils/mongo_raw_log_middleware.py index 7c998cb3f..fdc37692a 100644 --- a/utils/mongo_raw_log_middleware.py +++ b/utils/mongo_raw_log_middleware.py @@ -63,7 +63,9 @@ class MongoDumpMiddleware(object): result = original_method(*args, **kwargs) stop = time() duration = stop - start - connection.queries.append({ + if not getattr(connection, 'queriesx', False): + connection.queriesx = [] + connection.queriesx.append({ 'mongo': message, 'time': '%.3f' % duration, }) diff --git a/utils/redis_raw_log_middleware.py b/utils/redis_raw_log_middleware.py index a7d4e159d..7854a5c27 100644 --- a/utils/redis_raw_log_middleware.py +++ b/utils/redis_raw_log_middleware.py @@ -39,7 +39,9 @@ class RedisDumpMiddleware(object): result = original_method(*args, **kwargs) stop = time() duration = stop - start - connection.queries.append({ + if not getattr(connection, 'queriesx', False): + connection.queriesx = [] + connection.queriesx.append({ 'redis': message, 'time': '%.3f' % duration, })