tursom/blrec
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

perf: postprocessing one video only at the same time

Browse Source
This commit is contained in:
acgnhik 2022-06-19 13:32:56 +08:00
parent b74c25ebb6
commit 765df3ead9
1 changed files with 30 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/blrec/postprocess/postprocessor.py
View File

@ -4,7 +4,7 @@ import asyncio
import logging
from contextlib import suppress
from pathlib import PurePath
from typing import Any, Awaitable, Dict, Iterator, List, Optional, Union
from typing import Any, Awaitable, Dict, Final, Iterator, List, Optional, Union
from reactivex.scheduler import ThreadPoolScheduler
@ -48,6 +48,8 @@ class Postprocessor(
AsyncCooperationMixin,
SupportDebugMixin,
):
_worker_semaphore: Final = asyncio.Semaphore(value=1)
def __init__(
self,
live: Live,
@ -127,12 +129,15 @@ class Postprocessor(
@aio_task_with_room_id
async def _worker(self) -> None:
while True:
self._status = PostprocessorStatus.WAITING
self._postprocessing_path = None
self._postprocessing_progress = None
video_path = await self._queue.get()
async with self._worker_semaphore:
logger.debug(f'Postprocessing... {video_path}')
if not await self._is_vaild_flv_file(video_path):
@ -154,7 +159,9 @@ class Postprocessor(
await discard_file(extra_metadata_path(video_path), 'DEBUG')
self._completed_files.append(result_path)
await self._emit('video_postprocessing_completed', self, result_path)
await self._emit(
'video_postprocessing_completed', self, result_path
)
except Exception as exc:
submit_exception(exc)
finally: