wehub-resource-sync
81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
from typing import List, Tuple
|
|
|
|
from ray.autoscaler._private.util import format_readonly_node_type
|
|
from ray.autoscaler.node_provider import NodeProvider
|
|
from ray.autoscaler.tags import (
|
|
NODE_KIND_HEAD,
|
|
STATUS_UP_TO_DATE,
|
|
TAG_RAY_NODE_KIND,
|
|
TAG_RAY_NODE_NAME,
|
|
TAG_RAY_NODE_STATUS,
|
|
TAG_RAY_USER_NODE_TYPE,
|
|
)
|
|
|
|
|
|
class ReadOnlyNodeProvider(NodeProvider):
|
|
"""A node provider that merely reports the current cluster state.
|
|
|
|
This is used for laptop mode / manual cluster setup modes, in order to
|
|
provide status reporting in the same way for users."""
|
|
|
|
def __init__(self, provider_config, cluster_name):
|
|
NodeProvider.__init__(self, provider_config, cluster_name)
|
|
self.nodes = {}
|
|
|
|
def is_readonly(self):
|
|
return True
|
|
|
|
def _set_nodes(self, nodes: List[Tuple[str, str]]):
|
|
"""Update the set of nodes in the cluster.
|
|
|
|
Args:
|
|
nodes: List of (node_id, node_manager_address) tuples.
|
|
"""
|
|
new_nodes = {}
|
|
for node_id, node_manager_address in nodes:
|
|
# We make up a fake node type for each node (since each node
|
|
# could have its own unique configuration).
|
|
new_nodes[node_id] = {
|
|
# Keep prefix in sync with node config gen in monitor.py
|
|
"node_type": format_readonly_node_type(node_id),
|
|
"ip": node_manager_address,
|
|
}
|
|
self.nodes = new_nodes
|
|
|
|
def non_terminated_nodes(self, tag_filters):
|
|
return list(self.nodes.keys())
|
|
|
|
def is_running(self, node_id):
|
|
return node_id in self.nodes
|
|
|
|
def is_terminated(self, node_id):
|
|
return node_id not in self.nodes
|
|
|
|
def node_tags(self, node_id):
|
|
tags = {
|
|
TAG_RAY_NODE_KIND: NODE_KIND_HEAD,
|
|
TAG_RAY_USER_NODE_TYPE: self.nodes[node_id]["node_type"],
|
|
TAG_RAY_NODE_NAME: node_id,
|
|
TAG_RAY_NODE_STATUS: STATUS_UP_TO_DATE,
|
|
}
|
|
return tags
|
|
|
|
def external_ip(self, node_id):
|
|
return node_id
|
|
|
|
def internal_ip(self, node_id):
|
|
return node_id
|
|
|
|
def set_node_tags(self, node_id, tags):
|
|
raise AssertionError("Readonly node provider cannot be updated")
|
|
|
|
def create_node(self, node_config, tags, count):
|
|
raise AssertionError("Readonly node provider cannot be updated")
|
|
|
|
def terminate_node(self, node_id):
|
|
raise AssertionError("Readonly node provider cannot be updated")
|
|
|
|
@staticmethod
|
|
def bootstrap_config(cluster_config):
|
|
return cluster_config
|