SashegDev
aaa19df5e4
- Multiple workers cause contention and slow down large file downloads - Single worker with async handles concurrent requests fine
82 lines
3.0 KiB
Python
82 lines
3.0 KiB
Python
# cli.py
|
|
import argparse
|
|
import sys
|
|
import asyncio
|
|
from pathlib import Path
|
|
import structlog
|
|
|
|
logger = structlog.get_logger(__name__)
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(description="ZernMC Launcher Server")
|
|
|
|
# Mode selection (mutually exclusive)
|
|
mode_group = parser.add_mutually_exclusive_group()
|
|
mode_group.add_argument("--dev", action="store_true", help="Development mode with auto-reload")
|
|
mode_group.add_argument("--prod", action="store_true", help="Production mode with 4 workers")
|
|
mode_group.add_argument("--test", action="store_true", help="Test mode - validate builds and generate manifests")
|
|
|
|
# Additional options
|
|
parser.add_argument("--host", default="0.0.0.0", help="Host to bind to (default: 0.0.0.0)")
|
|
parser.add_argument("--port", type=int, default=1582, help="Port to bind to (default: 1582)")
|
|
parser.add_argument("--workers", type=int, default=1, help="Number of workers for production mode (default: 1, more causes file download slowdown)")
|
|
parser.add_argument("--reload", action="store_true", help="Enable auto-reload (development)")
|
|
|
|
return parser.parse_args()
|
|
|
|
async def run_test_mode():
|
|
"""Validate all packs and generate/update manifests"""
|
|
logger.info("Running in TEST mode - validating builds and generating manifests")
|
|
|
|
from pack_manager import scan_pack, PACKS_DIR
|
|
|
|
pack_count = 0
|
|
error_count = 0
|
|
|
|
for pack_dir in PACKS_DIR.iterdir():
|
|
if pack_dir.is_dir():
|
|
try:
|
|
logger.info(f"Validating pack: {pack_dir.name}")
|
|
meta = await scan_pack(pack_dir.name)
|
|
pack_count += 1
|
|
logger.info(f"✓ Pack validated: {pack_dir.name} v{meta.version}, {len(meta.files)} files")
|
|
except Exception as e:
|
|
error_count += 1
|
|
logger.error(f"✗ Failed to validate pack {pack_dir.name}", error=str(e), exc_info=True)
|
|
|
|
logger.info(f"Test mode completed: {pack_count} packs validated, {error_count} errors")
|
|
|
|
if error_count > 0:
|
|
logger.error("Some packs failed validation")
|
|
sys.exit(1)
|
|
else:
|
|
logger.info("All packs validated successfully")
|
|
sys.exit(0)
|
|
|
|
def run_production_mode(host: str, port: int, workers: int):
|
|
"""Run with multiple workers"""
|
|
logger.info(f"Starting in PRODUCTION mode with {workers} workers on {host}:{port}")
|
|
|
|
import uvicorn
|
|
uvicorn.run(
|
|
"main:app",
|
|
host=host,
|
|
port=port,
|
|
workers=workers,
|
|
log_config=None,
|
|
access_log=False # We have our own logging middleware
|
|
)
|
|
|
|
def run_development_mode(host: str, port: int, reload: bool = True):
|
|
"""Run with auto-reload for development"""
|
|
logger.info(f"Starting in DEVELOPMENT mode with reload={reload} on {host}:{port}")
|
|
|
|
import uvicorn
|
|
uvicorn.run(
|
|
"main:app",
|
|
host=host,
|
|
port=port,
|
|
reload=reload,
|
|
log_config=None,
|
|
access_log=False
|
|
) |