[downloader/fc2] Stop heatbeating once FFmpeg finishes

Authored by: Lesmiscore
This commit is contained in:
Lesmiscore 2022-04-25 00:45:19 +09:00
parent acbc642250
commit 69b59b4b4b
No known key found for this signature in database
GPG Key ID: 0EC2B52CF86236FF

View File

@ -18,6 +18,9 @@ class FC2LiveFD(FileDownloader):
heartbeat_state = [None, 1] heartbeat_state = [None, 1]
def heartbeat(): def heartbeat():
if heartbeat_state[1] < 0:
return
try: try:
heartbeat_state[1] += 1 heartbeat_state[1] += 1
ws.send('{"name":"heartbeat","arguments":{},"id":%d}' % heartbeat_state[1]) ws.send('{"name":"heartbeat","arguments":{},"id":%d}' % heartbeat_state[1])
@ -36,4 +39,8 @@ class FC2LiveFD(FileDownloader):
'ws': None, 'ws': None,
'protocol': 'live_ffmpeg', 'protocol': 'live_ffmpeg',
}) })
return FFmpegFD(self.ydl, self.params or {}).download(filename, new_info_dict) try:
return FFmpegFD(self.ydl, self.params or {}).download(filename, new_info_dict)
finally:
# stop heartbeating
heartbeat_state[1] = -1