Show More
udp.py
55 lines
| 1.6 KiB
| text/x-python
|
PythonLexer
r4909 | ||||
r4632 | ||||
import socket | ||||
from .base import StatsClientBase, PipelineBase | ||||
class Pipeline(PipelineBase): | ||||
def __init__(self, client): | ||||
super(Pipeline, self).__init__(client) | ||||
self._maxudpsize = client._maxudpsize | ||||
def _send(self): | ||||
data = self._stats.popleft() | ||||
while self._stats: | ||||
# Use popleft to preserve the order of the stats. | ||||
stat = self._stats.popleft() | ||||
if len(stat) + len(data) + 1 >= self._maxudpsize: | ||||
self._client._after(data) | ||||
data = stat | ||||
else: | ||||
data += '\n' + stat | ||||
self._client._after(data) | ||||
class StatsClient(StatsClientBase): | ||||
"""A client for statsd.""" | ||||
def __init__(self, host='localhost', port=8125, prefix=None, | ||||
maxudpsize=512, ipv6=False): | ||||
"""Create a new client.""" | ||||
fam = socket.AF_INET6 if ipv6 else socket.AF_INET | ||||
family, _, _, _, addr = socket.getaddrinfo( | ||||
host, port, fam, socket.SOCK_DGRAM)[0] | ||||
self._addr = addr | ||||
self._sock = socket.socket(family, socket.SOCK_DGRAM) | ||||
self._prefix = prefix | ||||
self._maxudpsize = maxudpsize | ||||
def _send(self, data): | ||||
"""Send data to statsd.""" | ||||
try: | ||||
self._sock.sendto(data.encode('ascii'), self._addr) | ||||
except (socket.error, RuntimeError): | ||||
# No time for love, Dr. Jones! | ||||
pass | ||||
def close(self): | ||||
if self._sock and hasattr(self._sock, 'close'): | ||||
self._sock.close() | ||||
self._sock = None | ||||
def pipeline(self): | ||||
return Pipeline(self) | ||||