57 changing the cluster conn function to use domain nameshostnames instead of ips

falkordb/cluster.py

@@ -1,4 +1,6 @@
-from redis.cluster import RedisCluster
+from redis.cluster import RedisCluster, ClusterNode
+from redis.retry import Retry
+from redis.backoff import default_backoff
 
 # detect if a connection is a sentinel
 def Is_Cluster(conn):
@@ -7,12 +9,10 @@
 
 # create a cluster connection from a Redis connection
 def Cluster_Conn(conn, ssl):
-    # current sentinel
+    info = conn.execute_command("CLUSTER NODES")
+    nodes = [ ClusterNode(v['hostname'],k.split(':')[1]) for k,v in info.items()]
     connection_kwargs = conn.connection_pool.connection_kwargs
-    host = connection_kwargs['host']
-    port = connection_kwargs['port']
     username = connection_kwargs['username']
     password = connection_kwargs['password']
-
-    return RedisCluster(host=host, port=port, username=username, password=password, ssl=ssl)
+    return RedisCluster(Retry(default_backoff(),6),cluster_error_retry_attempts=20,startup_nodes=nodes,username=username,password=password,ssl=ssl)
 

pyproject.toml

@@ -13,7 +13,7 @@ packages = [{include = "falkordb"}]
 
 [tool.poetry.dependencies]
 python = "^3.8"
-redis = "^5.0.1"
+redis = {git = "https://github.com/MuhammadQadora/redis-py", rev = "master"}
 
 [tool.poetry.group.test.dependencies]
 pytest-cov = ">=4.1,<6.0"