You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
2.3 KiB
Python
82 lines
2.3 KiB
Python
"""Structured JSON logging via structlog."""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
import sys
|
|
from contextvars import ContextVar
|
|
from typing import Any
|
|
|
|
import structlog
|
|
|
|
_job_run_id: ContextVar[str] = ContextVar("job_run_id", default="")
|
|
|
|
|
|
def bind_job_run_id(run_id: str) -> None:
|
|
_job_run_id.set(run_id)
|
|
|
|
|
|
def _add_job_run_id(
|
|
logger: Any, method: str, event_dict: dict[str, Any]
|
|
) -> dict[str, Any]:
|
|
run_id = _job_run_id.get()
|
|
if run_id:
|
|
event_dict["job_run_id"] = run_id
|
|
return event_dict
|
|
|
|
|
|
def _make_events_sink_processor():
|
|
"""Return a structlog processor that tees events to EventsStore.
|
|
|
|
Imported lazily to avoid a circular import at configure_logging() call time
|
|
(EventsStore lives in apps/web which is not always present in subprocess env).
|
|
Falls back to a no-op if the import fails.
|
|
"""
|
|
try:
|
|
from apps.web.services.events_store import structlog_sink_processor
|
|
return structlog_sink_processor
|
|
except Exception:
|
|
return None
|
|
|
|
|
|
def configure_logging(level: str = "INFO", enable_events_sink: bool = False) -> None:
|
|
import os as _os
|
|
logging.basicConfig(
|
|
format="%(message)s",
|
|
stream=sys.stdout,
|
|
level=getattr(logging, level.upper(), logging.INFO),
|
|
)
|
|
# In subprocess env, inherit JOB_RUN_ID from parent if set
|
|
_env_id = _os.environ.get("JOB_RUN_ID")
|
|
if _env_id:
|
|
bind_job_run_id(_env_id)
|
|
|
|
processors: list[Any] = [
|
|
structlog.contextvars.merge_contextvars,
|
|
_add_job_run_id,
|
|
structlog.stdlib.add_log_level,
|
|
structlog.processors.TimeStamper(fmt="iso"),
|
|
structlog.processors.StackInfoRenderer(),
|
|
structlog.processors.format_exc_info,
|
|
]
|
|
|
|
if enable_events_sink:
|
|
sink = _make_events_sink_processor()
|
|
if sink is not None:
|
|
processors.append(sink)
|
|
|
|
processors.append(structlog.processors.JSONRenderer())
|
|
|
|
structlog.configure(
|
|
processors=processors,
|
|
wrapper_class=structlog.make_filtering_bound_logger(
|
|
getattr(logging, level.upper(), logging.INFO)
|
|
),
|
|
context_class=dict,
|
|
logger_factory=structlog.PrintLoggerFactory(),
|
|
cache_logger_on_first_use=False,
|
|
)
|
|
|
|
|
|
def get_logger(name: str = "") -> structlog.BoundLogger:
|
|
return structlog.get_logger(name)
|