收集策略日志¶
RQAlpha 采用 logbook 作为默认的日志模块,开发者可以通过在 mod 中为 logger 添加 handler 实现自定义的日志收集。
以下是一个简单的 demo。
Mod 示例¶
首先要做的是实现 handler 对象,此处实现的 handler 对象接受 send_log_handler
函数作为参数,该函数会在打印日志的时候被调用。
from logbook.handlers import Handler, StringFormatterHandlerMixin
from logbook.base import NOTSET
from rqalpha.environment import Environment
class LogHandler(Handler, StringFormatterHandlerMixin):
def __init__(self, send_log_handler, level=NOTSET, format_string=None, filter=None, bubble=False):
Handler.__init__(self, level, filter, bubble)
StringFormatterHandlerMixin.__init__(self, format_string)
self.send_log_handler = send_log_handler
def _write(self, level_name, item):
dt = Environment.get_instance().calendar_dt
self.send_log_handler(dt, item, level_name)
def emit(self, record):
msg = self.format(record)
self.lock.acquire()
try:
self._write(record.level_name, msg)
finally:
self.lock.release()
Mod 的实现如下,该 Mod 所做的所有工作仅仅是初始化了 LogHandler
对象并将其传给 user_log 和 user_system_logger。另外需要实现 _send_log
方法,将日志送往需要的去处。
from rqalpha.interface import AbstractMod
from rqalpha.utils.logger import user_system_log, user_log
class CustomLogHandlerMod(AbstractMod):
def _send_log(self, dt, text, log_tag):
# TODO
pass
def start_up(self, env, mod_config):
user_log.handlers.append(LogHandler(self._send_log, bubble=True))
user_system_log.handlers.append(LogHandler(self._send_log, bubble=True))
def tear_down(self, code, exception=None):
pass
def load_mod():
return CustomLogHandlerMod()