Skip to content

Commit c4a19a5

Browse files
committed
added connection retry for redis connector class
1 parent 88739a6 commit c4a19a5

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

pjon_python/utils/RedisConn.py

+11
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
from redis import ConnectionError
44
import logging
55
import jsonpickle
6+
from retrying import retry
7+
68
log = logging.getLogger("redis-conn")
79

810

@@ -41,6 +43,7 @@ def listen(self, rcv_timeout=0.01):
4143
return message['data']
4244
return None
4345

46+
@retry(wait_fixed=1000, stop_max_attempt_number=3)
4447
def publish(self, payload, channel=None):
4548
if channel is None:
4649
channel = self._pub_channel_name
@@ -51,26 +54,34 @@ def publish(self, payload, channel=None):
5154
pass
5255
self._redis_conn.publish(channel, payload)
5356

57+
@retry(wait_fixed=1000, stop_max_attempt_number=3)
5458
def hgetall(self, *args, **kwargs):
5559
return self._redis_conn.hgetall(*args, **kwargs)
5660

61+
@retry(wait_fixed=1000, stop_max_attempt_number=3)
5762
def hget(self, *args, **kwargs):
5863
return self._redis_conn.hget(*args, **kwargs)
5964

65+
@retry(wait_fixed=1000, stop_max_attempt_number=3)
6066
def hmset(self, *args, **kwargs):
6167
return self._redis_conn.hmset(*args, **kwargs)
6268

69+
@retry(wait_fixed=1000, stop_max_attempt_number=3)
6370
def delete(self, *args, **kwargs):
6471
return self._redis_conn.delete(*args, **kwargs)
6572

73+
@retry(wait_fixed=1000, stop_max_attempt_number=3)
6674
def hdel(self, *args, **kwargs):
6775
return self._redis_conn.hdel(*args, **kwargs)
6876

77+
@retry(wait_fixed=1000, stop_max_attempt_number=3)
6978
def hset(self, *args, **kwargs):
7079
return self._redis_conn.hset(*args, **kwargs)
7180

81+
@retry(wait_fixed=1000, stop_max_attempt_number=3)
7282
def set(self, *args, **kwargs):
7383
return self._redis_conn.set(*args, **kwargs)
7484

85+
@retry(wait_fixed=1000, stop_max_attempt_number=3)
7586
def get(self, *args, **kwargs):
7687
return self._redis_conn.get(*args, **kwargs)

requirements.txt

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
11
mock
22
nose
3-
pyserial >= 3.1.1
3+
pyserial >= 3.1.1
4+
retrying

0 commit comments

Comments
 (0)