Django框架中间件(Middleware)用法实例分析

下面是详细讲解“Django框架中间件(Middleware)用法实例分析”的完整攻略。

一、中间件概述

Django框架中间件(Middleware)，指的是处于请求和响应处理之间的一层拦截层。中间件是在Django处理请求的过程中运行的，而非在应用程序中直接运行。

在Django中，中间件可以用于拦截请求或响应，处理请求或响应，以及更改请求或响应。它增加了框架的灵活性，使得我们能够对请求和响应进行更加细致的控制和定制。

二、中间件使用

中间件在settings.py文件中进行配置，将中间件类的路径添加到MIDDLEWARE列表中。

例如，要在Django应用程序中添加中间件“my_middleware.middleware.MyMiddleware”，可以在settings.py文件中添加以下代码：

MIDDLEWARE = [
    # ...
    'my_middleware.middleware.MyMiddleware',
    # ...
]

三、中间件实例说明

下面介绍两个中间件实例：

1. 统计请求时间中间件

这个中间件用于统计请求的处理时间，计算每个请求的处理时间，并将处理结果输出到控制台。

具体代码实现如下：

import time

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

    def __call__(self, request):
        start_time = time.time()  # 程序开始时间
        response = self.get_response(request)
        end_time = time.time()  # 程序结束时间

        print("Request to {} took {:.2f} seconds.".format(request.path, end_time - start_time))

        return response

可以看到，这个中间件实现了一个__call__方法，该方法接收request对象，计算请求处理的时间，并打印到控制台上。

2. JWT中间件

这个中间件用于验证用户的JWT Token，在视图函数被调用之前需要进行验证，如果Token不合法，需要返回一个401 Unauthorized响应。

JWT Token是一个基于JSON的开放标准，用于将声明性数据存储在JSON格式中。JWT的一个优点是它可以用于身份验证，因为它可以包含用户的身份信息。

具体代码实现如下：

import jwt
from django.conf import settings
from django.http import JsonResponse

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

    def __call__(self, request):
        if not request.META.get('HTTP_AUTHORIZATION'):
            # 请求头Authorization不存在
            return JsonResponse({'error': 'Authorization header not found.'}, status=401)

        # 获取Token
        token = request.META.get('HTTP_AUTHORIZATION').split(' ')[1]

        try:
            # 解码Token
            payload = jwt.decode(token, settings.JWT_SECRET_KEY, algorithms=[settings.JWT_ALGORITHM])
            # 将Token中的用户信息如user_id保存到request中
            request.user_id = payload['user_id']
        except:
            # Token解码失败，返回401 Unauthorized响应
            return JsonResponse({'error': 'Invalid Token.'}, status=401)

        response = self.get_response(request)
        return response

这个中间件实现了一个__call__方法，该方法接收request对象，检查请求头中是否存在Authorization，并从中获取Token。对Token进行解码，获取用户信息，并将用户ID信息存储在request对象中。

如果Token解码失败，则返回401 Unauthorized响应，如果Token解码成功，则调用后续的视图函数。