[ PARENT ]

EDIT :: base_subprocess_log.py

# Module for logging in subprocess. Process related info (call stack) is logged # to simplify investigation of main process hangings # # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2024 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENSE.TXT import atexit import collections import dataclasses import logging import logging.handlers import os import pathlib import queue import time from datetime import datetime from multiprocessing import Event, Process, Queue from raven.handlers.logging import SentryHandler import psutil from lve_utils.sentry import init_lve_utils_sentry_client # NOTE: set records history size relatively small, # otherwise sentry will trim the log message with history records and process ancestors DEFAULT_RECORDS_HISTORY_SIZE = 10 DEFAULT_LOG_EVENTS_TIMEOUT = 300 DEFAULT_LOG_FORMAT = '%(asctime)s - %(name)s - %(levelname)s - %(message)s' @dataclasses.dataclass class ProcessDescription: """ Process description object pid: int - process id ppid: int | None - parent process id user: str | None - process owner status: str | None - process status start_time: datetime | None - process start time cmdline: str | None - process command line """ pid: int ppid: int | None user: str | None status: str | None start_time: datetime | None cmdline: str | None @classmethod def from_psutil(cls, process: psutil.Process): """Build process description object using psutil.Process object""" try: ppid = process.ppid() except Exception: ppid = None try: cmdline = ' '.join(process.cmdline()) except Exception: cmdline = None try: start_time = datetime.fromtimestamp(process.create_time()) except Exception: start_time = None try: user = process.username() except Exception: user = None try: status = process.status() except Exception: status = None return cls(process.pid, ppid, user, status, start_time, cmdline) def to_str(self): return f"PID: {self.pid}, PPID: {self.ppid}, User: {self.user}, Status: {self.status} " \ f"Start time: {self.start_time} Command: {self.cmdline}" class QueueListenerWithHistory(logging.handlers.QueueListener): def __init__( self, queue, *handlers, timeout: int = DEFAULT_LOG_EVENTS_TIMEOUT, records_history_size: int = DEFAULT_RECORDS_HISTORY_SIZE, ): """Queue listener with log records history :param multiprocessing.Queue queue: log records queue :param int timeout: log records flow timeout, defaults to DEFAULT_LOG_EVENTS_TIMEOUT In case timeput is reached, last log records and process ancestors are logged :param int records_history_size: the max number of records to log in case of timeout """ super().__init__(queue, *handlers, respect_handler_level=True) self._records_history_queue: collections.deque[logging.LogRecord] = collections.deque( iterable=[], maxlen=records_history_size ) self._timeout = timeout self._timout_registred = False def dequeue(self, _block): """Dequeue log event""" while True: try: record = self.queue.get(timeout=self._timeout) self._records_history_queue.append(record) break except queue.Empty: self._handle_queue_timeout() return record def _monitor(self): """Main loop to process log events""" self._handle_monitor_started() super()._monitor() self._handle_monitor_finished() def _handle_monitor_started(self): """Start monitoring log events processing time""" self._start_ts = time.time() # pylint: disable=attribute-defined-outside-init def _handle_monitor_finished(self): """Log that monitor finished processing log records""" if self._timout_registred is True: r = self._create_record( logging.ERROR, 'Processing log queue took %.4f seconds. Pid: %s', time.time() - self._start_ts, os.getpid(), ) self.handle(r) def _get_ancestors(self): """Collect process ancestors up to pid 1""" ancestors: list[ProcessDescription] = [] try: process = psutil.Process() ancestors = [ProcessDescription.from_psutil(process)] while (ppid := process.ppid()) != 0: process = psutil.Process(ppid) ancestors.append(ProcessDescription.from_psutil(process)) return ancestors except Exception: return ancestors def _create_record(self, level, msg, *args): """Create log record""" return logging.LogRecord( name=__name__, level=level, pathname='', lineno=1, msg=msg, args=args, exc_info=None, ) def _handle_queue_timeout(self): """ Handle log queue timeout event. Print last events (incl. omitted because debug disabled) and process ancestors to log. """ if self._timout_registred is False: proc_ancestors = '\n'.join( proc_descr.to_str() for proc_descr in self._get_ancestors()[::-1] ) last_events = '\n'.join( f'[{record.levelname}] - {datetime.fromtimestamp(record.created)}: {record.message}' for record in self._records_history_queue ) error_msg = 'Log queue timeout. Last events:\n%s\nProcess ancestors:\n%s' error_args = (last_events, proc_ancestors) r = self._create_record(logging.ERROR, error_msg, *error_args) self.handle(r) self._timout_registred = True def get_log_level(file_name: str): """Get log level based on debug flag presence""" return logging.DEBUG if pathlib.Path(f'{file_name}.debug.flag').exists() else logging.INFO def run_subpprocess_logger(q: Queue, stop_event: Event, file_name: str, timeout: int): """ Main function to process log events in subprocess :param Queue q: log records queue :param Event stop_event: stop event :param str file_name: log file name """ sentry_handler = SentryHandler(init_lve_utils_sentry_client('lvectl'), level=logging.ERROR) fh = logging.FileHandler(file_name) fh.setFormatter(logging.Formatter(DEFAULT_LOG_FORMAT)) fh.setLevel(get_log_level(file_name)) q_handler = QueueListenerWithHistory(q, fh, sentry_handler, timeout=timeout) q_handler.start() stop_event.wait() q_handler.stop() @dataclasses.dataclass class SubprocessLogger: logger_process: Process stop_event: Event subprocess_loggers: dict[str | None, SubprocessLogger] = {} def init_subprocess_logger( logger_name: str | None, file_name: str, timeout: int = DEFAULT_LOG_EVENTS_TIMEOUT, ) -> logging.Logger: """ Get asynchronous logger running in subprocess Log records asynchronously in subprocess (>=info by default, >= debug when debug flag is present). Log records from debug and higher in subprocess in case of timeout. WARNING: multiprocessing.Queue is used to pass log records. It works asynchornously (passing to subprocess in handled by separate thread). This may lead to log records not being passed to the subprocess in case main process got stuck e.g. in kmodlve kernel space. :param str | None logger_name: logger name :param str file_name: log file :param int timeout: log events processing timeout """ global subprocess_loggers subprocess_logger = subprocess_loggers.get(logger_name) if subprocess_logger is not None: return logging.getLogger(logger_name) logger = logging.getLogger(logger_name) # NOTE: We want to pass all the records (incl. debug) to the subprocess # so that we can log them in case of timeout logger.setLevel(logging.DEBUG) subprocess_logger_queue: Queue = Queue() queue_handler = logging.handlers.QueueHandler(subprocess_logger_queue) queue_handler.setLevel(logging.DEBUG) logger.addHandler(queue_handler) stop_event = Event() logger_process = Process(target=run_subpprocess_logger, args=(subprocess_logger_queue, stop_event, file_name, timeout)) logger_process.start() atexit.register(stop_event.set) subprocess_loggers[logger_name] = SubprocessLogger(logger_process, stop_event) return logger def stop_subprocess_logger(name: str): """ Stop subprocess logger """ global subprocess_loggers subprocess_logger = subprocess_loggers.get(name) if subprocess_logger is None: return subprocess_logger.stop_event.set() subprocess_logger.logger_process.join() del subprocess_loggers[name]

SAVE CANCEL