github/sprockets-statsd
1
0
Fork 0
mirror of https://github.com/sprockets/sprockets-statsd.git synced 2024-12-26 19:34:54 +00:00
Code Issues Projects Releases Packages Wiki Activity

Reorder logic to emphasize common paths.

Browse source 
I snuck in a call to verify that the connected event is set before the
call to _process_metric.  This closes a small loophole where the
transport could be None inside of _process_metric.
This commit is contained in:
Dave Shawley 2021-03-11 07:03:18 -05:00
parent 00759ed20b
commit 65b5bacbee
No known key found for this signature in database
GPG key ID: 44A9C9992CCFAB82
1 changed files with 13 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
sprockets_statsd/statsd.py
View file

@ -155,6 +155,7 @@ class Processor(asyncio.Protocol):
while not self.should_terminate:
try:
await self._connect_if_necessary()
if self.connected.is_set():
await self._process_metric()
except asyncio.CancelledError:
self.logger.info('task cancelled, exiting')
@ -221,14 +222,11 @@ class Processor(asyncio.Protocol):
async def _process_metric(self):
processing_failed_send = False
if self._failed_sends:
self.logger.debug('using previous send attempt')
metric = self._failed_sends[0]
processing_failed_send = True
else:
if not self._failed_sends:
try:
metric = await asyncio.wait_for(self.queue.get(), 0.1)
self.logger.debug('received %r from queue', metric)
self.queue.task_done()
except asyncio.TimeoutError:
return
else:
@ -240,15 +238,17 @@ class Processor(asyncio.Protocol):
self.logger.debug('preventing send on closed transport')
self._failed_sends.append(metric)
return
else:
self.logger.debug('using previous send attempt')
metric = self._failed_sends[0]
processing_failed_send = True
self.transport.write(metric)
if self.transport.is_closing():
# Writing to a transport does not raise exceptions, it
# will close the transport if a low-level error occurs.
self.logger.debug('transport closed by writing')
else:
if not self.transport.is_closing():
self.logger.debug('sent %r to statsd', metric)
if processing_failed_send:
self._failed_sends.pop(0)
else:
self.queue.task_done()
# Writing to a transport does not raise exceptions, it
# will close the transport if a low-level error occurs.
self.logger.debug('transport closed by writing')