From 0750169af968477a1fd9680e5db195a589fb6437 Mon Sep 17 00:00:00 2001 From: bensonhome <410554565@qq.com> Date: Thu, 1 Feb 2024 16:20:12 +0800 Subject: [PATCH] =?UTF-8?q?:art:=E5=A2=9E=E5=8A=A0=E5=BF=83=E8=B7=B3?= =?UTF-8?q?=E5=BC=82=E5=B8=B8=E6=8D=95=E8=8E=B7=EF=BC=8C=E9=81=BF=E5=85=8D?= =?UTF-8?q?=E7=9F=AD=E6=97=B6api=E8=AE=BF=E9=97=AE=E5=BC=82=E5=B8=B8?= =?UTF-8?q?=E5=AF=BC=E8=87=B4=E5=BF=83=E8=B7=B3=E7=BA=BF=E7=A8=8B=E9=80=80?= =?UTF-8?q?=E5=87=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- client/node/localtask/jobheartbeat.py | 5 ++++- client/node/servertask/nodemgr.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/client/node/localtask/jobheartbeat.py b/client/node/localtask/jobheartbeat.py index effe4d05f..e73fc9111 100644 --- a/client/node/localtask/jobheartbeat.py +++ b/client/node/localtask/jobheartbeat.py @@ -36,8 +36,11 @@ def __init__(self, event, org_sid, team_name, repo_id, project_id, job_id, serve def run(self): while self._event.is_set(): + try: + self._server.job_heart_beat(self._org_sid, self._team_name, self._repo_id, self._project_id, self._job_id) + except Exception as err: + logger.exception(f"job heartbeat error: {str(err)}") time.sleep(self._sleep_interval) - self._server.job_heart_beat(self._org_sid, self._team_name, self._repo_id, self._project_id, self._job_id) class JobHeartBeat(object): diff --git a/client/node/servertask/nodemgr.py b/client/node/servertask/nodemgr.py index 998fa8762..3593a7c58 100644 --- a/client/node/servertask/nodemgr.py +++ b/client/node/servertask/nodemgr.py @@ -118,7 +118,10 @@ def _thread_beat(self): host_ip = HostNetMgr().get_host_ip() data = {"puppy_ip": host_ip} # LogPrinter.info(f">>> data: {data}") - self._server.heart_beat(data) + try: + self._server.heart_beat(data) + except Exception as err: + LogPrinter.exception(f"heart beat error: {str(err)}") time.sleep(self._beat_interval) def start(self):