redis cluster add timeout ssl

hhyo · Aug 24, 2024 · 1936d8c · 1936d8c
1 parent d181a03
commit 1936d8c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sql/engines/redis.py b/sql/engines/redis.py
@@ -11,13 +11,13 @@
 import shlex
 
 import redis
+import rediscluster
 import logging
 import traceback
 
 from common.utils.timer import FuncTimer
 from . import EngineBase
 from .models import ResultSet, ReviewSet, ReviewResult
-from rediscluster import RedisCluster
 
 __author__ = "hhyo"
 
@@ -28,9 +28,11 @@ class RedisEngine(EngineBase):
     def get_connection(self, db_name=None):
         db_name = db_name or self.db_name
         if self.mode == "cluster":
-            return RedisCluster(
+            return rediscluster.RedisCluster(
                 startup_nodes=[{"host": self.host, "port": self.port}],
                 decode_responses=True,
+                socket_connect_timeout=10,
+                ssl=self.is_ssl,
                 password=self.password,
             )
         else: