如何缓存 Django Rest Framework API 调用？

Question

我使用 Memcached 作为我的 django 应用程序的后端。此代码在正常的 django 查询中运行良好：

def get_myobj():
        cache_key = 'mykey'
        result = cache.get(cache_key, None)
        if not result:
            result = Product.objects.all().filter(draft=False)
            cache.set(cache_key, result)
        return result

但是与 django-rest-framework api 调用一起使用时它不起作用：

class ProductListAPIView(generics.ListAPIView):
    def get_queryset(self):
        product_list = Product.objects.all()
        return product_list
    serializer_class = ProductSerializer

我即将尝试提供缓存功能的 DRF 扩展：

https://github.com/chibisov/drf-extensions

但 github 上的构建状态当前显示“构建失败”。

我的应用程序对 api 调用的读取量很大。有没有办法缓存这些调用？

谢谢你。

Answer 1

好的，所以，为了对您的查询集使用缓存：

class ProductListAPIView(generics.ListAPIView):
    def get_queryset(self):
        return get_myobj()
    serializer_class = ProductSerializer

您可能希望在缓存集上设置超时（例如 60 秒）：

cache.set(cache_key, result, 60)

如果你想缓存整个视图：

from django.utils.decorators import method_decorator
from django.views.decorators.cache import cache_page

class ProductListAPIView(generics.ListAPIView):
    serializer_class = ProductSerializer

    @method_decorator(cache_page(60))
    def dispatch(self, *args, **kwargs):
        return super(ProductListAPIView, self).dispatch(*args, **kwargs)

Answer 2

我刚刚实现了这个以在我的序列化器上使用

def cache_me(cache):
    def true_decorator(f):
        @wraps(f)
        def wrapper(*args, **kwargs):
            instance = args[1]
            cache_key = '%s.%s' % (instance.facility, instance.id)
            logger.debug('%s cache_key: %s' % (cache, cache_key))
            try:
                data = caches[cache].get(cache_key)
                if data is not None:
                    return data
            except:
                pass
            logger.info('did not cache')
            data = f(*args, **kwargs)
            try:
                caches[cache].set(cache_key, data)
            except:
                pass
            return data
        return wrapper
    return true_decorator

然后我重写序列化器上的 to_representation 方法，以便它缓存每个实例的序列化输出。

class MyModelSerializer(serializers.ModelSerializer):

    class Meta:
        model = MyModel
        exclude = ('is_deleted', 'facility',)

    @cache_me('mymodel')
    def to_representation(self, instance):
       return super(MyModelSerializer, self).to_representation(instance)

Answer 3

下面的代码使用调度工作正常。

 @method_decorator(cache_page(60 * 60 * 2), 'dispatch')
 @method_decorator(vary_on_cookie, 'dispatch')
 class SampleViewSet(viewsets.ModelViewSet):
    queryset         = Sample.objects.all()
    serializer_class = SampleSerializer

    def get_queryset(self):
        queryset = Sample.objects.filter(is_active=True)
        return queryset