Django REST framework - 缓存

一位女性有着非常敏锐的意识，但几乎没有记忆力……她的记忆力足以工作，而且她工作很努力。
——莉迪亚·戴维斯

REST Framework 中的缓存与 Django 提供的缓存工具配合得很好。

在 APIView 和 ViewSet 中使用缓存

Django 提供了一个 method_decorator，用于在基于类的视图中使用装饰器。这可以与其他缓存装饰器一起使用，如 cache_page、vary_on_cookie 和 vary_on_headers。

from django.utils.decorators import method_decorator
from django.views.decorators.cache import cache_page
from django.views.decorators.vary import vary_on_cookie, vary_on_headers

from rest_framework.response import Response
from rest_framework.views import APIView
from rest_framework import viewsets

class UserViewSet(viewsets.ViewSet):
    # 带 cookie：为每个用户缓存请求的 URL 2 小时
    @method_decorator(cache_page(60 * 60 * 2))
    @method_decorator(vary_on_cookie)
    def list(self, request, format=None):
        content = {
            "user_feed": request.user.get_user_feed(),
        }
        return Response(content)

class ProfileView(APIView):
    # 带身份验证：为每个用户缓存请求的 URL 2 小时
    @method_decorator(cache_page(60 * 60 * 2))
    @method_decorator(vary_on_headers("Authorization"))
    def get(self, request, format=None):
        content = {
            "user_feed": request.user.get_user_feed(),
        }
        return Response(content)

class PostView(APIView):
    # 缓存请求的 URL 页面
    @method_decorator(cache_page(60 * 60 * 2))
    def get(self, request, format=None):
        content = {
            "title": "文章标题",
            "body": "文章内容",
        }
        return Response(content)

使用 @api_view 装饰器时的缓存

当使用 @api_view 装饰器时，可以直接调用 Django 提供的基于方法的缓存装饰器，如 cache_page、vary_on_cookie 和 vary_on_headers。

from django.views.decorators.cache import cache_page
from django.views.decorators.vary import vary_on_cookie

from rest_framework.decorators import api_view
from rest_framework.response import Response

@cache_page(60 * 15)
@vary_on_cookie
@api_view(["GET"])
def get_user_list(request):
    content = {"user_feed": request.user.get_user_feed()}
    return Response(content)

注意： cache_page 装饰器仅缓存状态为 200 的 GET 和 HEAD 响应。