Fix stat_suffix

getsentry · Jan 28, 2014 · 1aa8f66 · 1aa8f66
1 parent a1e73f7
commit 1aa8f66
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/sentry/tasks/base.py b/src/sentry/tasks/base.py
@@ -12,13 +12,12 @@
 
 
 def instrumented_task(name, queue, stat_suffix=None, **kwargs):
-    statsd_key = 'jobs.duration.{name}'.format(name=name)
-    if stat_suffix:
-        statsd_key += '.{key}'.format(key=stat_suffix)
-
     def wrapped(func):
         @wraps(func)
         def _wrapped(*args, **kwargs):
+            statsd_key = 'jobs.duration.{name}'.format(name=name)
+            if stat_suffix:
+                statsd_key += '.{key}'.format(key=stat_suffix(*args, **kwargs))
             with statsd.timer(statsd_key):
                 result = func(*args, **kwargs)
             return result