233 lines
7.1 KiB
Python
233 lines
7.1 KiB
Python
"""pyzmq logging handlers.
|
|
|
|
This mainly defines the PUBHandler object for publishing logging messages over
|
|
a zmq.PUB socket.
|
|
|
|
The PUBHandler can be used with the regular logging module, as in::
|
|
|
|
>>> import logging
|
|
>>> handler = PUBHandler('tcp://127.0.0.1:12345')
|
|
>>> handler.root_topic = 'foo'
|
|
>>> logger = logging.getLogger('foobar')
|
|
>>> logger.setLevel(logging.DEBUG)
|
|
>>> logger.addHandler(handler)
|
|
|
|
Or using ``dictConfig``, as in::
|
|
|
|
>>> from logging.config import dictConfig
|
|
>>> socket = Context.instance().socket(PUB)
|
|
>>> socket.connect('tcp://127.0.0.1:12345')
|
|
>>> dictConfig({
|
|
>>> 'version': 1,
|
|
>>> 'handlers': {
|
|
>>> 'zmq': {
|
|
>>> 'class': 'zmq.log.handlers.PUBHandler',
|
|
>>> 'level': logging.DEBUG,
|
|
>>> 'root_topic': 'foo',
|
|
>>> 'interface_or_socket': socket
|
|
>>> }
|
|
>>> },
|
|
>>> 'root': {
|
|
>>> 'level': 'DEBUG',
|
|
>>> 'handlers': ['zmq'],
|
|
>>> }
|
|
>>> })
|
|
|
|
|
|
After this point, all messages logged by ``logger`` will be published on the
|
|
PUB socket.
|
|
|
|
Code adapted from StarCluster:
|
|
|
|
https://github.com/jtriley/StarCluster/blob/StarCluster-0.91/starcluster/logger.py
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from copy import copy
|
|
|
|
import zmq
|
|
|
|
# Copyright (C) PyZMQ Developers
|
|
# Distributed under the terms of the Modified BSD License.
|
|
|
|
|
|
TOPIC_DELIM = "::" # delimiter for splitting topics on the receiving end.
|
|
|
|
|
|
class PUBHandler(logging.Handler):
|
|
"""A basic logging handler that emits log messages through a PUB socket.
|
|
|
|
Takes a PUB socket already bound to interfaces or an interface to bind to.
|
|
|
|
Example::
|
|
|
|
sock = context.socket(zmq.PUB)
|
|
sock.bind('inproc://log')
|
|
handler = PUBHandler(sock)
|
|
|
|
Or::
|
|
|
|
handler = PUBHandler('inproc://loc')
|
|
|
|
These are equivalent.
|
|
|
|
Log messages handled by this handler are broadcast with ZMQ topics
|
|
``this.root_topic`` comes first, followed by the log level
|
|
(DEBUG,INFO,etc.), followed by any additional subtopics specified in the
|
|
message by: log.debug("subtopic.subsub::the real message")
|
|
"""
|
|
|
|
ctx: zmq.Context
|
|
socket: zmq.Socket
|
|
|
|
def __init__(
|
|
self,
|
|
interface_or_socket: str | zmq.Socket,
|
|
context: zmq.Context | None = None,
|
|
root_topic: str = '',
|
|
) -> None:
|
|
logging.Handler.__init__(self)
|
|
self.root_topic = root_topic
|
|
self.formatters = {
|
|
logging.DEBUG: logging.Formatter(
|
|
"%(levelname)s %(filename)s:%(lineno)d - %(message)s\n"
|
|
),
|
|
logging.INFO: logging.Formatter("%(message)s\n"),
|
|
logging.WARN: logging.Formatter(
|
|
"%(levelname)s %(filename)s:%(lineno)d - %(message)s\n"
|
|
),
|
|
logging.ERROR: logging.Formatter(
|
|
"%(levelname)s %(filename)s:%(lineno)d - %(message)s - %(exc_info)s\n"
|
|
),
|
|
logging.CRITICAL: logging.Formatter(
|
|
"%(levelname)s %(filename)s:%(lineno)d - %(message)s\n"
|
|
),
|
|
}
|
|
if isinstance(interface_or_socket, zmq.Socket):
|
|
self.socket = interface_or_socket
|
|
self.ctx = self.socket.context
|
|
else:
|
|
self.ctx = context or zmq.Context()
|
|
self.socket = self.ctx.socket(zmq.PUB)
|
|
self.socket.bind(interface_or_socket)
|
|
|
|
@property
|
|
def root_topic(self) -> str:
|
|
return self._root_topic
|
|
|
|
@root_topic.setter
|
|
def root_topic(self, value: str):
|
|
self.setRootTopic(value)
|
|
|
|
def setRootTopic(self, root_topic: str):
|
|
"""Set the root topic for this handler.
|
|
|
|
This value is prepended to all messages published by this handler, and it
|
|
defaults to the empty string ''. When you subscribe to this socket, you must
|
|
set your subscription to an empty string, or to at least the first letter of
|
|
the binary representation of this string to ensure you receive any messages
|
|
from this handler.
|
|
|
|
If you use the default empty string root topic, messages will begin with
|
|
the binary representation of the log level string (INFO, WARN, etc.).
|
|
Note that ZMQ SUB sockets can have multiple subscriptions.
|
|
"""
|
|
if isinstance(root_topic, bytes):
|
|
root_topic = root_topic.decode("utf8")
|
|
self._root_topic = root_topic
|
|
|
|
def setFormatter(self, fmt, level=logging.NOTSET):
|
|
"""Set the Formatter for this handler.
|
|
|
|
If no level is provided, the same format is used for all levels. This
|
|
will overwrite all selective formatters set in the object constructor.
|
|
"""
|
|
if level == logging.NOTSET:
|
|
for fmt_level in self.formatters.keys():
|
|
self.formatters[fmt_level] = fmt
|
|
else:
|
|
self.formatters[level] = fmt
|
|
|
|
def format(self, record):
|
|
"""Format a record."""
|
|
return self.formatters[record.levelno].format(record)
|
|
|
|
def emit(self, record):
|
|
"""Emit a log message on my socket."""
|
|
|
|
# LogRecord.getMessage explicitly allows msg to be anything _castable_ to a str
|
|
try:
|
|
topic, msg = str(record.msg).split(TOPIC_DELIM, 1)
|
|
except ValueError:
|
|
topic = ""
|
|
else:
|
|
# copy to avoid mutating LogRecord in-place
|
|
record = copy(record)
|
|
record.msg = msg
|
|
|
|
try:
|
|
bmsg = self.format(record).encode("utf8")
|
|
except Exception:
|
|
self.handleError(record)
|
|
return
|
|
|
|
topic_list = []
|
|
|
|
if self.root_topic:
|
|
topic_list.append(self.root_topic)
|
|
|
|
topic_list.append(record.levelname)
|
|
|
|
if topic:
|
|
topic_list.append(topic)
|
|
|
|
btopic = '.'.join(topic_list).encode("utf8", "replace")
|
|
|
|
self.socket.send_multipart([btopic, bmsg])
|
|
|
|
|
|
class TopicLogger(logging.Logger):
|
|
"""A simple wrapper that takes an additional argument to log methods.
|
|
|
|
All the regular methods exist, but instead of one msg argument, two
|
|
arguments: topic, msg are passed.
|
|
|
|
That is::
|
|
|
|
logger.debug('msg')
|
|
|
|
Would become::
|
|
|
|
logger.debug('topic.sub', 'msg')
|
|
"""
|
|
|
|
def log(self, level, topic, msg, *args, **kwargs):
|
|
"""Log 'msg % args' with level and topic.
|
|
|
|
To pass exception information, use the keyword argument exc_info
|
|
with a True value::
|
|
|
|
logger.log(level, "zmq.fun", "We have a %s",
|
|
"mysterious problem", exc_info=1)
|
|
"""
|
|
logging.Logger.log(self, level, f'{topic}{TOPIC_DELIM}{msg}', *args, **kwargs)
|
|
|
|
|
|
# Generate the methods of TopicLogger, since they are just adding a
|
|
# topic prefix to a message.
|
|
for name in "debug warn warning error critical fatal".split():
|
|
try:
|
|
meth = getattr(logging.Logger, name)
|
|
except AttributeError:
|
|
# some methods are missing, e.g. Logger.warn was removed from Python 3.13
|
|
continue
|
|
setattr(
|
|
TopicLogger,
|
|
name,
|
|
lambda self, level, topic, msg, *args, **kwargs: meth(
|
|
self, level, topic + TOPIC_DELIM + msg, *args, **kwargs
|
|
),
|
|
)
|