diff --git a/app/utils/api_secret_utils.py b/app/utils/api_secret_utils.py index 105bda8..996a67e 100644 --- a/app/utils/api_secret_utils.py +++ b/app/utils/api_secret_utils.py @@ -1,5 +1,7 @@ from enum import Enum +from app.utils.redis_utils import redis_utils, remove_redis_cache + class ApiSecretStatus(Enum): valid = "valid" # 秘钥有效 @@ -13,22 +15,23 @@ class ApiSecretUtils(): # 验证秘钥状态 async def verify_secret(self, secret: str) -> ApiSecretStatus: - api_secret_status = self.cache.get(secret, None) - print("verify_secret", self.cache) + async with redis_utils.get_redis_connection() as redis_client: + cache_value = await redis_client.get(f"api_secret_status_${secret}") + print("cache_value", cache_value) + + api_secret_status = cache_value + print("verify_secret", api_secret_status, secret) return api_secret_status or ApiSecretStatus.not_exist # 保存秘钥状态 async def save_secret(self, secret: str, status: ApiSecretStatus): - print("save_secret", self.cache) - self.cache[secret] = status + print("save_secret", secret) + async with redis_utils.get_redis_connection() as redis_client: + await redis_client.set(f"api_secret_status_${secret}", str(status)) # 移除秘钥缓存 async def remove_secret(self, secret: str): - # self.cache.pop(secret, None) - if secret in self.cache: - del self.cache[secret] - print("remove_secret", secret) - print(await self.verify_secret(secret)) + await remove_redis_cache(f"api_secret_status_${secret}") api_secret_utils = ApiSecretUtils()