mirror of
https://github.com/wassname/ray.git
synced 2026-07-04 04:44:43 +08:00
Tao Wang
67 lines
2.7 KiB
C++
67 lines
2.7 KiB
C++
// Copyright 2017 The Ray Authors.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include "stats_handler_impl.h"
|
|
|
|
namespace ray {
|
|
namespace rpc {
|
|
|
|
void DefaultStatsHandler::HandleAddProfileData(const AddProfileDataRequest &request,
|
|
AddProfileDataReply *reply,
|
|
SendReplyCallback send_reply_callback) {
|
|
ClientID node_id = ClientID::FromBinary(request.profile_data().component_id());
|
|
RAY_LOG(DEBUG) << "Adding profile data, component type = "
|
|
<< request.profile_data().component_type() << ", node id = " << node_id;
|
|
auto profile_table_data = std::make_shared<ProfileTableData>();
|
|
profile_table_data->CopyFrom(request.profile_data());
|
|
auto on_done = [node_id, request, reply, send_reply_callback](Status status) {
|
|
if (!status.ok()) {
|
|
RAY_LOG(ERROR) << "Failed to add profile data, component type = "
|
|
<< request.profile_data().component_type()
|
|
<< ", node id = " << node_id;
|
|
}
|
|
GCS_RPC_SEND_REPLY(send_reply_callback, reply, status);
|
|
};
|
|
|
|
Status status = gcs_table_storage_->ProfileTable().Put(UniqueID::FromRandom(),
|
|
*profile_table_data, on_done);
|
|
if (!status.ok()) {
|
|
on_done(status);
|
|
}
|
|
RAY_LOG(DEBUG) << "Finished adding profile data, component type = "
|
|
<< request.profile_data().component_type() << ", node id = " << node_id;
|
|
}
|
|
|
|
void DefaultStatsHandler::HandleGetAllProfileInfo(
|
|
const rpc::GetAllProfileInfoRequest &request, rpc::GetAllProfileInfoReply *reply,
|
|
rpc::SendReplyCallback send_reply_callback) {
|
|
RAY_LOG(DEBUG) << "Getting all profile info.";
|
|
auto on_done = [reply, send_reply_callback](
|
|
const std::unordered_map<UniqueID, ProfileTableData> &result) {
|
|
for (auto &data : result) {
|
|
reply->add_profile_info_list()->CopyFrom(data.second);
|
|
}
|
|
RAY_LOG(DEBUG) << "Finished getting all profile info.";
|
|
GCS_RPC_SEND_REPLY(send_reply_callback, reply, Status::OK());
|
|
};
|
|
|
|
Status status = gcs_table_storage_->ProfileTable().GetAll(on_done);
|
|
if (!status.ok()) {
|
|
on_done(std::unordered_map<UniqueID, ProfileTableData>());
|
|
}
|
|
}
|
|
|
|
} // namespace rpc
|
|
} // namespace ray
|