Robert Wojciechowski
3058b11ff2
- Remove DistributedListSplitter and DistributedListCollector nodes - Remove /distributed/init_list_queue and /distributed/request_list_item endpoints - Clean up all references from prompt_transform, queue_orchestration, constants, and tests - Refactor and reorganize distributed modules for quality improvements
67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
"""Single startup owner for node surface and runtime initialization."""
|
|
from __future__ import annotations
|
|
|
|
import atexit
|
|
import os
|
|
from functools import lru_cache
|
|
from typing import Any
|
|
|
|
|
|
@lru_cache(maxsize=1)
|
|
def build_node_mappings() -> tuple[dict[str, Any], dict[str, str]]:
|
|
"""Build and cache the full exported node mapping surface."""
|
|
from ..nodes import (
|
|
NODE_CLASS_MAPPINGS as distributed_class_mappings,
|
|
NODE_DISPLAY_NAME_MAPPINGS as distributed_display_name_mappings,
|
|
)
|
|
from ..nodes.distributed_upscale import (
|
|
NODE_CLASS_MAPPINGS as upscale_class_mappings,
|
|
NODE_DISPLAY_NAME_MAPPINGS as upscale_display_name_mappings,
|
|
)
|
|
|
|
return (
|
|
{**distributed_class_mappings, **upscale_class_mappings},
|
|
{**distributed_display_name_mappings, **upscale_display_name_mappings},
|
|
)
|
|
|
|
|
|
@lru_cache(maxsize=1)
|
|
def _runtime_initialized() -> dict[str, bool]:
|
|
return {"done": False}
|
|
|
|
|
|
def initialize_runtime(prompt_server: Any | None = None) -> None:
|
|
"""Initialize distributed runtime side effects once."""
|
|
state = _runtime_initialized()
|
|
if state["done"]:
|
|
return
|
|
|
|
import server
|
|
|
|
from ..api import bootstrap_routes
|
|
from ..api.queue_orchestration import ensure_distributed_state
|
|
from ..upscale.job_store import ensure_tile_jobs_initialized
|
|
from ..utils.config import CONFIG_FILE, ensure_config_exists
|
|
from ..utils.logging import debug_log
|
|
from ..workers.startup import delayed_auto_launch, register_async_signals, sync_cleanup
|
|
|
|
ensure_config_exists()
|
|
bootstrap_routes()
|
|
|
|
server_instance = prompt_server if prompt_server is not None else server.PromptServer.instance
|
|
ensure_distributed_state(server_instance)
|
|
ensure_tile_jobs_initialized()
|
|
|
|
atexit.register(lambda: None) # placeholder; real cleanup in sync_cleanup
|
|
if not os.environ.get("COMFYUI_IS_WORKER"):
|
|
atexit.register(sync_cleanup)
|
|
delayed_auto_launch()
|
|
register_async_signals()
|
|
|
|
state["done"] = True
|
|
|
|
node_mappings, _ = build_node_mappings()
|
|
debug_log("Loaded Distributed nodes.")
|
|
debug_log(f"Config file: {CONFIG_FILE}")
|
|
debug_log(f"Available nodes: {list(node_mappings.keys())}")
|