o bi3=@sddlZddlZddlZddlZddlZddlZddlZddlmm Z ddl m m Z ddlm mZddlmZddlmZddlmZddlmZddlmZddlmZeeZGdd d Zed krej d d Z!e!j"d de#dde!j"dde#dde!j"dde#dde!j"dde$dde!j"dde$dde!j"dde$dde!j"dde#ddde!j"dde$e j%d de!j"d!de#dd"de!j"d#dd$d%e j&e j'e j(d&e!j"d'de#e j)e j*de!j"d(de#e j+d),e j+de!j"d*de$d+de!j"d,de$d-de!j"d.de#dd/de!j"d0de#dd1de!j"d2de#dd3de!j"d4d5d6d7e!j"d8d5d9d7e!j"d:de#dd;de!j"dde!j"d?de#d=d@de!-Z.znej/dAkr^e.j0ndZ1ej/dAkrie.j2ndBZ3ee.j4e.j5e.j6e.j7e1e3dCZe8e.j9e.j:e1e3eZ;ee.je.j?e.j@e.jAe.jBe.j6e.jCe.jDe.jEe.jFe.jGe.jHe.jIdDZJdEdFZKej/dAkre;LejMeKe;NeJOWdSePyeQdGeRdBYdSwdS)HN)get_or_create_event_loop) logging_utils)create_check_raylet_task)AGENT_GRPC_MAX_MESSAGE_LENGTH)setup_component_logger) GcsClientc@speZdZddejdfdedededededed efd d Zd d ZddZ e ddZ defddZ ddZ dS)DashboardAgentNFdisable_metrics_collectionobject_store_name raylet_namelog_dirtemp_dir session_dir session_namec Cs|dusJ||_||_|dusJ||_||_| |_||_| |_||_||_||_ ||_ | |_ | |_ t jd|_| |_||_d|_d|_t|j|jd|_|jsW|dSdS)z%Initialize the DashboardAgent object.N RAY_NODE_ID)address cluster_id)ipminimal gcs_addresscluster_id_hexr rr dashboard_agent_portmetrics_export_portnode_manager_port listen_portr r osenvironnode_idmetrics_collection_disabledrserver http_serverr gcs_client_init_non_minimal)selfnode_ip_addressrrrrrrrr r r r r rrr%G/home/ubuntu/.local/lib/python3.10/site-packages/ray/dashboard/agent.py__init__s6    zDashboardAgent.__init__cCs&ddlm}ddlm}ddlm}tjjdkrBtjj dkrBddl }| |j dt d|Wdn1sz j|IdHWnty=}ztd|dd}WYd}~nd}~ww|rjsEdnjj}jsNdnj}j j t j j tj||gdtjd}j j t jjtj ||gdtjd}t||IdHfdd|D}tjd vrfd d } tjj | t} || jr|jn d d } || tj|IdHjr߈jIdHdSdS)NTz,Failed to start HTTP server with exception: zB. The agent will stay alive but the HTTP service will be disabled.F) namespacecsg|]}|jqSr%)runr).0mrPr%r& sz&DashboardAgent.run..)win32cygwincs$tdjdjd|dS)NzTerminated Raylet: ip=z , node_id=z. )r?rArr)msgrPr%r&callbacksz$DashboardAgent.run..callbackcs tdIdHq)NTi)asynciosleepr%r%r%r& wait_foreversz(DashboardAgent.run..wait_forever)rstartrNr r>r?r@ http_portr=r!async_internal_kv_putdashboard_consts#DASHBOARD_AGENT_ADDR_NODE_ID_PREFIXrencodejsondumpsr ray_constantsKV_NAMESPACE_DASHBOARDDASHBOARD_AGENT_ADDR_IP_PREFIXr^gatherr4platformrr looprHwait_for_terminationcleanup) r#rJlaunch_http_servererbr=put_by_node_id put_by_iptasksr]check_parent_taskr`r%rPr&rVs\       zDashboardAgent.run)rG __module__ __qualname__ri#DEFAULT_DASHBOARD_AGENT_LISTEN_PORTboolstrr'r"rNpropertyrOrSrVr%r%r%r&rs4    89  r__main__zDashboard agent.) descriptionz--node-ip-addressTzthe IP address of this node.)requiredtypehelpz --gcs-addresszThe address (ip:port) of GCS.z--cluster-id-hexzThe cluster id in hex.z--metrics-export-portz.The port to expose metrics through Prometheus.z--dashboard-agent-portz9The port on which the dashboard agent will receive GRPCs.z--node-manager-portz-The port to use for starting the node managerz--object-store-namez#The socket name of the plasma store)rrdefaultrz --listen-portFz!Port for HTTP server to listen onz --raylet-namez%The socket path of the raylet processz--logging-levelcCst|SrR)logging getLevelNameupper)sr%r%r&'sr)rrrchoicesrz--logging-formatz--logging-filenamezJSpecify the name of log file, log to stdout if set empty, default is "{}".z--logging-rotate-bytesz,Specify the max bytes for rotating log file.z--logging-rotate-backup-countz-Specify the backup count of rotated log file.z --log-dirz"Specify the path of log directory.z --temp-dirz?Specify the path of the temporary directory use by Ray process.z --session-dirz!Specify the path of this session.z --minimal store_truezMinimal agent only contains a subset of features that don't require additional dependencies installed when ray is installed by `pip install 'ray[default]'`.)actionrz--disable-metrics-collectionzCIf this arg is set, metrics report won't be enabled from the agent.z--session-namez.The session name (cluster id) of this cluster.z--stdout-filepathz,The filepath to dump dashboard agent stdout.z--stderr-filepathz,The filepath to dump dashboard agent stderr.rZ) logging_levellogging_formatr filename max_bytes backup_count) r rr rrrr r r rcCstdttjdS)Nz#Exiting with SIGTERM immediately...)r?warningr_exitsignalSIGTERMr%r%r%r&sigterm_handlers rz6Agent is working abnormally. It will exit immediately.)Sargparser^rgrrrr4ray._private.ray_constants_privateriray.dashboard.consts dashboardconstsrdray.dashboard.utilsutilsrDray._common.utilsr ray._privaterray._private.process_watcherrrray._private.ray_loggingr ray._rayletr getLoggerrGr?rArgumentParserparser add_argumentr{intry LOGGER_LEVELLOGGER_LEVEL_CHOICESLOGGER_LEVEL_HELP LOGGER_FORMATLOGGER_FORMAT_HELPDASHBOARD_AGENT_LOG_FILENAMEformat parse_argsargsrmlogging_rotate_byteslogging_rotation_byteslogging_rotate_backup_countlogging_rotation_backup_countrrr logging_filename redirect_stdout_stderr_if_neededstdout_filepathstderr_filepathrnr$rrrrr rrrrr r r ragentradd_signal_handlerrrun_until_completerVr>r@exitr%r%r%r&s        W