Zergling_man/moment
Zergling_man/moment
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Abort net manager talk for certain error codes

Browse Source
This commit is contained in:
miruka 2019-04-12 04:48:00 -04:00
parent 5d4c7b8520
commit 381c6b5b1c
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
harmonyqml/backend/network_manager.py
View File

@ -23,6 +23,9 @@ class NioErrorResponse(Exception):
class NetworkManager: class NetworkManager:
http_retry_codes = {408, 429, 500, 502, 503, 504, 507}
def __init__(self, client: Client) -> None: def __init__(self, client: Client) -> None:
self.client = client self.client = client
self._ssl_context: ssl.SSLContext = ssl.create_default_context() self._ssl_context: ssl.SSLContext = ssl.create_default_context()
@ -80,12 +83,15 @@ class NetworkManager:
response = self.read(sock) response = self.read(sock)
except NioErrorResponse as err: except NioErrorResponse as err:
logging.error("read bad response: %s", err) logging.error("read bad response for %s: %s", nio_func, err)
self._close_socket(sock) self._close_socket(sock)
if self._should_abort_talk(err):
logging.error("aborting talk")
break
time.sleep(10) time.sleep(10)
except Exception as err: except Exception as err:
logging.error("talk exception: %r", err) logging.error("talk exception for %s: %r", nio_func, err)
break break
else: else:
@ -93,3 +99,9 @@ class NetworkManager:
self._close_socket(sock) self._close_socket(sock)
return response return response
def _should_abort_talk(self, err: NioErrorResponse) -> bool:
if err.response.status_code in self.http_retry_codes:
return False
return True