python - DRF 扩展缓存忽略查询参数

Question

我正在使用 drf-extension 来缓存我的 API。但它与 cache_response 装饰器没有按预期工作。

它缓存 say 的响应/api/get-cities/?country=india。但是当我点击时 /api/get-cities/?country=usa，我得到了相同的响应。

这是示例代码：

设置.py

CACHES = {
   "default": {
       "BACKEND": "django_redis.cache.RedisCache",
       "LOCATION": "redis://127.0.0.1:6379/0",
       "OPTIONS": {
           "CLIENT_CLASS": "django_redis.client.DefaultClient"
       },
       "KEY_PREFIX": "city"
   }
}

REST_FRAMEWORK_EXTENSIONS = {
   'DEFAULT_USE_CACHE': 'default',
   'DEFAULT_CACHE_RESPONSE_TIMEOUT': 86400,
}

视图.py

class GetCities(APIView):

    @cache_response()
    def get(self, request):
        country = request.GET.get("country", "")
        return get_cities_function(country)

请帮助解决这个问题。

Answer 1

我能够找到解决问题的方法。我使用 api 名称和参数名称（在我的例子中是国家/地区）的组合在 redis 中创建了自己的键。因此，当 API 被查询参数命中时，我会检查是否存在对应的键，如果存在则返回缓存的响应。

class GetCities(APIView):

    def calculate_cache_key(self, view_instance, view_method, request, args, kwargs):
        api = view_instance.get_view_name().replace(' ', '')
        return "api:" + api + "country:" + str(request.GET.get("country", ""))

    @cache_response(key_func='calculate_cache_key')
    def get(self, request):
        country = request.GET.get("country", "")
        return get_cities_function(country)