o $i @s2ddlmZmZmZddlmZGdddZdS))ListSetTuple) constantsc@speZdZdZddZdedefddZdeded efd d Zdefd d Zde ede e eeffddZ dS) NodeTrackeraJMap nodes to their corresponding logs. We need to be a little careful here. At an given point in time, node_id <-> ip can be interchangeably used, but the node_id -> ip relation is not bijective _across time_ since IP addresses can be reused. Therefore, we should treat node_id as the only unique identifier. cCsi|_g|_dSN) node_mapping lru_order)selfr a/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/autoscaler/_private/node_tracker.py__init__s zNodeTracker.__init__node_idvaluecCsd||jvrdSt|jt|jksJt|jtjkr%|jd}|j|=||j|<|j|dS)Nr)rlenr rAUTOSCALER_MAX_NODES_TRACKEDpopappend)r rrr r r _add_node_mappings   zNodeTracker._add_node_mappingip node_typecCs"||jvr||||fdSdS)z Begin to track a new node. Args: node_id: The node id. ip: The node ip address. node_type: The node type. N)rr)r rrrr r r track#s zNodeTracker.trackcCs&||jvr|j||j|=dSdS)zGracefully stop tracking a node. If a node is intentionally removed from the cluster, we should stop tracking it so we don't mistakenly mark it as failed. Args: node_id: The node id which failed. N)rr remove)r rr r r untrack/s   zNodeTracker.untracknon_failed_idsreturncs@|j|g}tfdd|jD] }||j|q|S)aGet the information about all failed nodes. A failed node is any node which we began to track that is not pending or alive (i.e. not failed). Args: non_failed_ids: Nodes are failed unless they are in this set. Returns: List[Tuple[str, str]]: A list of tuples. Each tuple is the ip address and type of a failed node. cs|vSrr )r failed_nodesr r Ksz6NodeTracker.get_all_failed_node_info..)rkeysfilterr r)r r failed_inforr rr get_all_failed_node_info;s  z$NodeTracker.get_all_failed_node_infoN) __name__ __module__ __qualname____doc__r strrrrrrrr"r r r r rs   rN)typingrrrray.autoscaler._privaterrr r r r s