diff options
Diffstat (limited to 'redis/cluster.py')
| -rw-r--r-- | redis/cluster.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/redis/cluster.py b/redis/cluster.py index 235d8f2..5f730a8 100644 --- a/redis/cluster.py +++ b/redis/cluster.py @@ -121,6 +121,7 @@ REDIS_ALLOWED_KEYS = ( "charset", "connection_class", "connection_pool", + "connection_pool_class", "client_name", "credential_provider", "db", @@ -1267,6 +1268,7 @@ class NodesManager: require_full_coverage=False, lock=None, dynamic_startup_nodes=True, + connection_pool_class=ConnectionPool, **kwargs, ): self.nodes_cache = {} @@ -1277,6 +1279,7 @@ class NodesManager: self.from_url = from_url self._require_full_coverage = require_full_coverage self._dynamic_startup_nodes = dynamic_startup_nodes + self.connection_pool_class = connection_pool_class self._moved_exception = None self.connection_kwargs = kwargs self.read_load_balancer = LoadBalancer() @@ -1420,7 +1423,7 @@ class NodesManager: # Create a redis node with a costumed connection pool kwargs.update({"host": host}) kwargs.update({"port": port}) - r = Redis(connection_pool=ConnectionPool(**kwargs)) + r = Redis(connection_pool=self.connection_pool_class(**kwargs)) else: r = Redis(host=host, port=port, **kwargs) return r |