from contextvars import ContextVar from typing import Optional from slack_bolt import BoltContext # Middleware correlation_id: ContextVar[Optional[str]] = ContextVar("correlation_id", default=None)

from logging import Filter, LogRecord from typing import Optional from.modules.logging import correlation_id def _trim_string(string: Optional[str], string_length: Optional[int]) -> Optional[str]: return string[:string_length] if string_length is not None and string else string class CorrelationIdFilter(Filter): """Logging filter to attached correlation IDs to log records""" def __init__( self, name: str = "", uuid_length: int | None = None, default_value: str | None = None, ): super().__init__(name=name) self.uuid_length = uuid_length self.default_value = default_value def filter(self, record: "LogRecord") -> bool: """  Attach a correlation ID to the log record.   Since the correlation ID is defined in the middleware layer, any  log generated from a request after this point can easily be searched  for, if the correlation ID is added to the message, or included as  metadata.  """ cid = correlation_id.get(self.default_value) record.correlation_id = _trim_string(cid, self.uuid_length) return True

 from.modules.logging import correlation_id def call(context: BoltContext, logger: Logger, next: Callable[[], None]) -> None: # Capture the current context current_context = copy_context() current_thread = threading.current_thread() id_value: str | None = correlation_id.get() if id_value is not None: logger.info( f"{__name__} {id_value=} found id in contextvar {current_thread=}" ) elif context.get("correlation_id", None) is not None: id_value = context["correlation_id"] token = correlation_id.set(id_value) logger.info(f"{__name__} {id_value=} found id in context {current_thread=}") else: id_value = uuid4().hex context["correlation_id"] = id_value token = correlation_id.set(id_value) logger.info(f"{__name__} {id_value=} generated new id {current_thread=}") try: current_context.run(next) finally: if token: correlation_id.reset(token)