##// END OF EJS Templates
redis-lock: add rhodecode logging prefix to properly log things.
redis-lock: add rhodecode logging prefix to properly log things.

File last commit:

r4632:ed2c6dae stable
r4707:5ead8f47 stable
Show More
timer.py
71 lines | 2.0 KiB | text/x-python | PythonLexer
app: added statsd calls for monitoring.
r4632 from __future__ import absolute_import, division, unicode_literals
import functools
# Use timer that's not susceptible to time of day adjustments.
try:
# perf_counter is only present on Py3.3+
from time import perf_counter as time_now
except ImportError:
# fall back to using time
from time import time as time_now
def safe_wraps(wrapper, *args, **kwargs):
"""Safely wraps partial functions."""
while isinstance(wrapper, functools.partial):
wrapper = wrapper.func
return functools.wraps(wrapper, *args, **kwargs)
class Timer(object):
"""A context manager/decorator for statsd.timing()."""
def __init__(self, client, stat, rate=1):
self.client = client
self.stat = stat
self.rate = rate
self.ms = None
self._sent = False
self._start_time = None
def __call__(self, f):
"""Thread-safe timing function decorator."""
@safe_wraps(f)
def _wrapped(*args, **kwargs):
start_time = time_now()
try:
return f(*args, **kwargs)
finally:
elapsed_time_ms = 1000.0 * (time_now() - start_time)
self.client.timing(self.stat, elapsed_time_ms, self.rate)
return _wrapped
def __enter__(self):
return self.start()
def __exit__(self, typ, value, tb):
self.stop()
def start(self):
self.ms = None
self._sent = False
self._start_time = time_now()
return self
def stop(self, send=True):
if self._start_time is None:
raise RuntimeError('Timer has not started.')
dt = time_now() - self._start_time
self.ms = 1000.0 * dt # Convert to milliseconds.
if send:
self.send()
return self
def send(self):
if self.ms is None:
raise RuntimeError('No data recorded.')
if self._sent:
raise RuntimeError('Already sent data.')
self._sent = True
self.client.timing(self.stat, self.ms, self.rate)