o 1 i=@sddlZddlZddlZddlZddlZddlZddlZddlmm Z ddl m m Z ddlm mZddlmZmZddlmZddlmZddlmZddlmZddlmZddlmZee Z!Gd d d Z"e d krej#d d Z$e$j%dde&dde$j%dde&dde$j%dde&dde$j%dde'dde$j%dde'dde$j%dde'dde$j%dde&ddde$j%dd e'e j(d!de$j%d"de&dd#de$j%d$d d%d&e j)e j*e j+d'e$j%d(d e&e j,e j-de$j%d)d e&e j.d*/e j.de$j%d+de'd,de$j%d-de'd.de$j%d/de&dd0de$j%d1de&dd2de$j%d3de&dd4de$j%d5d6d7d8e$j%d9d6d:d8e$j%d;d e&ddd?de$j%d@d e&d>dAde$0Z1znej2dBkrfe1j3ndZ4ej2dBkrqe1j5ndCZ6ee1j7e1j8e1j9e1j:e4e6dDZ!e;e1je"e1j?e1j@e1jAe1jBe1jCe1jDe1jEe1j9e1jFe1jGe1jHe1jIe1jJe1jKe1jLdEZMdFdGZNej2dBkre>OejPeNe>QeMRWdSeSye!TdHeUdCYdSwdS)IN) build_address is_localhost)get_or_create_event_loop) logging_utils)create_check_raylet_task)AGENT_GRPC_MAX_MESSAGE_LENGTH)setup_component_logger) GcsClientc@sreZdZdddejdfdedededededed 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||_||_||_||_| |_||_||_||_ ||_ | |_ | |_ | |_ tjd|_| |_||_d|_d|_t|j|jd|_|jsZ|dSdS)z%Initialize the DashboardAgent object.NZ RAY_NODE_ID)addressZ cluster_id)ipminimal gcs_addresscluster_id_hexrrr grpc_portmetrics_export_portnode_manager_portevents_export_addr listen_portr r osenvironnode_idZmetrics_collection_disabledrserver http_serverr gcs_client_init_non_minimal)selfnode_ip_addressrrrrrrrrr r r rrrrr%_/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/dashboard/agent.py__init__s8    zDashboardAgent.__init__cCs4ddlm}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).0mrEr%r& sz&DashboardAgent.run..)win32cygwincs$tdjdjd|dS)NzTerminated Raylet: ip=z , node_id=z. )r9r;rr)msgrEr%r&callbacksz$DashboardAgent.run..callbackcs tdIdHq)NTi)asynciosleepr%r%r%r& wait_foreversz(DashboardAgent.run..wait_forever)rstartrCr r8r9r: http_portrr!Zasync_internal_kv_putdashboard_constsZ#DASHBOARD_AGENT_ADDR_NODE_ID_PREFIXrencodejsondumpsr ray_constantsZKV_NAMESPACE_DASHBOARDZDASHBOARD_AGENT_ADDR_IP_PREFIXrSgatherr0platformrrloopr>Zwait_for_terminationcleanup) r#r@Zlaunch_http_servererWrZput_by_node_idZ put_by_iptasksrRZcheck_parent_taskrUr%rEr&rKs\       zDashboardAgent.run)r= __module__ __qualname__r\#DEFAULT_DASHBOARD_AGENT_LISTEN_PORTboolstrr'r"rCpropertyrDrHrKr%r%r%r&r s6   :;  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 --grpc-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)rkrldefaultrmz --listen-portFz!Port for HTTP server to listen onz --raylet-namez%The socket path of the raylet processz--logging-levelcCst|SrG)logging getLevelNameupper)sr%r%r&,srs)rkrlrnchoicesrmz--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]'`.)actionrmz--disable-metrics-collectionzCIf this arg is set, metrics report won't be enabled from the agent.z--session-namezThe current Ray session name.z--stdout-filepathz,The filepath to dump dashboard agent stdout.z--stderr-filepathz,The filepath to dump dashboard agent stderr.rO) logging_levellogging_formatrfilenameZ max_bytesZ backup_count) rrrrrrr r r rcCstdttjdS)Nz#Exiting with SIGTERM immediately...)r9warningr_exitsignalSIGTERMr%r%r%r&sigterm_handlers rz6Agent is working abnormally. It will exit immediately.)VargparserSrZrorr~r0Zray._private.ray_constantsZ_privater\Zray.dashboard.constsZ dashboardZconstsrXZray.dashboard.utilsutilsr<Zray._common.network_utilsrrZray._common.utilsrZ ray._privaterZray._private.process_watcherrrZray._private.ray_loggingrZ ray._rayletr getLoggerr=r9r ArgumentParserparser add_argumentrgintreZ LOGGER_LEVELZLOGGER_LEVEL_CHOICESZLOGGER_LEVEL_HELPZ LOGGER_FORMATZLOGGER_FORMAT_HELPZDASHBOARD_AGENT_LOG_FILENAMEformat parse_argsargsr^Zlogging_rotate_bytesZlogging_rotation_bytesZlogging_rotate_backup_countZlogging_rotation_backup_countryrzrZlogging_filenameZ redirect_stdout_stderr_if_neededZstdout_filepathZstderr_filepathr_r$rrrrrrrrrr r r rZagentradd_signal_handlerrrun_until_completerKr8r:exitr%r%r%r&s        [