DRF使用缓存

DRF使用缓存

在做项目的时候，有时很多数据都是经常被访问的而且数据是基本不变的，比如省市区的数据是经常被用户查询使用的，而且数据基本不变化，所以我们可以将省市区数据进行缓存处理，减少数据库的查询次数，这里将以缓存省市区数据为例，简单介绍一下在使用drf开发前后端分离项目的时候缓存的使用。

在Django REST framework中使用缓存，可以通过drf-extensions扩展来实现。

关于扩展使用缓存的文档，可参考链接http://chibisov.github.io/drf-extensions/docs/#caching

(1) 安装

pip install drf-extensions

(2) 使用方法

1） 直接添加装饰器

可以在使用rest_framework_extensions.cache.decorators中的cache_response装饰器来装饰返回数据的类视图的对象方法，如

class CityView(views.APIView):
    @cache_response()
    def get(self, request, *args, **kwargs):
        ...

cache_response装饰器可以接收两个参数

@cache_response(timeout=60*60, cache='default')

• timeout 缓存时间
• cache 缓存使用的Django缓存后端（即CACHES配置中的键名称）

如果在使用cache_response装饰器时未指明timeout或者cache参数，则会使用配置文件中的默认配置，可以通过如下方法指明：

# DRF扩展
REST_FRAMEWORK_EXTENSIONS = {
    # 缓存时间
    'DEFAULT_CACHE_RESPONSE_TIMEOUT': 60 * 60,
    # 缓存存储
    'DEFAULT_USE_CACHE': 'default',
}

• DEFAULT_CACHE_RESPONSE_TIMEOUT 缓存有效期，单位秒
• DEFAULT_USE_CACHE 缓存的存储方式，与配置文件中的CACHES的键对应。

注意，cache_response装饰器既可以装饰在类视图中的get方法上，也可以装饰在REST framework扩展类提供的list或retrieve方法上。使用cache_response装饰器无需使用method_decorator进行转换。

2）使用drf-extensions提供的扩展类

drf-extensions扩展对于缓存提供了三个扩展类：

• ListCacheResponseMixin

  用于缓存返回列表数据的视图，与ListModelMixin扩展类配合使用，实际是为list方法添加了cache_response装饰器

• RetrieveCacheResponseMixin

  用于缓存返回单一数据的视图，与RetrieveModelMixin扩展类配合使用，实际是为retrieve方法添加了cache_response装饰器

• CacheResponseMixin

  为视图集同时补充List和Retrieve两种缓存，与ListModelMixin和RetrieveModelMixin一起配合使用。

三个扩展类都是在rest_framework_extensions.cache.mixins中。

(3) 为省市区视图添加缓存

由于视图使用了ListAPIView和RetrieveAPIView分别返回多条和单一数据，所以使用ListCacheResponseMixin和RetrieveCacheResponseMixin类来装饰视图类

修改返回省市区信息的视图

class AreaListView(ListCacheResponseMixin, ListAPIView):
    """查询所有省"""
    queryset = Area.objects.filter(parent=None)
    serializer_class = AreaSerializer


class AreaDetailView(RetrieveCacheResponseMixin, RetrieveAPIView):
    """查询单一省或市"""
    serializer_class = SubAreaSerializer
    queryset = Area.objects.all()

(4)缓存数据保存位置与有效期的设置

我们想把缓存数据保存在redis中，且设置有效期，可以通过在配置文件中定义的方式来实现。

在配置文件中增加如下配置：

# DRF扩展
REST_FRAMEWORK_EXTENSIONS = {
    # 缓存时间
    'DEFAULT_CACHE_RESPONSE_TIMEOUT': 60 * 60,
    # 缓存存储
    'DEFAULT_USE_CACHE': 'default',
}