Как регистрировать идентификатор запроса в django middleware?

Я написал код, как показано ниже, но он вызывает исключение

здесь request_id_middleware.py

import uuid

from django.middleware.common import CommonMiddleware
from taiji.logger import logger

class RequestIDMiddleware(CommonMiddleware):
    # pass
    def process_request(self, request):
        request.META['request_id'] = str(uuid.uuid4())
        logger.info(f"start request id: {request.META['request_id']}")
        return request

    def process_response(self, request, response):
        if request.META.get('request_id') is None:
            response.headers['X-REQUEST-ID'] = request.META['request_id']
            logger.info(f"finish request id: {response.headers['X-REQUEST-ID']}")
        return response

logger.py

import logging


def set_logger(name):
    logger=logging.getLogger(name)
    handler=logging.StreamHandler()
    handler.setLevel(logging.DEBUG)
    fmt=logging.Formatter(fmt='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    handler.setFormatter(fmt)
    logger.addHandler(handler)
    return logger

logger=set_logger('gg')

Просмотров

def index(request: HttpRequest):
    logger.info("hello world")
    return JsonResponse("hello world")

но он говорит мне

Traceback (most recent call last):
  File "E:\miniconda\envs\py312\Lib\wsgiref\handlers.py", line 137, in run
    self.result = application(self.environ, self.start_response)
                  ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\miniconda\envs\py312\Lib\site-packages\django\contrib\staticfiles\handlers.py", line 80, in __call__
    return self.application(environ, start_response)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\miniconda\envs\py312\Lib\site-packages\django\core\handlers\wsgi.py", line 124, in __call__
    response = self.get_response(request)
               ^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\miniconda\envs\py312\Lib\site-packages\django\core\handlers\base.py", line 141, in get_response
    response._resource_closers.append(request.close)
    ^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'WSGIRequest' object has no attribute '_resource_closers'
[20/Apr/2024 22:32:59] "GET /index/ HTTP/1.1" 500 59

правильно ли реализовано промежуточное ПО? почему он вызывает ошибку WSGIRequest? Это должна быть ошибка respone-like object Любая помощь будет признательна

Похоже, что проблема связана с вашим методом process_response в промежуточном ПО. Вы пытаетесь получить доступ к response.headers, но объекты ответа Django не имеют атрибута headers. Вместо этого вы должны установить заголовок непосредственно в объекте ответа. Попробуйте обновить ваше промежуточное ПО следующим образом:

import uuid
from django.http import JsonResponse

class RequestIDMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        request.META['request_id'] = str(uuid.uuid4())
        response = self.get_response(request)
        response['X-REQUEST-ID'] = request.META['request_id']
        return response

После этой модификации промежуточное ПО должно устанавливать заголовок X-REQUEST-ID в каждом ответе. Просто убедитесь, что вы соответствующим образом обновили регистрацию промежуточного ПО в настройках MIDDLEWARE вашего проекта Django.

Вернуться на верх