51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
# -*- coding: utf-8 -*-
|
||
from threading import Thread
|
||
import time
|
||
from traceback import format_exc
|
||
|
||
from loguru import logger
|
||
|
||
from entity.FeedBack import recording_feedback
|
||
from enums.RecordingStatusEnum import RecordingStatus
|
||
from util.QueUtil import get_no_block_queue, put_queue, clear_queue
|
||
|
||
|
||
class RecordingHeartbeat(Thread):
|
||
__slots__ = ('_fb_queue', '_hb_queue', '_request_id')
|
||
|
||
def __init__(self, fb_queue, hb_queue, request_id):
|
||
super().__init__()
|
||
self._fb_queue = fb_queue
|
||
self._hb_queue = hb_queue
|
||
self._request_id = request_id
|
||
|
||
def run(self):
|
||
request_id = self._request_id
|
||
hb_queue, fb_queue = self._hb_queue, self._fb_queue
|
||
logger.info("开始启动录屏心跳线程！requestId:{}", request_id)
|
||
hb_init_num, progress = 0, '0.0000'
|
||
status = RecordingStatus.RECORDING_WAITING.value[0]
|
||
try:
|
||
while True:
|
||
time.sleep(3)
|
||
hb_msg = get_no_block_queue(hb_queue)
|
||
if hb_msg is not None and len(hb_msg) > 0:
|
||
command_que = hb_msg.get("command")
|
||
progress_que = hb_msg.get("progress")
|
||
status_que = hb_msg.get("status")
|
||
if progress_que is not None:
|
||
progress = progress_que
|
||
if status_que is not None:
|
||
status = status_que
|
||
if 'stop' == command_que:
|
||
logger.info("开始终止心跳线程, requestId:{}", request_id)
|
||
break
|
||
if hb_init_num % 30 == 0:
|
||
put_queue(fb_queue, recording_feedback(request_id, status, progress=progress), timeout=5, is_ex=True)
|
||
hb_init_num += 3
|
||
except Exception:
|
||
logger.error("心跳线程异常:{}, requestId:{}", format_exc(), request_id)
|
||
finally:
|
||
clear_queue(hb_queue)
|
||
logger.info("心跳线程停止完成！requestId:{}", request_id)
|