فهرست منبع

Merge branch 'debug-timeout' into pre-master

liqian 3 سال پیش
والد
کامیت
ac01d1970a
1فایلهای تغییر یافته به همراه30 افزوده شده و 19 حذف شده
  1. 30 19
      db_helper.py

+ 30 - 19
db_helper.py

@@ -1,3 +1,5 @@
+import traceback
+
 import redis
 import psycopg2
 from psycopg2 import pool as pgpool
@@ -5,10 +7,11 @@ from config import set_config
 from log import Log
 
 config_ = set_config()
-log = Log()
+log_ = Log()
 
 conn_redis = None
 
+
 class RedisHelper(object):
     def __init__(self):
         """
@@ -28,9 +31,9 @@ class RedisHelper(object):
         global conn_redis
         if conn_redis is None:
             pool = redis.ConnectionPool(host=self.host,
-                                    port=self.port,
-                                    password=self.password,
-                                    decode_responses=True)
+                                        port=self.port,
+                                        password=self.password,
+                                        decode_responses=True)
             conn = redis.Redis(connection_pool=pool)
             conn_redis = conn
         return conn_redis
@@ -231,7 +234,10 @@ class RedisHelper(object):
 #hologres_info = config_.HOLOGRES_INFO
 #conn = psycopg2.connect(**hologres_info)
 #cur = conn.cursor()
+
 connectPool = pgpool.SimpleConnectionPool(1, 10, **config_.HOLOGRES_INFO)
+
+
 class HologresHelper(object):
     def __init__(self):
         """初始化hologres连接信息"""
@@ -239,21 +245,26 @@ class HologresHelper(object):
 
     def get_data(self, sql):
         #global conn
-        # 连接Hologres
-        #conn = psycopg2.connect(**self.hologres_info)
-        conn = connectPool.getconn()
-        # 创建游标
-        cur = conn.cursor()
-        # 查询数据
-        cur.execute(sql)
-        data = cur.fetchall()
-        cur.close()
-        connectPool.putconn(conn, close=False)
-        # 提交事务
-        #conn.commit()
-        # 释放资源
-        #cur.close()
-        #conn.close()
+        try:
+            # 连接Hologres
+            #conn = psycopg2.connect(**self.hologres_info)
+            conn = connectPool.getconn()
+            # 创建游标
+            cur = conn.cursor()
+            # 查询数据
+            cur.execute(sql)
+            data = cur.fetchall()
+            cur.close()
+            connectPool.putconn(conn, close=False)
+            # 提交事务
+            #conn.commit()
+            # 释放资源
+            #cur.close()
+            #conn.close()
+        except Exception as e:
+            log_.error('hologress error...')
+            log_.error(traceback.format_exc())
+            return []
         return data