Merge pull request #66 from DataDog/yann/timed-context-cosmetic

Yann/timed context cosmetic
DataDog · Jul 1, 2015 · 16666d2 · 16666d2
2 parents bcfccf2 + 5a44b92
commit 16666d2
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/datadog/dogstatsd/base.py b/datadog/dogstatsd/base.py
@@ -143,12 +143,8 @@ def __call__(self, func):
             """Decorator which returns the elapsed time of the function call."""
             @wraps(func)
             def wrapped(*args, **kwargs):
-                start = time()
-                result = func(*args, **kwargs)
-                self.statsd.timing(
-                    self.metric, time() - start, tags=self.tags,
-                    sample_rate=self.sample_rate)
-                return result
+                with self:
+                    return func(*args, **kwargs)
             return wrapped
 
         def __enter__(self):