GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-09-05 23:47:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix redis key

Browse Source
This commit is contained in:
jxxghp
2025-08-23 15:40:38 +08:00
parent 354165aa0a
commit 3b04506893
2 changed files with 59 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/helper/redis.py
+32 -20
View File
@@ -140,20 +140,32 @@ class RedisHelper(metaclass=Singleton):
logger.error(f"Failed to set Redis maxmemory or policy: {e}")
@staticmethod
def get_region(region: Optional[str] = "DEFAULT"):
def __get_region(region: Optional[str] = "DEFAULT"):
"""
获取缓存的区
"""
return f"region:{region}" if region else "region:default"
def get_redis_key(self, region: str, key: str) -> str:
def __make_redis_key(self, region: str, key: str) -> str:
"""
获取缓存Key
"""
# 使用region作为缓存键的一部分
region = self.get_region(quote(region))
region = self.__get_region(quote(region))
return f"{region}:key:{quote(key)}"
@staticmethod
def __get_original_key(redis_key: str) -> str:
"""
从Redis键中提取原始key
"""
try:
parts = redis_key.split(":key:")
return parts[-1]
except Exception as e:
logger.warn(f"Failed to parse redis key: {redis_key}, error: {e}")
return redis_key
def set(self, key: str, value: Any, ttl: Optional[int] = None,
region: Optional[str] = "DEFAULT", **kwargs) -> None:
"""
@@ -167,7 +179,7 @@ class RedisHelper(metaclass=Singleton):
"""
try:
self._connect()
redis_key = self.get_redis_key(region, key)
redis_key = self.__make_redis_key(region, key)
# 对值进行序列化
serialized_value = serialize(value)
kwargs.pop("maxsize", None)
@@ -185,7 +197,7 @@ class RedisHelper(metaclass=Singleton):
"""
try:
self._connect()
redis_key = self.get_redis_key(region, key)
redis_key = self.__make_redis_key(region, key)
return self.client.exists(redis_key) == 1
except Exception as e:
logger.error(f"Failed to exists key: {key} region: {region}, error: {e}")
@@ -201,7 +213,7 @@ class RedisHelper(metaclass=Singleton):
"""
try:
self._connect()
redis_key = self.get_redis_key(region, key)
redis_key = self.__make_redis_key(region, key)
value = self.client.get(redis_key)
if value is not None:
return deserialize(value)
@@ -219,7 +231,7 @@ class RedisHelper(metaclass=Singleton):
"""
try:
self._connect()
redis_key = self.get_redis_key(region, key)
redis_key = self.__make_redis_key(region, key)
self.client.delete(redis_key)
except Exception as e:
logger.error(f"Failed to delete key: {key} in region: {region}, error: {e}")
@@ -233,7 +245,7 @@ class RedisHelper(metaclass=Singleton):
try:
self._connect()
if region:
cache_region = self.get_region(quote(region))
cache_region = self.__get_region(quote(region))
redis_key = f"{cache_region}:key:*"
with self.client.pipeline() as pipe:
for key in self.client.scan_iter(redis_key):
@@ -256,17 +268,17 @@ class RedisHelper(metaclass=Singleton):
try:
self._connect()
if region:
cache_region = self.get_region(quote(region))
cache_region = self.__get_region(quote(region))
redis_key = f"{cache_region}:key:*"
for key in self.client.scan_iter(redis_key):
value = self.client.get(key)
if value is not None:
yield key, deserialize(value)
yield self.__get_original_key(key), deserialize(value)
else:
for key in self.client.scan_iter("*"):
value = self.client.get(key)
if value is not None:
yield key, deserialize(value)
yield self.__get_original_key(key), deserialize(value)
except Exception as e:
logger.error(f"Failed to get items from Redis, region: {region}, error: {e}")
@@ -367,18 +379,18 @@ class AsyncRedisHelper(metaclass=Singleton):
logger.error(f"Failed to set Redis maxmemory or policy (async): {e}")
@staticmethod
def get_region(region: Optional[str] = "DEFAULT"):
def __get_region(region: Optional[str] = "DEFAULT"):
"""
获取缓存的区
"""
return f"region:{region}" if region else "region:default"
def get_redis_key(self, region: str, key: str) -> str:
def __make_redis_key(self, region: str, key: str) -> str:
"""
获取缓存Key
"""
# 使用region作为缓存键的一部分
region = self.get_region(quote(region))
region = self.__get_region(quote(region))
return f"{region}:key:{quote(key)}"
async def set(self, key: str, value: Any, ttl: Optional[int] = None,
@@ -394,7 +406,7 @@ class AsyncRedisHelper(metaclass=Singleton):
"""
try:
await self._connect()
redis_key = self.get_redis_key(region, key)
redis_key = self.__make_redis_key(region, key)
# 对值进行序列化
serialized_value = serialize(value)
kwargs.pop("maxsize", None)
@@ -412,7 +424,7 @@ class AsyncRedisHelper(metaclass=Singleton):
"""
try:
await self._connect()
redis_key = self.get_redis_key(region, key)
redis_key = self.__make_redis_key(region, key)
result = await self.client.exists(redis_key)
return result == 1
except Exception as e:
@@ -429,7 +441,7 @@ class AsyncRedisHelper(metaclass=Singleton):
"""
try:
await self._connect()
redis_key = self.get_redis_key(region, key)
redis_key = self.__make_redis_key(region, key)
value = await self.client.get(redis_key)
if value is not None:
return deserialize(value)
@@ -447,7 +459,7 @@ class AsyncRedisHelper(metaclass=Singleton):
"""
try:
await self._connect()
redis_key = self.get_redis_key(region, key)
redis_key = self.__make_redis_key(region, key)
await self.client.delete(redis_key)
except Exception as e:
logger.error(f"Failed to delete key (async): {key} in region: {region}, error: {e}")
@@ -461,7 +473,7 @@ class AsyncRedisHelper(metaclass=Singleton):
try:
await self._connect()
if region:
cache_region = self.get_region(quote(region))
cache_region = self.__get_region(quote(region))
redis_key = f"{cache_region}:key:*"
async with self.client.pipeline() as pipe:
async for key in self.client.scan_iter(redis_key):
@@ -484,7 +496,7 @@ class AsyncRedisHelper(metaclass=Singleton):
try:
await self._connect()
if region:
cache_region = self.get_region(quote(region))
cache_region = self.__get_region(quote(region))
redis_key = f"{cache_region}:key:*"
async for key in self.client.scan_iter(redis_key):
value = await self.client.get(key)