2020-08-26 18:04:57 +08:00
|
|
|
from typing import Tuple
|
|
|
|
|
|
|
|
|
|
from aiomcache import Client
|
|
|
|
|
|
|
|
|
|
from fastapi_cache.backends import Backend
|
|
|
|
|
|
|
|
|
|
|
2020-11-09 20:32:23 +08:00
|
|
|
class MemcachedBackend(Backend):
|
2020-08-26 18:04:57 +08:00
|
|
|
def __init__(self, mcache: Client):
|
|
|
|
|
self.mcache = mcache
|
|
|
|
|
|
|
|
|
|
async def get_with_ttl(self, key: str) -> Tuple[int, str]:
|
2020-11-10 10:34:52 +08:00
|
|
|
return 3600, await self.mcache.get(key.encode())
|
2020-08-26 18:04:57 +08:00
|
|
|
|
|
|
|
|
async def get(self, key: str):
|
|
|
|
|
return await self.mcache.get(key, key.encode())
|
|
|
|
|
|
|
|
|
|
async def set(self, key: str, value: str, expire: int = None):
|
2021-03-20 14:42:29 +08:00
|
|
|
return await self.mcache.set(key.encode(), value.encode(), exptime=expire or 0)
|
2020-11-03 18:08:06 +08:00
|
|
|
|
|
|
|
|
async def clear(self, namespace: str = None, key: str = None):
|
|
|
|
|
raise NotImplementedError
|