mirror of
https://github.com/long2ice/fastapi-cache.git
synced 2026-03-25 04:57:54 +00:00
30 lines
917 B
Python
30 lines
917 B
Python
from typing import Optional, Tuple
|
|
|
|
from aiomcache import Client
|
|
|
|
from fastapi_cache.types import Backend
|
|
|
|
|
|
class MemcachedBackend(Backend):
|
|
def __init__(self, mcache: Client):
|
|
self.mcache = mcache
|
|
|
|
async def get_with_ttl(self, key: str) -> Tuple[int, Optional[bytes]]:
|
|
return 3600, await self.get(key)
|
|
|
|
async def get(self, key: str) -> Optional[bytes]:
|
|
return await self.mcache.get(key.encode())
|
|
|
|
async def set(self, key: str, value: bytes, expire: Optional[int] = None) -> None:
|
|
await self.mcache.set(key.encode(), value, exptime=expire or 0)
|
|
|
|
async def clear(self, namespace: Optional[str] = None, key: Optional[str] = None) -> int:
|
|
is_deleted = False
|
|
if key:
|
|
is_deleted = await self.mcache.delete(key=key.encode())
|
|
else:
|
|
await self.mcache.flush_all()
|
|
is_deleted = True
|
|
|
|
return int(is_deleted)
|