fix connection reset error

陈帅民 · 陈帅民 · commit c352e78c3241 · 2024-03-29T10:24:19.000+08:00
diff --git a/Pipfile b/Pipfile
@@ -21,6 +21,7 @@ six = ">=1.11.0"
 qiniu = ">=7.1.4,<7.2.4"
 "urllib3" = ">=1.24.3,<=1.25.3"
 requests = ">=2.20.0,<=2.22.0"
+requests-toolbelt = ">=1.0.0"
 Werkzeug = ">=0.11.11,<1.0.0"
 gevent = ">=22.10.2,<23.0.0"
 typing = { version = "*", markers = "python_version < '3.5.0'" }
diff --git a/leancloud/client.py b/leancloud/client.py
@@ -13,6 +13,7 @@
 
 import six
 import requests
+from requests_toolbelt.adapters.socket_options import TCPKeepAliveAdapter
 
 import leancloud
 from leancloud import utils
@@ -43,6 +44,8 @@
 
 app_router = None
 session = requests.Session()
+session.mount("http://", TCPKeepAliveAdapter())
+session.mount("https://", TCPKeepAliveAdapter())
 request_hooks = {}
 
 SERVER_VERSION = "1.1"
diff --git a/requirements.txt b/requirements.txt
@@ -4,6 +4,7 @@ iso8601>=0.1.14
 six>=1.11.0
 qiniu>=7.3.1
 requests>=2.25.1
+requests-toolbelt>=1.0.0
 Werkzeug>=0.16.0,<2.0.0
 secure-cookie>=0.1.0,<1.0.0
 gevent>=22.10.2,<23.0.0
diff --git a/setup.py b/setup.py
@@ -12,6 +12,7 @@
     'six>=1.11.0',
     'qiniu==7.3.1',
     'requests>=2.25.1',
+    'requests-toolbelt>=1.0.0',
     'Werkzeug>=0.16.0,<2.0.0',
     'secure-cookie>=0.1.0,<1.0.0',
     'gevent>=22.10.2,<23.0.0',
