mirror of
https://github.com/ihmily/StreamCap.git
synced 2026-05-07 22:15:58 +08:00
142c437dbd
* refactor: restructure app directory for better organization * update: add missing changes to __init__.py
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
import asyncio
|
|
import os
|
|
import threading
|
|
|
|
from ...utils.logger import logger
|
|
|
|
|
|
class BackgroundService:
|
|
|
|
_instance = None
|
|
|
|
@classmethod
|
|
def get_instance(cls):
|
|
if cls._instance is None:
|
|
cls._instance = BackgroundService()
|
|
return cls._instance
|
|
|
|
def __init__(self):
|
|
self.tasks = []
|
|
self.is_running = False
|
|
self.worker_thread = None
|
|
|
|
def add_task(self, task_func, *args, **kwargs):
|
|
self.tasks.append((task_func, args, kwargs))
|
|
logger.info(f"Added background task: {task_func.__name__}")
|
|
|
|
if not self.is_running:
|
|
self.start()
|
|
|
|
def start(self):
|
|
if self.is_running:
|
|
return
|
|
|
|
self.is_running = True
|
|
self.worker_thread = threading.Thread(target=self._process_tasks, daemon=False)
|
|
self.worker_thread.start()
|
|
logger.info("Background service started")
|
|
|
|
def _process_tasks(self):
|
|
while self.tasks:
|
|
task_func, args, kwargs = self.tasks.pop(0)
|
|
try:
|
|
logger.info(f"Executing background task: {task_func.__name__}")
|
|
task_func(*args, **kwargs)
|
|
logger.info(f"Background task completed: {task_func.__name__}")
|
|
except Exception as e:
|
|
logger.error(f"Background task execution failed: {e}")
|
|
|
|
logger.info("All background tasks completed, service stopped")
|
|
self.is_running = False
|
|
|
|
|
|
class AsyncProcessManager:
|
|
def __init__(self):
|
|
self.ffmpeg_processes = []
|
|
|
|
def add_process(self, process):
|
|
self.ffmpeg_processes.append(process)
|
|
|
|
async def cleanup(self):
|
|
for process in self.ffmpeg_processes[:]:
|
|
try:
|
|
if process.returncode is None:
|
|
logger.debug(f"Terminating process {process.pid}")
|
|
if os.name == "nt":
|
|
if process.stdin:
|
|
process.stdin.write(b"q")
|
|
await process.stdin.drain()
|
|
else:
|
|
process.terminate()
|
|
|
|
try:
|
|
await asyncio.wait_for(process.wait(), timeout=5.0)
|
|
except asyncio.TimeoutError:
|
|
logger.warning(f"Process {process.pid} did not terminate, killing it")
|
|
process.kill()
|
|
await process.wait()
|
|
|
|
self.ffmpeg_processes.remove(process)
|
|
except Exception as e:
|
|
logger.error(f"Error cleaning up process: {e}")
|
|
if process in self.ffmpeg_processes:
|
|
self.ffmpeg_processes.remove(process)
|
|
|
|
logger.debug("All processes cleaned up")
|