coze_znjd/commom/traceId_util.py

# -*- coding:utf-8 -*-
"""
@author: isaacqyang
@time: 2023/5/16
@desc: 
"""

import logging
from contextvars import ContextVar

request_id_context = ContextVar('request_id')


class TraceIdFilter(logging.Filter):

    def filter(self, record):
        record.requestId = request_id_context.get()
        return True