Python3使用元类实现一个线程安全的logger

最新推荐文章于 2024-09-08 02:13:18 发布

原创最新推荐文章于 2024-09-08 02:13:18 发布 · 483 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

python 专栏收录该内容

28 篇文章

订阅专栏

该博客介绍了如何使用Python的logging模块创建一个线程安全的单例日志类。日志类实现了将信息日志和错误日志分别写入不同的文件，并且使用TimedRotatingFileHandler按天进行日志轮转。同时，通过过滤器设置，使得信息日志只记录警告及以下级别，而错误日志记录错误及以上级别。

import threading
import logging  # 引入logging模块
import logging.handlers
# 这个路径抽离到配置文件中去
info_log_path = "/tmp/service/info.log"
err_log_path = "/tmp/service/error.log"


class SingletonType(type):
    """线程安全的单例的元类,只要继承了metaclass=SingletonType,就是一个线程安全的单例类,可以抽离到公共文件中被引用"""
    _instance_lock = threading.Lock()

    def __call__(cls, *args, **kwargs):
        if not hasattr(cls, "_instance"):
            with SingletonType._instance_lock:
                if not hasattr(cls, "_instance"):
                    cls._instance = super(SingletonType, cls).__call__(*args, **kwargs)
        return cls._instance


class SingleLogger(metaclass=SingletonType):
    def __init__(self):
        # 第一步，创建一个logger,这里默认是root
        self.logger = logging.getLogger()
        self.logger.setLevel(logging.INFO)  # Log等级总开关

        # 第二步，创建一个handler，用于写入日志文件
        # fh = logging.FileHandler(info_log_path,mode="w")
        info_fh = logging.handlers.TimedRotatingFileHandler(info_log_path, when="midnight")

        # fh.setLevel(logging.DEBUG)  # 输出到file的log等级的开关
        # 设置日志过滤器
        info_filter = logging.Filter()
        info_filter.filter = lambda record: record.levelno <= logging.WARNING  # 设置过滤等级
        info_fh.setLevel(logging.INFO)
        info_fh.addFilter(info_filter)

        # fh = logging.FileHandler(info_log_path,mode="w")
        err_fh = logging.handlers.TimedRotatingFileHandler(err_log_path, when="midnight")

        err_filter = logging.Filter()
        err_filter.filter = lambda record: record.levelno > logging.WARNING
        err_fh.setLevel(logging.ERROR)
        err_fh.addFilter(err_filter)

        # 第三步，定义handler的输出格式
        formatter = logging.Formatter("%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s: %(message)s")
        info_fh.setFormatter(formatter)
        err_fh.setFormatter(formatter)
        # 第四步，将logger添加到handler里面
        self.logger.addHandler(info_fh)
        self.logger.addHandler(err_fh)


logger = SingleLogger().logger