实现

在app里新建一个performance.py文件,用来定义我们的中间件函数,这里实现的函数功能为记录页面的访问时间。

import time
import logging

logger = logging.getLogger(__name__)


def performance_logger_middleware(get_response):
    def middleware(request):
        # 计算响应耗时
        start_time = time.time()
        response = get_response(request)
        duration = time.time() - start_time
        # 在response header中写入页面访问耗时的属性
        response["X-Page-Duration-ms"] = int(duration * 1000)
        # 载入日志
        logger.info("%s %s %s", duration, request.path, request.GET.dict() )
        return response

    return middleware

随后在django项目设置文件里面注册中间件,我的performance.py文件是创建在interviewapp里面的,所以我引入的链接为interview.performance.performance_logger_middleware

注册中间件

测试

这里我已经设置了将日志打印至命令行以及写入日志文件,具体配置请查看django文档中日志部分。

命令行打印出响应时间日志

日志文件

响应头中也有我们写入的信息

至此,一个简单的中间件就实现了,同样的,也可以使用类的方法类定义一个中间件(可以参考django中间件的源码,都是基于类的方法类定义中间件的),本文例子中使用函数的方法定义只是为了方便。

最后修改:2022 年 01 月 25 日
如果觉得我的文章对你有用,请随意赞赏