support more redis connection options

tschellenbach · Mar 3, 2017 · 1099dbb · 1099dbb
1 parent 70fcab3
commit 1099dbb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/stream_framework/__init__.py b/stream_framework/__init__.py
@@ -4,7 +4,7 @@
 
 
 __license__ = 'BSD'
-__version__ = '1.3.6'
+__version__ = '1.3.7'
 __maintainer__ = 'Thierry Schellenbach'
 __email__ = 'thierryschellenbach@gmail.com'
 __status__ = 'Production'
diff --git a/stream_framework/storage/redis/connection.py b/stream_framework/storage/redis/connection.py
@@ -29,7 +29,13 @@ def setup_redis():
             port=config['port'],
             password=config.get('password'),
             db=config['db'],
-            decode_responses=True
+            decode_responses=config.get('decode_responses', True),
+            # connection options
+            socket_timeout=config.get('socket_timeout', None),
+            socket_connect_timeout=config.get('socket_connect_timeout', None),
+            socket_keepalive=config.get('socket_keepalive', False),
+            socket_keepalive_options=config.get('socket_keepalive_options', None),
+            retry_on_timeout=config.get('retry_on_timeout', False),
         )
         pools[name] = pool
     return pools