Show More
base.py
107 lines
| 2.8 KiB
| text/x-python
|
PythonLexer
r4632 | from __future__ import absolute_import, division, unicode_literals | |||
import random | ||||
from collections import deque | ||||
from datetime import timedelta | ||||
from .timer import Timer | ||||
class StatsClientBase(object): | ||||
"""A Base class for various statsd clients.""" | ||||
def close(self): | ||||
"""Used to close and clean up any underlying resources.""" | ||||
raise NotImplementedError() | ||||
def _send(self): | ||||
raise NotImplementedError() | ||||
def pipeline(self): | ||||
raise NotImplementedError() | ||||
def timer(self, stat, rate=1): | ||||
return Timer(self, stat, rate) | ||||
def timing(self, stat, delta, rate=1): | ||||
""" | ||||
Send new timing information. | ||||
`delta` can be either a number of milliseconds or a timedelta. | ||||
""" | ||||
if isinstance(delta, timedelta): | ||||
# Convert timedelta to number of milliseconds. | ||||
delta = delta.total_seconds() * 1000. | ||||
self._send_stat(stat, '%0.6f|ms' % delta, rate) | ||||
def incr(self, stat, count=1, rate=1): | ||||
"""Increment a stat by `count`.""" | ||||
self._send_stat(stat, '%s|c' % count, rate) | ||||
def decr(self, stat, count=1, rate=1): | ||||
"""Decrement a stat by `count`.""" | ||||
self.incr(stat, -count, rate) | ||||
def gauge(self, stat, value, rate=1, delta=False): | ||||
"""Set a gauge value.""" | ||||
if value < 0 and not delta: | ||||
if rate < 1: | ||||
if random.random() > rate: | ||||
return | ||||
with self.pipeline() as pipe: | ||||
pipe._send_stat(stat, '0|g', 1) | ||||
pipe._send_stat(stat, '%s|g' % value, 1) | ||||
else: | ||||
prefix = '+' if delta and value >= 0 else '' | ||||
self._send_stat(stat, '%s%s|g' % (prefix, value), rate) | ||||
def set(self, stat, value, rate=1): | ||||
"""Set a set value.""" | ||||
self._send_stat(stat, '%s|s' % value, rate) | ||||
def _send_stat(self, stat, value, rate): | ||||
self._after(self._prepare(stat, value, rate)) | ||||
def _prepare(self, stat, value, rate): | ||||
if rate < 1: | ||||
if random.random() > rate: | ||||
return | ||||
value = '%s|@%s' % (value, rate) | ||||
if self._prefix: | ||||
stat = '%s.%s' % (self._prefix, stat) | ||||
return '%s:%s' % (stat, value) | ||||
def _after(self, data): | ||||
if data: | ||||
self._send(data) | ||||
class PipelineBase(StatsClientBase): | ||||
def __init__(self, client): | ||||
self._client = client | ||||
self._prefix = client._prefix | ||||
self._stats = deque() | ||||
def _send(self): | ||||
raise NotImplementedError() | ||||
def _after(self, data): | ||||
if data is not None: | ||||
self._stats.append(data) | ||||
def __enter__(self): | ||||
return self | ||||
def __exit__(self, typ, value, tb): | ||||
self.send() | ||||
def send(self): | ||||
if not self._stats: | ||||
return | ||||
self._send() | ||||
def pipeline(self): | ||||
return self.__class__(self) | ||||