mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
20 lines
601 B
Python
20 lines
601 B
Python
from redis import Redis
|
|
from sqlalchemy import orm
|
|
|
|
|
|
async def db_count_cache(redis: Redis, key: str, query: orm.Query,
|
|
expire: int = None) -> int:
|
|
""" Store and retrieve a query.count() via redis cache.
|
|
|
|
:param redis: Redis handle
|
|
:param key: Redis key
|
|
:param query: SQLAlchemy ORM query
|
|
:param expire: Optional expiration in seconds
|
|
:return: query.count()
|
|
"""
|
|
result = redis.get(key)
|
|
if result is None:
|
|
redis.set(key, (result := int(query.count())))
|
|
if expire:
|
|
redis.expire(key, expire)
|
|
return int(result)
|