flibusta-apps/book_library_server
1
0
Fork 0
mirror of https://github.com/flibusta-apps/book_library_server.git synced 2025-12-06 07:05:36 +01:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add no_cache param

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-01-31 11:47:53 +01:00
parent 70682ee050
commit b3b3852ff3
2 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
fastapi_book_server/app/filters/book.py
View File

@@ -11,6 +11,7 @@ def get_book_filter(
allowed_langs: Optional[list[str]] = Query(None), # type: ignore allowed_langs: Optional[list[str]] = Query(None), # type: ignore
uploaded_gte: Optional[date] = None, uploaded_gte: Optional[date] = None,
uploaded_lte: Optional[date] = None, uploaded_lte: Optional[date] = None,
no_cache: bool = False,
) -> dict: ) -> dict:
result = {} result = {}
@@ -26,4 +27,6 @@ def get_book_filter(
if uploaded_lte: if uploaded_lte:
result["uploaded__lte"] = uploaded_lte result["uploaded__lte"] = uploaded_lte
result["no_cache"] = no_cache
return result return result

16
fastapi_book_server/app/services/common.py
View File

@@ -128,28 +128,28 @@ class BaseSearchService(Generic[MODEL, QUERY], BaseService[MODEL, QUERY]):
@classmethod @classmethod
async def get_object_ids( async def get_object_ids(
cls, query: QUERY, redis: aioredis.Redis cls, query: QUERY, redis: aioredis.Redis, no_cache: bool
) -> tuple[int, list[int]]: ) -> tuple[int, list[int]]:
params = cls.get_raw_params() params = cls.get_raw_params()
if ( if not no_cache and (
cached_object_ids := await cls.get_cached_ids(query, redis, params) cached_object_ids := await cls.get_cached_ids(query, redis, params)
) is not None: ):
return cached_object_ids return cached_object_ids
object_ids = await cls._get_object_ids(query) object_ids = await cls._get_object_ids(query)
limited_object_ids = object_ids[params.offset : params.offset + params.limit] limited_object_ids = object_ids[params.offset : params.offset + params.limit]
if len(object_ids) != 0: if not no_cache and len(object_ids) != 0:
await cls.cache_object_ids(query, object_ids, redis) await cls.cache_object_ids(query, object_ids, redis)
return len(object_ids), limited_object_ids return len(object_ids), limited_object_ids
@classmethod @classmethod
async def get_limited_objects( async def get_limited_objects(
cls, query: QUERY, redis: aioredis.Redis cls, query: QUERY, redis: aioredis.Redis, no_cache: bool
) -> tuple[int, list[MODEL]]: ) -> tuple[int, list[MODEL]]:
count, object_ids = await cls.get_object_ids(query, redis) count, object_ids = await cls.get_object_ids(query, redis, no_cache)
queryset: QuerySet[MODEL] = cls.model.objects queryset: QuerySet[MODEL] = cls.model.objects
@@ -164,9 +164,11 @@ class BaseSearchService(Generic[MODEL, QUERY], BaseService[MODEL, QUERY]):
@classmethod @classmethod
async def get(cls, query: QUERY, redis: aioredis.Redis) -> Page[MODEL]: async def get(cls, query: QUERY, redis: aioredis.Redis) -> Page[MODEL]:
no_cache: bool = query.get("no_cache", False) # type: ignore
params = cls.get_params() params = cls.get_params()
total, objects = await cls.get_limited_objects(query, redis) total, objects = await cls.get_limited_objects(query, redis, no_cache)
return CustomPage.create(items=objects, total=total, params=params) return CustomPage.create(items=objects, total=total, params=params)