trans_log.py 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. # -*- coding: utf-8 -*-
  2. # @Time : 2024/5/16
  3. # @Author : 魏志亮
  4. import datetime
  5. import logging
  6. import os
  7. import sys
  8. # 创建一个ThreadLocal对象来存储链路ID
  9. def set_trance_id(trace_id):
  10. """设置当前线程的链路ID"""
  11. os.environ['trace_id'] = trace_id
  12. class ContextFilter(logging.Filter):
  13. """一个自定义的日志过滤器,用于在日志记录中添加链路ID"""
  14. def filter(self, record):
  15. record.trace_id = ''
  16. if 'trace_id' in os.environ.keys():
  17. record.trace_id = os.environ['trace_id']
  18. return True
  19. logger = logging.getLogger(__name__)
  20. logger.setLevel(logging.INFO)
  21. stout_handle = logging.StreamHandler(sys.stdout)
  22. stout_handle.setFormatter(
  23. logging.Formatter("%(asctime)s-%(trace_id)s-%(levelname)s-%(filename)-8s:%(lineno)s: %(message)s"))
  24. stout_handle.setLevel(logging.INFO)
  25. stout_handle.addFilter(ContextFilter())
  26. logger.addHandler(stout_handle)
  27. log_path = r'/data/logs/trans_data'
  28. file_path = os.path.join(log_path)
  29. if not os.path.exists(file_path):
  30. os.makedirs(file_path, exist_ok=True)
  31. file_name = file_path + os.sep + str(datetime.date.today()) + '.log'
  32. file_handler = logging.FileHandler(file_name, encoding='utf-8')
  33. file_handler.setFormatter(
  34. logging.Formatter("%(asctime)s-%(trace_id)s-%(levelname)s-%(filename)-8s:%(lineno)s: %(message)s"))
  35. file_handler.setLevel(logging.INFO)
  36. file_handler.addFilter(ContextFilter())
  37. logger.addHandler(file_handler)
  38. def trans_print(*args):
  39. logger.info(" ".join([str(a) for a in args]))