chore: import upstream snapshot with attribution

python/ray/autoscaler/_private/readonly/node_provider.py

@@ -0,0 +1,80 @@
+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