From 39fa397bc5399ad1b0c883a02ad286c5e0a998a6 Mon Sep 17 00:00:00 2001 From: Taha Yassine Kraiem Date: Fri, 22 Apr 2022 17:35:24 +0200 Subject: [PATCH] feat(api): chalice-assist configurable-timeout --- api/chalicelib/core/assist.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/api/chalicelib/core/assist.py b/api/chalicelib/core/assist.py index dc70eaa6f..b2926fd0c 100644 --- a/api/chalicelib/core/assist.py +++ b/api/chalicelib/core/assist.py @@ -25,7 +25,8 @@ def get_live_sessions_ws(project_id, user_id=None): if user_id and len(user_id) > 0: params["userId"] = user_id try: - connected_peers = requests.get(config("assist") % config("S3_KEY") + f"/{project_key}", params, timeout=6) + connected_peers = requests.get(config("assist") % config("S3_KEY") + f"/{project_key}", params, + timeout=config("assistTimeout", cast=int, default=5)) if connected_peers.status_code != 200: print("!! issue with the peer-server") print(connected_peers.text) @@ -63,7 +64,8 @@ def is_live(project_id, session_id, project_key=None): if project_key is None: project_key = projects.get_project_key(project_id) try: - connected_peers = requests.get(config("assistList") % config("S3_KEY") + f"/{project_key}", timeout=6) + connected_peers = requests.get(config("assistList") % config("S3_KEY") + f"/{project_key}", + timeout=config("assistTimeout", cast=int, default=5)) if connected_peers.status_code != 200: print("!! issue with the peer-server") print(connected_peers.text)