Files
n8n/packages/@n8n/task-runner-python/src/main.py

73 lines
2.0 KiB
Python

import asyncio
import logging
import sys
import platform
from src.constants import ERROR_WINDOWS_NOT_SUPPORTED
from src.config.health_check_config import HealthCheckConfig
from src.config.sentry_config import SentryConfig
from src.config.task_runner_config import TaskRunnerConfig
from src.errors import ConfigurationError
from src.logs import setup_logging
from src.task_runner import TaskRunner
from src.shutdown import Shutdown
async def main():
setup_logging()
logger = logging.getLogger(__name__)
sentry = None
sentry_config = SentryConfig.from_env()
if sentry_config.enabled:
from src.sentry import setup_sentry
sentry = setup_sentry(sentry_config)
try:
health_check_config = HealthCheckConfig.from_env()
except ConfigurationError as e:
logger.error(f"Invalid health check configuration: {e}")
sys.exit(1)
health_check_server: "HealthCheckServer | None" = None
if health_check_config.enabled:
from src.health_check_server import HealthCheckServer
health_check_server = HealthCheckServer()
try:
await health_check_server.start(health_check_config)
except OSError as e:
logger.error(f"Failed to start health check server: {e}")
sys.exit(1)
try:
task_runner_config = TaskRunnerConfig.from_env()
except ConfigurationError as e:
logger.error(str(e))
sys.exit(1)
task_runner = TaskRunner(task_runner_config)
logger.info("Starting runner...")
shutdown = Shutdown(task_runner, health_check_server, sentry)
task_runner.on_idle_timeout = shutdown.start_auto_shutdown
try:
await task_runner.start()
except Exception:
logger.error("Unexpected error", exc_info=True)
await shutdown.start_shutdown()
exit_code = await shutdown.wait_for_shutdown()
sys.exit(exit_code)
if __name__ == "__main__":
if platform.system() == "Windows":
print(ERROR_WINDOWS_NOT_SUPPORTED, file=sys.stderr)
sys.exit(1)
asyncio.run(main())