bugfix(test): restore support for Redis cluster testing

- Extend ruff.toml to include tests directory
- Update _catches.py to support Redis cluster nodes
- Add new cluster policies for different caching strategies
- Implement cluster-specific Redis function caches

Author: tanbro

.ruff.toml

@@ -1,4 +1,4 @@
-src = ["src"]
+src = ["src", "tests"]
 extend-exclude = ["docs"]
 line-length = 120
 

tests/_catches.py

@@ -1,11 +1,12 @@
 import asyncio
 import atexit
 from os import getenv
-from typing import Optional
+from typing import Callable, Dict, List, Optional
 from warnings import warn
 
 from redis import Redis
 from redis.asyncio import Redis as AsyncRedis
+from redis.cluster import ClusterNode, RedisCluster
 
 from redis_func_cache import (
     FifoClusterMultiplePolicy,
@@ -22,11 +23,11 @@
     RrClusterMultiplePolicy,
     RrPolicy,
 )
-from redis_func_cache.policies.fifo import FifoMultiplePolicy
-from redis_func_cache.policies.lfu import LfuMultiplePolicy
-from redis_func_cache.policies.lru import LruMultiplePolicy, LruTMultiplePolicy
-from redis_func_cache.policies.mru import MruMultiplePolicy
-from redis_func_cache.policies.rr import RrMultiplePolicy
+from redis_func_cache.policies.fifo import FifoClusterPolicy, FifoMultiplePolicy
+from redis_func_cache.policies.lfu import LfuClusterPolicy, LfuMultiplePolicy
+from redis_func_cache.policies.lru import LruClusterPolicy, LruMultiplePolicy, LruTClusterPolicy, LruTMultiplePolicy
+from redis_func_cache.policies.mru import MruClusterPolicy, MruMultiplePolicy
+from redis_func_cache.policies.rr import RrClusterPolicy, RrMultiplePolicy
 
 try:
     from dotenv import load_dotenv
@@ -69,6 +70,35 @@ async def close_async_redis_client():
 ASYNC_REDIS_FACTORY = lambda: AsyncRedis.from_url(REDIS_URL)  # noqa: E731
 REDIS_CLUSTER_NODES = getenv("REDIS_CLUSTER_NODES")
 
+# 解析 Redis 集群节点
+CLUSTER_NODES: List[ClusterNode] = []
+CLUSTER_CACHES: Dict[str, RedisFuncCache] = {}
+CLUSTER_MULTI_CACHES: Dict[str, RedisFuncCache] = {}
+
+if REDIS_CLUSTER_NODES:
+    CLUSTER_NODES = [
+        ClusterNode(cluster.split(":")[-2], int(cluster.split(":")[-1])) for cluster in REDIS_CLUSTER_NODES.split()
+    ]
+    REDIS_CLUSTER_FACTORY: Callable[[], RedisCluster] = lambda: RedisCluster(startup_nodes=CLUSTER_NODES)  # type: ignore[abstract]  # noqa: E731
+
+    CLUSTER_CACHES = {
+        "tlru": RedisFuncCache(__name__, LruTClusterPolicy, client=REDIS_CLUSTER_FACTORY, maxsize=MAXSIZE),
+        "lru": RedisFuncCache(__name__, LruClusterPolicy, client=REDIS_CLUSTER_FACTORY, maxsize=MAXSIZE),
+        "mru": RedisFuncCache(__name__, MruClusterPolicy, client=REDIS_CLUSTER_FACTORY, maxsize=MAXSIZE),
+        "rr": RedisFuncCache(__name__, RrClusterPolicy, client=REDIS_CLUSTER_FACTORY, maxsize=MAXSIZE),
+        "fifo": RedisFuncCache(__name__, FifoClusterPolicy, client=REDIS_CLUSTER_FACTORY, maxsize=MAXSIZE),
+        "lfu": RedisFuncCache(__name__, LfuClusterPolicy, client=REDIS_CLUSTER_FACTORY, maxsize=MAXSIZE),
+    }
+
+    CLUSTER_MULTI_CACHES = {
+        "tlru": RedisFuncCache(__name__, LruTClusterMultiplePolicy, client=REDIS_CLUSTER_FACTORY, maxsize=MAXSIZE),
+        "lru": RedisFuncCache(__name__, LruClusterMultiplePolicy, client=REDIS_CLUSTER_FACTORY, maxsize=MAXSIZE),
+        "mru": RedisFuncCache(__name__, MruClusterMultiplePolicy, client=REDIS_CLUSTER_FACTORY, maxsize=MAXSIZE),
+        "rr": RedisFuncCache(__name__, RrClusterMultiplePolicy, client=REDIS_CLUSTER_FACTORY, maxsize=MAXSIZE),
+        "fifo": RedisFuncCache(__name__, FifoClusterMultiplePolicy, client=REDIS_CLUSTER_FACTORY, maxsize=MAXSIZE),
+        "lfu": RedisFuncCache(__name__, LfuClusterMultiplePolicy, client=REDIS_CLUSTER_FACTORY, maxsize=MAXSIZE),
+    }
+
 
 CACHES = {
     "tlru": RedisFuncCache(__name__, LruTPolicy, client=REDIS_FACTORY, maxsize=MAXSIZE),