mirror of
https://github.com/samuelclay/NewsBlur.git
synced 2025-04-13 09:42:01 +00:00
78 lines
2.9 KiB
Python
78 lines
2.9 KiB
Python
from django.core.exceptions import MiddlewareNotUsed
|
|
from django.conf import settings
|
|
from django.db import connection
|
|
from redis.connection import Connection
|
|
from time import time
|
|
|
|
class RedisDumpMiddleware(object):
|
|
|
|
def __init__(self, get_response=None):
|
|
self.get_response = get_response
|
|
|
|
def activated(self, request):
|
|
return (settings.DEBUG_QUERIES or
|
|
(hasattr(request, 'activated_segments') and
|
|
'db_profiler' in request.activated_segments))
|
|
|
|
def process_view(self, request, callback, callback_args, callback_kwargs):
|
|
if not self.activated(request): return
|
|
if not getattr(Connection, '_logging', False):
|
|
# save old methods
|
|
setattr(Connection, '_logging', True)
|
|
Connection.pack_command = \
|
|
self._instrument(Connection.pack_command)
|
|
|
|
def process_celery(self, profiler):
|
|
if not self.activated(profiler): return
|
|
if not getattr(Connection, '_logging', False):
|
|
# save old methods
|
|
setattr(Connection, '_logging', True)
|
|
Connection.pack_command = \
|
|
self._instrument(Connection.pack_command)
|
|
|
|
def process_response(self, request, response):
|
|
# if settings.DEBUG and hasattr(self, 'orig_pack_command'):
|
|
# # remove instrumentation from redis
|
|
# setattr(Connection, '_logging', False)
|
|
# Connection.pack_command = \
|
|
# self.orig_pack_command
|
|
return response
|
|
|
|
def _instrument(self, original_method):
|
|
def instrumented_method(*args, **kwargs):
|
|
message = self.process_message(*args, **kwargs)
|
|
if not message:
|
|
return original_method(*args, **kwargs)
|
|
start = time()
|
|
result = original_method(*args, **kwargs)
|
|
stop = time()
|
|
duration = stop - start
|
|
if not getattr(connection, 'queriesx', False):
|
|
connection.queriesx = []
|
|
connection.queriesx.append({
|
|
'redis': message,
|
|
'time': '%.3f' % duration,
|
|
})
|
|
return result
|
|
return instrumented_method
|
|
|
|
def process_message(self, *args, **kwargs):
|
|
query = []
|
|
for a, arg in enumerate(args):
|
|
if isinstance(arg, Connection):
|
|
continue
|
|
if len(str(arg)) > 100:
|
|
arg = "[%s bytes]" % len(str(arg))
|
|
query.append(str(arg).replace('\n', ''))
|
|
return { 'query': ' '.join(query) }
|
|
|
|
def __call__(self, request):
|
|
response = None
|
|
if hasattr(self, 'process_request'):
|
|
response = self.process_request(request)
|
|
if not response:
|
|
response = self.get_response(request)
|
|
if hasattr(self, 'process_response'):
|
|
response = self.process_response(request, response)
|
|
|
|
return response
|