o `Û·i¸ ã@s2ddlmZmZmZddlmZGdd„dƒZdS)é)ÚListÚSetÚTuple)Ú constantsc@speZdZdZdd„Zdedefdd„Zdeded efd d „Zdefd d „Zde ede e eeffdd„Z 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|_dS©N)Ú node_mappingÚ lru_order)Úself©r úZ/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/autoscaler/_private/node_tracker.pyÚ__init__s zNodeTracker.__init__Únode_idÚvaluecCsd||jvrdSt|jƒt|jƒksJ‚t|jƒtjkr%|j d¡}|j|=||j|<|j |¡dS)Nr)rÚlenr rÚAUTOSCALER_MAX_NODES_TRACKEDÚpopÚappend)r rrr r r Ú_add_node_mappings   zNodeTracker._add_node_mappingÚipÚ node_typecCs"||jvr| |||f¡dSdS)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)zñGracefully 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.untrackÚnon_failed_idsÚreturncs@|j ¡|‰g}t‡fdd„|jƒD] }| |j|¡q|S)a‚Get 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..)rÚkeysÚfilterr 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)ÚtypingrrrÚray.autoscaler._privaterrr r r r Ús