58 lines
2.5 KiB
Python
58 lines
2.5 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
from threading import Thread
|
|||
|
|
from time import sleep, time
|
|||
|
|
from traceback import format_exc
|
|||
|
|
|
|||
|
|
from loguru import logger
|
|||
|
|
|
|||
|
|
from common.Constant import init_progess
|
|||
|
|
from enums.AnalysisStatusEnum import AnalysisStatus
|
|||
|
|
from entity.FeedBack import message_feedback
|
|||
|
|
from enums.ExceptionEnum import ExceptionType
|
|||
|
|
from exception.CustomerException import ServiceException
|
|||
|
|
from util.QueUtil import get_no_block_queue, put_queue, clear_queue
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Heartbeat(Thread):
|
|||
|
|
__slots__ = ('__fb_queue', '__hb_queue', '__request_id', '__analyse_type', "_context")
|
|||
|
|
|
|||
|
|
def __init__(self, *args):
|
|||
|
|
super().__init__()
|
|||
|
|
self.__fb_queue, self.__hb_queue, self.__request_id, self.__analyse_type, self._context = args
|
|||
|
|
|
|||
|
|
def run(self):
|
|||
|
|
request_id, hb_queue, progress = self.__request_id, self.__hb_queue, init_progess
|
|||
|
|
analyse_type, fb_queue = self.__analyse_type, self.__fb_queue
|
|||
|
|
service_timeout = int(self._context["service"]["timeout"]) + 120
|
|||
|
|
try:
|
|||
|
|
logger.info("开始启动心跳线程！requestId:{}", request_id)
|
|||
|
|
start_time = time()
|
|||
|
|
hb_init_num = 0
|
|||
|
|
while True:
|
|||
|
|
if time() - start_time > service_timeout:
|
|||
|
|
logger.error("心跳运行超时, requestId: {}", request_id)
|
|||
|
|
raise ServiceException(ExceptionType.TASK_EXCUTE_TIMEOUT.value[0],
|
|||
|
|
ExceptionType.TASK_EXCUTE_TIMEOUT.value[1])
|
|||
|
|
sleep(3)
|
|||
|
|
hb_msg = get_no_block_queue(hb_queue)
|
|||
|
|
if hb_msg is not None:
|
|||
|
|
command = hb_msg.get("command")
|
|||
|
|
hb_value = hb_msg.get("hb_value")
|
|||
|
|
if 'stop' == command:
|
|||
|
|
logger.info("开始终止心跳线程, requestId:{}", request_id)
|
|||
|
|
break
|
|||
|
|
if hb_value is not None:
|
|||
|
|
progress = hb_value
|
|||
|
|
if hb_init_num % 30 == 0:
|
|||
|
|
hb_init_num = 0
|
|||
|
|
put_queue(fb_queue, message_feedback(request_id, AnalysisStatus.RUNNING.value, analyse_type,
|
|||
|
|
progress=progress), timeout=3, is_ex=True)
|
|||
|
|
|
|||
|
|
hb_init_num += 3
|
|||
|
|
del hb_msg
|
|||
|
|
except Exception:
|
|||
|
|
logger.error("心跳线程异常:{}, requestId:{}", format_exc(), request_id)
|
|||
|
|
finally:
|
|||
|
|
clear_queue(hb_queue)
|
|||
|
|
logger.info("心跳线程停止完成！requestId:{}", request_id)
|