drf實現(xiàn)常用數(shù)據(jù)緩存

ghnor 發(fā)布于2019-07-31 11:22 / 3416人閱讀

摘要：使用擴展類使用了視圖集用于緩存返回列表數(shù)據(jù)的視圖，與擴展類配合使用，實際是為方法添加了裝飾器用于緩存返回單一數(shù)據(jù)的視圖，與擴展類配合使用，實際是為方法添加了裝飾器為視圖集同時補充和兩種緩存，與和一起配合使用。

在以往的后臺數(shù)據(jù)訪問時，我們往往都會進行數(shù)據(jù)庫查詢，基本的流程是這樣的：

圖中發(fā)生了三次請求，則很正常向數(shù)據(jù)庫查詢了三次。但是現(xiàn)在有這樣一個場景：我們有1000個人在一個十分鐘內(nèi)向一個我們網(wǎng)站都看了同一個文章，那么我們有沒有哪些可以優(yōu)化我們的后端代碼，因為這只是一篇文章在短時間內(nèi)就被訪問了1000次，當(dāng)然我們網(wǎng)站的文章是海量的，那我們該怎么辦？那么我們的數(shù)據(jù)緩存就派上用場了，基本的流程是這樣的：

流程

在第一次請求的時候查看緩存中(redis)是否有數(shù)據(jù)，有數(shù)據(jù)則直接返回響應(yīng)

若redis中沒有數(shù)據(jù)，則查詢數(shù)據(jù)庫

查詢數(shù)據(jù)庫并將數(shù)據(jù)保存到redis中，返回響應(yīng)

這就是我們?yōu)槭裁粗徊樵兞艘淮螖?shù)據(jù)庫，若有1000次，我們在緩存時間內(nèi)也只需要查詢一次數(shù)據(jù)庫，這里向redis中獲取數(shù)據(jù)也需要耗時，但是由于redis數(shù)據(jù)存儲在內(nèi)存中，數(shù)據(jù)獲取性能較數(shù)據(jù)庫高了不止一點半點。
那么在drf項目中如何去實現(xiàn)呢？只需要簡單的三步

安裝

pip install drf-extensions

配置(可以省略)

# DRF擴展
REST_FRAMEWORK_EXTENSIONS = {
    # 緩存時間
    "DEFAULT_CACHE_RESPONSE_TIMEOUT": 60 * 60,
}

使用

使用cache_response裝飾器

from rest_framework.response import Response
from rest_framework import views
from rest_framework_extensions.cache.decorators import (
    cache_response
)
from myapp.models import City

class CityView(views.APIView):
    @cache_response()
    def get(self, request, *args, **kwargs):
        cities = City.objects.all().values_list("name", flat=True)
        return Response(cities)

注意，cache_response裝飾器既可以裝飾在類視圖中的get方法上，也可以裝飾在REST framework擴展類提供的list或retrieve方法上。使用cache_response裝飾器無需使用method_decorator進行轉(zhuǎn)換。

使用擴展類(使用了視圖集ViewSet)

ListCacheResponseMixin:用于緩存返回列表數(shù)據(jù)的視圖，與ListModelMixin擴展類配合使用，實際是為list方法添加了cache_response裝飾器
RetrieveCacheResponseMixin:用于緩存返回單一數(shù)據(jù)的視圖，與RetrieveModelMixin擴展類配合使用，實際是為retrieve方法添加了cache_response裝飾器
CacheResponseMixin:為視圖集同時補充List和Retrieve兩種緩存，與ListModelMixin和RetrieveModelMixin一起配合使用。

from myapps.serializers import UserSerializer
from rest_framework_extensions.cache.mixins import CacheResponseMixin

class UserViewSet(CacheResponseMixin, viewsets.ModelViewSet):#繼承順序一定在ViewSet前,其實必須在對應(yīng)的mixin前
    serializer_class = UserSerializer

def-extensions官方文檔：http://chibisov.github.io/drf...