liqian пре 1 година
родитељ
комит
3f3e4c2bc4
1 измењених фајлова са 22 додато и 9 уклоњено
  1. 22 9
      db_helper.py

+ 22 - 9
db_helper.py

@@ -11,6 +11,7 @@ config_ = set_config()
 log_ = Log()
 
 conn_redis = None
+conn_filter_redis = None
 
 
 class RedisHelper(object):
@@ -19,6 +20,7 @@ class RedisHelper(object):
         初始化redis连接信息
         redis_info: redis连接信息, 格式:dict, {'host': '', 'port': '', 'password': ''}
         """
+        self.redis_info = redis_info
         self.host = redis_info['host']
         self.port = redis_info['port']
         self.password = redis_info['password']
@@ -29,15 +31,26 @@ class RedisHelper(object):
         连接redis
         :return: conn
         """
-        global conn_redis
-        if conn_redis is None:
-            pool = redis.ConnectionPool(host=self.host,
-                                        port=self.port,
-                                        password=self.password,
-                                        decode_responses=True)
-            conn = redis.Redis(connection_pool=pool)
-            conn_redis = conn
-        return conn_redis
+        if self.redis_info == config_.REDIS_INFO_FILTER:
+            global conn_filter_redis
+            if conn_filter_redis is None:
+                pool = redis.ConnectionPool(host=self.host,
+                                            port=self.port,
+                                            password=self.password,
+                                            decode_responses=True)
+                conn = redis.Redis(connection_pool=pool)
+                conn_filter_redis = conn
+            return conn_filter_redis
+        else:
+            global conn_redis
+            if conn_redis is None:
+                pool = redis.ConnectionPool(host=self.host,
+                                            port=self.port,
+                                            password=self.password,
+                                            decode_responses=True)
+                conn = redis.Redis(connection_pool=pool)
+                conn_redis = conn
+            return conn_redis
 
     def key_exists(self, key_name):
         """