Asynchronous StatsD reporter
Find a file
Dave Shawley 05650286da
Remove usage of typing.Protocol.
It wasn't adding too much and it isn't available in Python 3.7.
2021-03-23 21:21:16 -04:00
.github/workflows Ensure that docs build as part of CI. 2021-03-11 07:43:17 -05:00
docs Expose UDP support in the mixins. 2021-03-23 07:58:45 -04:00
sprockets_statsd Expose UDP support in the mixins. 2021-03-23 07:58:45 -04:00
tests Remove usage of typing.Protocol. 2021-03-23 21:21:16 -04:00
CHANGELOG.rst Add support for UDP sockets. 2021-03-23 07:57:32 -04:00
example.py Add Application & RequestHandler mixins. 2021-03-10 06:46:06 -05:00
LICENSE Project scaffolding. 2021-03-08 07:35:31 -05:00
MANIFEST.in Add Application & RequestHandler mixins. 2021-03-10 06:46:06 -05:00
README.rst Add support for UDP sockets. 2021-03-23 07:57:32 -04:00
setup.cfg Add support for UDP sockets. 2021-03-23 07:57:32 -04:00
setup.py Project scaffolding. 2021-03-08 07:35:31 -05:00
tox.ini Switch to asynctest for Py 3.7 support. 2021-03-10 06:46:17 -05:00

Asynchronously send metrics to a statsd_ instance.

This library provides connectors to send metrics to a statsd_ instance using either TCP or UDP.

.. code-block:: python

   import asyncio
   import time

   import sprockets_statsd.statsd

   statsd = sprockets_statsd.statsd.Connector(
      host=os.environ.get('STATSD_HOST', '127.0.0.1'))

   async def do_stuff():
      start = time.time()
      response = make_some_http_call()
      statsd.inject_metric(f'timers.http.something.{response.code}',
                           (time.time() - start) * 1000.0, 'ms')

   async def main():
      await statsd.start()
      try:
         do_stuff()
      finally:
         await statsd.stop()

The ``Connector`` instance maintains a resilient connection to the target StatsD instance, formats the metric data
into payloads, and sends them to the StatsD target.  It defaults to using TCP as the transport but will use UDP if
the ``ip_protocol`` keyword is set to ``socket.IPPROTO_UDP``.  The ``Connector.start`` method starts a background
``asyncio.Task`` that is responsible for maintaining the connection.  The ``inject_metric`` method enqueues metric
data to send and the task consumes the internal queue when it is connected.

Tornado helpers
===============
The ``sprockets_statsd.mixins`` module contains mix-in classes that make reporting metrics from your tornado_ web
application simple.

.. code-block:: python

   import asyncio
   import logging
   
   from tornado import ioloop, web
   
   import sprockets_statsd.mixins
   
   
   class MyHandler(sprockets_statsd.mixins.RequestHandler,
                   web.RequestHandler):
       async def get(self):
           with self.execution_timer('some-operation'):
               await self.do_something()
           self.set_status(204)
   
       async def do_something(self):
           await asyncio.sleep(1)
   
   
   class Application(sprockets_statsd.mixins.Application, web.Application):
       def __init__(self, **settings):
           super().__init__([web.url('/', MyHandler)], **settings)
   
       async def on_start(self):
           await self.start_statsd()
   
       async def on_stop(self):
           await self.stop_statsd()
   
   
   if __name__ == '__main__':
       logging.basicConfig(level=logging.DEBUG)
       app = Application()
       app.listen(8888)
       iol = ioloop.IOLoop.current()
       try:
           iol.add_callback(app.on_start)
           iol.start()
       except KeyboardInterrupt:
           iol.add_future(asyncio.ensure_future(app.on_stop()),
                          lambda f: iol.stop())
           iol.start()

This application will emit two timing metrics each time that the endpoint is invoked::

   applications.timers.some-operation:1001.3449192047119|ms
   applications.timers.MyHandler.GET.204:1002.4960041046143|ms

You will need to set the ``$STATSD_HOST`` environment variable to enable the statsd processing inside of the
application.  The ``RequestHandler`` class exposes methods that send counter and timing metrics to a statsd server.
The connection is managed by the ``Application`` provided that you call the ``start_statsd`` method during application
startup.

Metrics are sent by a ``asyncio.Task`` that is started by ``start_statsd``.  The request handler methods insert the
metric data onto a ``asyncio.Queue`` that the task reads from.  Metric data remains on the queue when the task is
not connected to the server and will be sent in the order received when the task establishes the server connection.

.. _statsd: https://github.com/statsd/statsd/
.. _tornado: https://tornadoweb.org/