Python多进程环境下日志管理的最佳实践与实战指南

import logging

# 配置基本的日志格式
logging.basicConfig(level=logging.INFO, 
                    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')

# 创建一个日志记录器
logger = logging.getLogger(__name__)

# 使用日志记录器
logger.info("这是一条信息日志")
logger.warning("这是一条警告日志")
logger.error("这是一条错误日志")

import logging
import multiprocessing
import random
import time

def worker_process(queue):
    logger = logging.getLogger(f"Worker-{multiprocessing.current_process().name}")
    for _ in range(5):
        time.sleep(random.random())
        logger.info(f"Worker {multiprocessing.current_process().name} is working")
        queue.put(logger.name + ": " + f"Worker {multiprocessing.current_process().name} is working")

def logger_process(queue):
    logger = logging.getLogger("LoggerProcess")
    logger.setLevel(logging.INFO)
    handler = logging.FileHandler("multiprocess.log")
    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    handler.setFormatter(formatter)
    logger.addHandler(handler)

    while True:
        try:
            record = queue.get()
            if record == "STOP":
                break
            logger.info(record)
        except Exception:
            import sys, traceback
            print('Whoops! Problem:', file=sys.stderr)
            traceback.print_exc(file=sys.stderr)

if __name__ == "__main__":
    queue = multiprocessing.Queue(-1)
    logger_p = multiprocessing.Process(target=logger_process, args=(queue,))
    logger_p.start()

    workers = []
    for i in range(5):
        worker = multiprocessing.Process(target=worker_process, args=(queue,))
        workers.append(worker)
        worker.start()

    for worker in workers:
        worker.join()

    queue.put("STOP")
    logger_p.join()

import multiprocessing
import logging
from logging.handlers import RotatingFileHandler
import time
import random
import os

class MultiProcessSafeHandler(RotatingFileHandler):
    def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=False):
        super().__init__(filename, mode, maxBytes, backupCount, encoding, delay)
        self.mode = mode
        self.encoding = encoding
        self.delay = delay
        self.maxBytes = maxBytes
        self.backupCount = backupCount

    def emit(self, record):
        try:
            if self.shouldRollover(record):
                self.doRollover()
            logging.FileHandler.emit(self, record)
        except Exception:
            self.handleError(record)

    def doRollover(self):
        if self.stream:
            self.stream.close()
            self.stream = None
        if self.backupCount > 0:
            for i in range(self.backupCount - 1, 0, -1):
                sfn = self.rotation_filename("%s.%d" % (self.baseFilename, i))
                dfn = self.rotation_filename("%s.%d" % (self.baseFilename, i + 1))
                if os.path.exists(sfn):
                    if os.path.exists(dfn):
                        os.remove(dfn)
                    os.rename(sfn, dfn)
            dfn = self.rotation_filename(self.baseFilename + ".1")
            if os.path.exists(dfn):
                os.remove(dfn)
            self.rotate(self.baseFilename, dfn)
        if not self.delay:
            self.stream = self._open()

    def shouldRollover(self, record):
        if self.stream is None:
            self.stream = self._open()
        if self.maxBytes > 0:
            msg = "%s\n" % self.format(record)
            self.stream.seek(0, 2)
            if self.stream.tell() + len(msg) >= self.maxBytes:
                return 1
        return 0

def worker_process(name):
    logger = logging.getLogger(name)
    for _ in range(5):
        time.sleep(random.random())
        logger.info(f"Worker {name} is working")

if __name__ == "__main__":
    log_file = "multiprocess_safe.log"
    handler = MultiProcessSafeHandler(log_file, maxBytes=1024, backupCount=5)
    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    handler.setFormatter(formatter)

    root_logger = logging.getLogger()
    root_logger.setLevel(logging.INFO)
    root_logger.addHandler(handler)

    processes = []
    for i in range(5):
        p = multiprocessing.Process(target=worker_process, args=(f"Worker-{i}",))
        processes.append(p)
        p.start()

    for p in processes:
        p.join()

import multiprocessing
import logging
import time
import random

