updated logging
This commit is contained in:
1 parent
de8026fcef
commit
e2600d1fda
1 file changed
+4
-3
@@ -40,17 +40,18 @@ class JSONFormatter(logging.Formatter):
|
|||||||
return json.dumps(payload)
|
return json.dumps(payload)
|
||||||
|
|
||||||
|
|
||||||
def configure_logging(monitor_name: str):
|
def configure_logging(monitor_name: str, log_dir: str = "/logs"):
|
||||||
handler = logging.StreamHandler()
|
handler = logging.StreamHandler()
|
||||||
handler.setFormatter(JSONFormatter())
|
handler.setFormatter(JSONFormatter())
|
||||||
|
|
||||||
def inject_monitor(record: logging.LogRecord) -> bool:
|
def inject_monitor(record: logging.LogRecord) -> bool:
|
||||||
record.monitor = monitor_name
|
record.monitor_name = monitor_name
|
||||||
return True
|
return True
|
||||||
|
|
||||||
handler.addFilter(inject_monitor)
|
handler.addFilter(inject_monitor)
|
||||||
|
|
||||||
file_handler = RotatingFileHandler(
|
file_handler = RotatingFileHandler(
|
||||||
f"/logs/{monitor_name}.jsonl",
|
f"{log_dir}/{monitor_name}.jsonl",
|
||||||
maxBytes=10 * 1024 * 1024,
|
maxBytes=10 * 1024 * 1024,
|
||||||
backupCount=5,
|
backupCount=5,
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in new issue
Block a user