[Core] Put raylet ip's in resource usage report (#13871)

* .

* done?

Co-authored-by: Alex Wu <alex@anyscale.com>
This commit is contained in:
Alex Wu
2021-02-03 11:28:56 -08:00
committed by Alex Wu
parent b4d87b8fc5
commit 34e0dfe934
3 changed files with 29 additions and 0 deletions
+26
View File
@@ -7,6 +7,7 @@ import time
import ray
import ray.ray_constants
import ray.services
import ray.test_utils
from ray._raylet import GlobalStateAccessor
@@ -332,6 +333,31 @@ def test_backlog_report(shutdown_only):
global_state_accessor.disconnect()
def test_heartbeat_ip(shutdown_only):
cluster = ray.init(
num_cpus=1, _system_config={
"report_worker_backlog": True,
})
global_state_accessor = GlobalStateAccessor(
cluster["redis_address"], ray.ray_constants.REDIS_DEFAULT_PASSWORD)
global_state_accessor.connect()
self_ip = ray.services.get_node_ip_address()
def self_ip_is_set():
message = global_state_accessor.get_all_resource_usage()
if message is None:
return False
resource_usage = ray.gcs_utils.ResourceUsageBatchData.FromString(
message)
resources_data = resource_usage.batch[0]
return resources_data.node_manager_address == self_ip
ray.test_utils.wait_for_condition(self_ip_is_set, timeout=2)
global_state_accessor.disconnect()
if __name__ == "__main__":
import pytest
import sys