forked from M-Labs/artiq
1
0
Fork 0

protocols/logging: add LogForwarder

This commit is contained in:
Sebastien Bourdeauducq 2015-10-16 18:35:02 +08:00
parent c0796249b3
commit cee8f288de
1 changed files with 37 additions and 0 deletions

View File

@ -2,6 +2,7 @@ import asyncio
import logging import logging
from artiq.protocols.asyncio_server import AsyncioServer from artiq.protocols.asyncio_server import AsyncioServer
from artiq.tools import TaskObject
_fwd_logger = logging.getLogger("fwd") _fwd_logger = logging.getLogger("fwd")
@ -70,3 +71,39 @@ class Server(AsyncioServer):
extra={"source": source}) extra={"source": source})
finally: finally:
writer.close() writer.close()
class LogForwarder(logging.Handler, TaskObject):
def __init__(self, host, port, reconnect_timer=5.0, queue_size=1000,
**kwargs):
logging.Handler.__init__(self, **kwargs)
self.host = host
self.port = port
self.setFormatter(logging.Formatter(
"%(source)s:%(levelno)d:%(name)s:%(message)s"))
self._queue = asyncio.Queue(queue_size)
self.reconnect_timer = reconnect_timer
def emit(self, record):
message = self.format(record)
try:
self._queue.put_nowait(message)
except asyncio.QueueFull:
pass
async def _do(self):
while True:
try:
reader, writer = await asyncio.open_connection(self.host,
self.port)
writer.write(_init_string)
while True:
message = await self._queue.get() + "\n"
writer.write(message.encode())
await writer.drain()
except asyncio.CancelledError:
return
except:
await asyncio.sleep(self.reconnect_timer)
finally:
writer.close()