mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
108 lines
4.1 KiB
Python
108 lines
4.1 KiB
Python
import logging
|
|
|
|
import ray.new_dashboard.consts as dashboard_consts
|
|
from ray.new_dashboard.utils import Dict, Signal
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class GlobalSignals:
|
|
node_info_fetched = Signal(dashboard_consts.SIGNAL_NODE_INFO_FETCHED)
|
|
|
|
|
|
class DataSource:
|
|
# {ip address(str): node stats(dict of GetNodeStatsReply
|
|
# in node_manager.proto)}
|
|
node_stats = Dict()
|
|
# {ip address(str): node physical stats(dict from reporter_agent.py)}
|
|
node_physical_stats = Dict()
|
|
# {actor id hex(str): actor table data(dict of ActorTableData
|
|
# in gcs.proto)}
|
|
actors = Dict()
|
|
# {ip address(str): dashboard agent grpc server port(int)}
|
|
agents = Dict()
|
|
# {ip address(str): gcs node info(dict of GcsNodeInfo in gcs.proto)}
|
|
nodes = Dict()
|
|
# {hostname(str): ip address(str)}
|
|
hostname_to_ip = Dict()
|
|
# {ip address(str): hostname(str)}
|
|
ip_to_hostname = Dict()
|
|
|
|
|
|
class DataOrganizer:
|
|
@staticmethod
|
|
async def purge():
|
|
# Purge data that is out of date.
|
|
# These data sources are maintained by DashboardHead,
|
|
# we do not needs to purge them:
|
|
# * agents
|
|
# * nodes
|
|
# * hostname_to_ip
|
|
# * ip_to_hostname
|
|
logger.info("Purge data.")
|
|
valid_keys = DataSource.ip_to_hostname.keys()
|
|
for key in DataSource.node_stats.keys() - valid_keys:
|
|
DataSource.node_stats.pop(key)
|
|
|
|
for key in DataSource.node_physical_stats.keys() - valid_keys:
|
|
DataSource.node_physical_stats.pop(key)
|
|
|
|
@classmethod
|
|
async def get_node_actors(cls, hostname):
|
|
ip = DataSource.hostname_to_ip[hostname]
|
|
node_stats = DataSource.node_stats.get(ip, {})
|
|
node_worker_id_set = set()
|
|
for worker_stats in node_stats.get("workersStats", []):
|
|
node_worker_id_set.add(worker_stats["workerId"])
|
|
node_actors = {}
|
|
for actor_id, actor_table_data in DataSource.actors.items():
|
|
if actor_table_data["workerId"] in node_worker_id_set:
|
|
node_actors[actor_id] = actor_table_data
|
|
return node_actors
|
|
|
|
@classmethod
|
|
async def get_node_info(cls, hostname):
|
|
ip = DataSource.hostname_to_ip[hostname]
|
|
node_physical_stats = DataSource.node_physical_stats.get(ip, {})
|
|
node_stats = DataSource.node_stats.get(ip, {})
|
|
|
|
# Merge coreWorkerStats (node stats) to workers (node physical stats)
|
|
workers_stats = node_stats.pop("workersStats", {})
|
|
pid_to_worker_stats = {}
|
|
pid_to_language = {}
|
|
pid_to_job_id = {}
|
|
for stats in workers_stats:
|
|
d = pid_to_worker_stats.setdefault(stats["pid"], {}).setdefault(
|
|
stats["workerId"], stats["coreWorkerStats"])
|
|
d["workerId"] = stats["workerId"]
|
|
pid_to_language.setdefault(stats["pid"],
|
|
stats.get("language", "PYTHON"))
|
|
pid_to_job_id.setdefault(stats["pid"],
|
|
stats["coreWorkerStats"]["jobId"])
|
|
|
|
for worker in node_physical_stats.get("workers", []):
|
|
worker_stats = pid_to_worker_stats.get(worker["pid"], {})
|
|
worker["coreWorkerStats"] = list(worker_stats.values())
|
|
worker["language"] = pid_to_language.get(worker["pid"], "")
|
|
worker["jobId"] = pid_to_job_id.get(worker["pid"], "ffff")
|
|
|
|
# Merge node stats to node physical stats
|
|
node_info = node_physical_stats
|
|
node_info["raylet"] = node_stats
|
|
node_info["actors"] = await cls.get_node_actors(hostname)
|
|
node_info["state"] = DataSource.nodes.get(ip, {}).get("state", "DEAD")
|
|
|
|
await GlobalSignals.node_info_fetched.send(node_info)
|
|
|
|
return node_info
|
|
|
|
@classmethod
|
|
async def get_all_node_summary(cls):
|
|
all_nodes_summary = []
|
|
for hostname in DataSource.hostname_to_ip.keys():
|
|
node_info = await cls.get_node_info(hostname)
|
|
node_info.pop("workers", None)
|
|
node_info["raylet"].pop("workersStats", None)
|
|
node_info["raylet"].pop("viewData", None)
|
|
all_nodes_summary.append(node_info)
|
|
return all_nodes_summary
|