From a0bbf2bfc26bc487d690261aa641ffe54be343ee Mon Sep 17 00:00:00 2001 From: Tao Wang Date: Tue, 5 Jan 2021 11:18:03 +0800 Subject: [PATCH] Notify listeners after registered node stored (#13069) --- src/ray/gcs/gcs_server/gcs_node_manager.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/ray/gcs/gcs_server/gcs_node_manager.cc b/src/ray/gcs/gcs_server/gcs_node_manager.cc index 3724bac5e..8f9da5b20 100644 --- a/src/ray/gcs/gcs_server/gcs_node_manager.cc +++ b/src/ray/gcs/gcs_server/gcs_node_manager.cc @@ -33,7 +33,6 @@ void GcsNodeManager::HandleRegisterNode(const rpc::RegisterNodeRequest &request, NodeID node_id = NodeID::FromBinary(request.node_info().node_id()); RAY_LOG(INFO) << "Registering node info, node id = " << node_id << ", address = " << request.node_info().node_manager_address(); - AddNode(std::make_shared(request.node_info())); auto on_done = [this, node_id, request, reply, send_reply_callback](const Status &status) { RAY_CHECK_OK(status); @@ -41,6 +40,7 @@ void GcsNodeManager::HandleRegisterNode(const rpc::RegisterNodeRequest &request, << ", address = " << request.node_info().node_manager_address(); RAY_CHECK_OK(gcs_pub_sub_->Publish(NODE_CHANNEL, node_id.Hex(), request.node_info().SerializeAsString(), nullptr)); + AddNode(std::make_shared(request.node_info())); GCS_RPC_SEND_REPLY(send_reply_callback, reply, status); }; RAY_CHECK_OK(