def worker_process(name):
    logger = multiprocessing.get_logger()
    for _ in range(5):
        time.sleep(random.random())
        logger.info(f"Worker {name} is working")

if __name__ == "__main__":
    multiprocessing.log_to_stderr(logging.INFO)

    processes = []
    for i in range(5):
        p = multiprocessing.Process(target=worker_process, args=(f"Worker-{i}",))
        processes.append(p)
        p.start()

    for p in processes:
        p.join()

import logging
import multiprocessing
from contextlib import contextmanager

@contextmanager
def log_manager(name):
    logger = logging.getLogger(name)
    handler = logging.FileHandler(f"{name}.log")
    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    handler.setFormatter(formatter)
    logger.addHandler(handler)
    logger.setLevel(logging.INFO)
    
    try:
        yield logger
    finally:
        handler.close()
        logger.removeHandler(handler)

def worker_process(name):
    with log_manager(name) as logger:
        for i in range(5):
            logger.info(f"Worker {name} is working - step {i}")

if __name__ == "__main__":
    processes = []
    for i in range(5):
        p = multiprocessing.Process(target=worker_process, args=(f"Worker-{i}",))
        processes.append(p)
        p.start()

    for p in processes:
        p.join()

# logging.yaml 配置文件内容
"""
version: 1
formatters:
  standard:
    format: '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
handlers:
  console:
    class: logging.StreamHandler
    level: DEBUG
    formatter: standard
    stream: ext://sys.stdout
  file:
    class: logging.handlers.RotatingFileHandler
    level: INFO
    formatter: standard
    filename: multiprocess_app.log
    maxBytes: 10485760
    backupCount: 5
    encoding: utf8
loggers:
  worker:
    level: INFO
    handlers: [console, file]
    propagate: no
root:
  level: INFO
  handlers: [console]
"""
```python
import logging.config
import multiprocessing
import yaml
import os
def setup_logging(config_path='logging.yaml', default_level=logging.INFO):
    if os.path.exists(config_path):
        with open(config_path, 'rt') as f:
            try:
                config = yaml.safe_load(f.read())
                logging.config.dictConfig(config)
            except Exception as e:
                print(f'Error in Logging Configuration: {e}')
                logging.basicConfig(level=default_level)
    else:
        logging.basicConfig(level=default_level)
        print('Failed to load configuration file. Using default configs')
def worker_process(name):
    logger = logging.getLogger(f"worker.{name}")
    for i in range(5):
        logger.info(f"Worker {name} processing task {i}")
        time.sleep(random.random())
if __name__ == "__main__":
    setup_logging()
    processes = []
    for i in range(5):
        p = multiprocessing.Process(target=worker_process, args=(f"Worker-{i}",))
        processes.append(p)
        p.start()
    for p in processes:
        p.join()

import logging
import multiprocessing
import time
import random

class ProcessFilter(logging.Filter):
    """自定义进程过滤器，用于过滤特定进程的日志"""
    
    def __init__(self, process_name=None):
        super().__init__()
        self.process_name = process_name

    def filter(self, record):
        if self.process_name is None:
            return True
        return record.processName == self.process_name

def setup_logger(name, log_file, level=logging.INFO, process_name=None):
    formatter = logging.Formatter(
        '%(asctime)s - %(processName)s - %(name)s - %(levelname)s - %(message)s'
    )
    
    handler = logging.FileHandler(log_file)
    handler.setFormatter(formatter)
    
    logger = logging.getLogger(name)
    logger.setLevel(level)
    
    if process_name:
        process_filter = ProcessFilter(process_name)
        handler.addFilter(process_filter)
    
    logger.addHandler(handler)
    return logger

def worker_task(name):
    logger = setup_logger(
        name=f"worker.{name}",
        log_file="filtered_processes.log",
        process_name=multiprocessing.current_process().name
    )
    
    for i in range(5):
        logger.info(f"Processing task {i}")
        time.sleep(random.random())

if __name__ == "__main__":
    processes = []
    for i in range(3):
        p = multiprocessing.Process(
            target=worker_task,
            name=f"Worker-{i}",
            args=(f"Worker-{i}",)
        )
        processes.append(p)
        p.start()

    for p in processes:
        p.join()

import logging
import multiprocessing
import queue
import threading
import time
import random
from datetime import datetime

class LogAggregator:
    def __init__(self, output_file):
        self.output_file = output_file
        self.log_queue = multiprocessing.Queue()
        self.should_stop = multiprocessing.Event()
        self.aggregator_process = None

    def start(self):
        self.aggregator_process = multiprocessing.Process(
            target=self._aggregate_logs
        )
        self.aggregator_process.start()

    def stop(self):
        self.should_stop.set()
        self.log_queue.put(None)  # 发送停止信号
        if self.aggregator_process:
            self.aggregator_process.join()

    def _aggregate_logs(self):
        with open(self.output_file, 'a') as f:
            while not self.should_stop.is_set():
                try:
                    log_entry = self.log_queue.get(timeout=1)
                    if log_entry is None:
                        break
                    f.write(f"{log_entry}\n")
                    f.flush()
                except queue.Empty:
                    continue

    def log(self, message, level="INFO", process_name=None):
        timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S.%f')[:-3]
        process_name = process_name or multiprocessing.current_process().name
        log_entry = f"{timestamp} - {process_name} - {level} - {message}"
        self.log_queue.put(log_entry)

def worker_process(aggregator, worker_id):
    for i in range(5):
        message = f"Worker {worker_id} processing task {i}"
        aggregator.log(message)
        time.sleep(random.random())

if __name__ == "__main__":
    # 创建日志聚合器
    aggregator = LogAggregator("aggregated_logs.log")
    aggregator.start()

    # 创建多个工作进程
    processes = []
    for i in range(3):
        p = multiprocessing.Process(
            target=worker_process,
            args=(aggregator, i)
        )
        processes.append(p)
        p.start()

    # 等待所有进程完成
    for p in processes:
        p.join()

    # 停止日志聚合器
    aggregator.stop()

import logging
import multiprocessing
import os
from datetime import datetime
import time
import random

class MultiLevelLogger:
    def __init__(self, base_dir="logs"):
        self.base_dir = base_dir
        self.levels = {
            'DEBUG': logging.DEBUG,
            'INFO': logging.INFO,
            'WARNING': logging.WARNING,
            'ERROR': logging.ERROR,
            'CRITICAL': logging.CRITICAL
        }
        self._setup_directories()
        self._setup_loggers()

    def _setup_directories(self):
        for level in self.levels.keys():
            dir_path = os.path.join(self.base_dir, level.lower())
            os.makedirs(dir_path, exist_ok=True)

    def _setup_loggers(self):
        self.loggers = {}
        for level_name, level_value in self.levels.items():
            logger = logging.getLogger(f"multi_level.{level_name}")
            logger.setLevel(level_value)

            # 创建文件处理器
            log_file = os.path.join(
                self.base_dir,
                level_name.lower(),
                f"{level_name.lower()}_{datetime.now().strftime('%Y%m%d')}.log"
            )
            handler = logging.FileHandler(log_file)
            
            # 设置格式化器
            formatter = logging.Formatter(
                '%(asctime)s - %(processName)s - %(name)s - %(levelname)s - %(message)s'
            )
            handler.setFormatter(formatter)
            
            logger.addHandler(handler)
            self.loggers[level_name] = logger

    def log(self, level, message):
        if level in self.loggers:
            self.loggers[level].log(self.levels[level], message)

def worker_process(logger, worker_id):
    levels = ['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL']
    for i in range(5):
        level = random.choice(levels)
        message = f"Worker {worker_id} generated {level} message for task {i}"
        logger.log(level, message)
        time.sleep(random.random())

if __name__ == "__main__":
    # 创建多级日志记录器
    multi_logger = MultiLevelLogger()

    # 创建多个工作进程
    processes = []
    for i in range(3):
        p = multiprocessing.Process(
            target=worker_process,
            args=(multi_logger, i)
        )
        processes.append(p)
        p.start()

    # 等待所有进程完成
    for p in processes:
        p.join()