mirror of
https://github.com/Dispatcharr/Dispatcharr.git
synced 2026-07-28 12:36:42 +00:00
21901498de
This update addresses issues related to geventpool database connection leaks by consistently invoking `close_old_connections()` in `finally` blocks within the backup scheduler, plugin repository refresh, and core notification synchronization processes. These changes ensure that idle connections are properly released, preventing resource exhaustion. HLS proxy server has been disabled (hasn't been used). Additionally, the proxy server cleanup logic has been enhanced to safely handle the absence of proxy instances. Tests have been updated to validate these improvements and ensure stability in high-load scenarios.
78 lines
No EOL
3 KiB
Python
78 lines
No EOL
3 KiB
Python
from django.core.management.base import BaseCommand
|
|
from django.apps import apps
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Manage proxy servers'
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
'action',
|
|
choices=['start', 'stop', 'restart'],
|
|
help='Action to perform'
|
|
)
|
|
parser.add_argument(
|
|
'--type',
|
|
choices=['hls', 'ts', 'all'],
|
|
default='all',
|
|
help='Type of proxy to manage'
|
|
)
|
|
parser.add_argument(
|
|
'--channel',
|
|
help='Channel ID to manage'
|
|
)
|
|
parser.add_argument(
|
|
'--url',
|
|
help='Stream URL (required for start)'
|
|
)
|
|
|
|
def handle(self, *args, **options):
|
|
proxy_app = apps.get_app_config('proxy')
|
|
action = options['action']
|
|
proxy_type = options['type']
|
|
channel = options.get('channel')
|
|
url = options.get('url')
|
|
|
|
try:
|
|
hls_proxy = getattr(proxy_app, 'hls_proxy', None)
|
|
live_proxy = getattr(proxy_app, 'live_proxy', None)
|
|
|
|
if action == 'start':
|
|
if not url:
|
|
raise ValueError("URL is required for start action")
|
|
if proxy_type == 'hls' or (proxy_type == 'all' and hls_proxy is not None):
|
|
if hls_proxy is None:
|
|
raise ValueError("HLS proxy is not enabled")
|
|
hls_proxy.initialize_channel(url, channel or 'default')
|
|
if proxy_type in ('ts', 'all'):
|
|
if live_proxy is None:
|
|
raise ValueError("Live proxy is not initialized")
|
|
live_proxy.initialize_channel(url, channel or 'default')
|
|
self.stdout.write(self.style.SUCCESS('Started proxy servers'))
|
|
|
|
elif action == 'stop':
|
|
if proxy_type == 'hls' or (proxy_type == 'all' and hls_proxy is not None):
|
|
if hls_proxy is None:
|
|
raise ValueError("HLS proxy is not enabled")
|
|
if channel:
|
|
hls_proxy.stop_channel(channel)
|
|
else:
|
|
hls_proxy.shutdown()
|
|
if proxy_type in ('ts', 'all'):
|
|
if live_proxy is None:
|
|
raise ValueError("Live proxy is not initialized")
|
|
if channel:
|
|
live_proxy.stop_channel(channel)
|
|
else:
|
|
live_proxy.shutdown()
|
|
self.stdout.write(self.style.SUCCESS('Stopped proxy servers'))
|
|
|
|
elif action == 'restart':
|
|
self.handle(*args, **dict(options, action='stop'))
|
|
self.handle(*args, **dict(options, action='start'))
|
|
self.stdout.write(self.style.SUCCESS('Restarted proxy servers'))
|
|
|
|
except Exception as e:
|
|
self.stderr.write(self.style.ERROR(f'Error: {e}')) |