From d8a05a5ec1d689398e7c62c1e1af07f7f9373944 Mon Sep 17 00:00:00 2001
From: weideguo <2515717245@qq.com>
Date: Mon, 10 Jan 2022 23:19:46 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0redis/pg=E5=88=9B=E5=BB=BA?=
 =?UTF-8?q?=E8=BF=9E=E6=8E=A5=E6=97=B6=E7=9A=84=E8=B6=85=E6=97=B6=E5=8F=82?=
 =?UTF-8?q?=E6=95=B0=20(hhyo#1311)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 sql/engines/pgsql.py | 2 +-
 sql/engines/redis.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/sql/engines/pgsql.py b/sql/engines/pgsql.py
index 27ef8272e8..1444116554 100644
--- a/sql/engines/pgsql.py
+++ b/sql/engines/pgsql.py
@@ -29,7 +29,7 @@ def get_connection(self, db_name=None):
         if self.conn:
             return self.conn
         self.conn = psycopg2.connect(host=self.host, port=self.port, user=self.user,
-                                     password=self.password, dbname=db_name)
+                                     password=self.password, dbname=db_name, connect_timeout=10)
         return self.conn
 
     @property
diff --git a/sql/engines/redis.py b/sql/engines/redis.py
index 64707ba814..99b90a94bb 100644
--- a/sql/engines/redis.py
+++ b/sql/engines/redis.py
@@ -26,7 +26,7 @@ class RedisEngine(EngineBase):
     def get_connection(self, db_name=None):
         db_name = db_name or self.db_name
         return redis.Redis(host=self.host, port=self.port, db=db_name, password=self.password,
-                           encoding_errors='ignore', decode_responses=True)
+                           encoding_errors='ignore', decode_responses=True, socket_connect_timeout=10)
 
     @property
     def name(self):