Added CacheRealTimeModule

Kronuz · Kronuz · commit 04582e2e6f95 · 2011-12-31T10:56:24.000-06:00
diff --git a/devserver/modules/cache.py b/devserver/modules/cache.py
@@ -7,6 +7,7 @@ class CacheSummaryModule(DevServerModule):
     """
     Outputs a summary of cache events once a response is ready.
     """
+    real_time = False
 
     logger_name = 'cache'
 
@@ -19,7 +20,7 @@ def process_init(self, request):
         self.old = dict((k, getattr(cache, k)) for k in self.attrs_to_track)
 
         for k in self.attrs_to_track:
-            setattr(cache, k, track(getattr(cache, k), 'cache'))
+            setattr(cache, k, track(getattr(cache, k), 'cache', self.logger if self.real_time else None))
 
     def process_complete(self, request):
         from devserver.utils.stats import stats
@@ -33,13 +34,18 @@ def process_complete(self, request):
         else:
             ratio = 100
 
-        self.logger.info('%(calls)s calls made with a %(ratio)d%% hit percentage (%(misses)s misses)' % dict(
-            calls=calls,
-            ratio=ratio,
-            hits=hits,
-            misses=misses,
-        ), duration=stats.get_total_time('cache'))
+        if not self.real_time:
+            self.logger.info('%(calls)s calls made with a %(ratio)d%% hit percentage (%(misses)s misses)' % dict(
+                calls=calls,
+                ratio=ratio,
+                hits=hits,
+                misses=misses,
+            ), duration=stats.get_total_time('cache'))
 
         # set our attributes back to their defaults
         for k, v in self.old.iteritems():
             setattr(cache, k, v)
+
+
+class CacheRealTimeModule(CacheSummaryModule):
+    real_time = True
diff --git a/devserver/utils/stats.py b/devserver/utils/stats.py
@@ -16,7 +16,7 @@ def __init__(self, *args, **kwargs):
         super(StatCollection, self).__init__(*args, **kwargs)
         self.reset()
 
-    def run(self, func, key, *args, **kwargs):
+    def run(self, func, key, logger, *args, **kwargs):
         """Profile a function and store its information."""
 
         start_time = datetime.now()
@@ -50,6 +50,9 @@ def run(self, func, key, *args, **kwargs):
         if value is not None:
             row['hits'] += 1
 
+        if logger:
+            logger.debug('%s("%s") %s (%s)', func.__name__, args[0], 'Miss' if value is None else 'Hit', row['hits'], duration=this_time)
+
         return value
 
     def reset(self):
@@ -88,12 +91,12 @@ def get_calls(self, key):
 stats = StatCollection()
 
 
-def track(func, key):
+def track(func, key, logger):
     """A decorator which handles tracking calls on a function."""
     def wrapped(*args, **kwargs):
         global stats
 
-        return stats.run(func, key, *args, **kwargs)
+        return stats.run(func, key, logger, *args, **kwargs)
     wrapped.__doc__ = func.__doc__
     wrapped.__name__ = func.__name__
     return wrapped
