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 _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)
