From 429e6a967c4133e2da0b0e576abb9537b810f33d Mon Sep 17 00:00:00 2001 From: Hossein Shafagh Date: Thu, 25 Jul 2019 18:49:19 -0700 Subject: [PATCH] better error handling for redis --- lemur/common/redis.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/lemur/common/redis.py b/lemur/common/redis.py index 34a8778f..ca15734f 100644 --- a/lemur/common/redis.py +++ b/lemur/common/redis.py @@ -3,7 +3,9 @@ Helper Class for Redis """ import redis +import sys from flask import current_app +from lemur.extensions import sentry from lemur.factory import create_app if current_app: @@ -23,7 +25,19 @@ class RedisHandler: def redis(self, db=0): # The decode_responses flag here directs the client to convert the responses from Redis into Python strings # using the default encoding utf-8. This is client specific. - red = redis.StrictRedis(host=self.host, port=self.port, db=self.db, encoding="utf-8", decode_responses=True) + function = f"{__name__}.{sys._getframe().f_code.co_name}" + try: + red = redis.StrictRedis(host=self.host, port=self.port, db=self.db, encoding="utf-8", decode_responses=True) + red.set("test", 0) + except redis.ConnectionError: + log_data = { + "function": function, + "message": "Redis Connection error", + "host": self.host, + "port": self.port + } + current_app.logger.error(log_data) + sentry.captureException() return red