From 429e6a967c4133e2da0b0e576abb9537b810f33d Mon Sep 17 00:00:00 2001
From: Hossein Shafagh <hshafagh@netflix.com>
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