def test_tcp_timeout(mock_socket): """Timeout on TCPStatsClient should be set on socket.""" test_timeout = 321 cl = TCPStatsClient(timeout=test_timeout) cl.incr('foo') cl._sock.settimeout.assert_called_once() cl._sock.settimeout.assert_called_with(test_timeout)
def _tcp_client(prefix=None, addr=None, port=None, timeout=None): if not addr: addr = ADDR[0] if not port: port = ADDR[1] sc = TCPStatsClient(host=addr, port=port, prefix=prefix, timeout=timeout) sc._sock = mock.Mock() return sc
def client(self): if self._client is None: if self.protocol is Protocols.udp: self._client = StatsClient(**self.config) else: # self.protocol is Protocols.tcp self._client = TCPStatsClient(**self.config) return self._client
import json import os import time import requests from requests.exceptions import ConnectionError from statsd import TCPStatsClient statsd = TCPStatsClient(host=os.environ.get('STATSD_HOST', '192.168.59.103'), port=os.environ.get('STATSD_PORT', 8125), prefix=os.environ.get('STATSD_PREFIX', 'locust')) def post_with_retries(session, url, payload, headers): backoff = 0 retries = 10 while retries > 0: try: response = session.post(url, data=json.dumps(payload), headers=headers) print response.text return response break except ConnectionError, e: print "Retrying in %s seconds" % 2**backoff time.sleep(2**backoff) backoff += 1 retries -= -1
def init_statsd(): statsd = TCPStatsClient(host= Config.STATSD_HOST, port= Config.STATSD_PORT, prefix= Config.STATSD_PREFIX) return statsd
def tcp_statsd_client(**client_params): return TCPStatsClient(**client_params)