Implement the APIs for gRPC calls in client/server
Implement the APIs defined in the RdcStandaloneHandler to make gRPC call to daemon Implement the APIs defined in the RdcAPIServiceImpl to handle the gRPC calls in daemon Add two APIs to get all GPU groups and field groups: rdc_group_get_all_ids() and rdc_group_field_all_ids() Those two APIs are required by the rdci group and fieldgroup sub-modules. Change-Id: I066091423146dea180c16af212688ed43dc44611
이 커밋은 다음에 포함됨:
Bill(Shuzhou) Liu
Chris Freehill
+319
-2
@@ -70,7 +70,7 @@ RdcAPIServiceImpl::~RdcAPIServiceImpl() {
|
||||
}
|
||||
uint32_t gpu_index_list[RDC_MAX_NUM_DEVICES];
|
||||
uint32_t count = 0;
|
||||
rdc_status_t result = rdc_get_all_devices(rdc_handle_,
|
||||
rdc_status_t result = rdc_device_get_all(rdc_handle_,
|
||||
gpu_index_list, &count);
|
||||
reply->set_status(result);
|
||||
if (result != RDC_ST_OK) {
|
||||
@@ -93,7 +93,7 @@ RdcAPIServiceImpl::~RdcAPIServiceImpl() {
|
||||
}
|
||||
uint32_t gpu_index = request->gpu_index();
|
||||
rdc_device_attributes_t attribute;
|
||||
rdc_status_t result = rdc_get_device_attributes(rdc_handle_,
|
||||
rdc_status_t result = rdc_device_get_attributes(rdc_handle_,
|
||||
gpu_index, &attribute);
|
||||
|
||||
::rdc::DeviceAttributes* attr = reply->mutable_attributes();
|
||||
@@ -104,6 +104,323 @@ RdcAPIServiceImpl::~RdcAPIServiceImpl() {
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::CreateGpuGroup(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::CreateGpuGroupRequest* request,
|
||||
::rdc::CreateGpuGroupResponse* reply) {
|
||||
(void)(context);
|
||||
if (!reply || !request) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_gpu_group_t group_id = 0;
|
||||
rdc_status_t result = rdc_group_gpu_create(rdc_handle_,
|
||||
static_cast<rdc_group_type_t>(request->type()),
|
||||
request->group_name().c_str(), &group_id);
|
||||
reply->set_status(result);
|
||||
if (result != RDC_ST_OK) {
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
reply->set_group_id(group_id);
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::AddToGpuGroup(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::AddToGpuGroupRequest* request,
|
||||
::rdc::AddToGpuGroupResponse* reply) {
|
||||
(void)(context);
|
||||
if (!reply || !request) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_status_t result = rdc_group_gpu_add(rdc_handle_,
|
||||
request->group_id(), request->gpu_index());
|
||||
reply->set_status(result);
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::GetGpuGroupInfo(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::GetGpuGroupInfoRequest* request,
|
||||
::rdc::GetGpuGroupInfoResponse* reply) {
|
||||
(void)(context);
|
||||
if (!reply || !request) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_group_info_t group_info;
|
||||
rdc_status_t result = rdc_group_gpu_get_info(
|
||||
rdc_handle_, request->group_id(), &group_info);
|
||||
reply->set_status(result);
|
||||
if (result != RDC_ST_OK) {
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
reply->set_group_name(group_info.group_name);
|
||||
for (uint32_t i = 0; i < group_info.count; i++) {
|
||||
reply->add_entity_ids(group_info.entity_ids[i]);
|
||||
}
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::GetGroupAllIds(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::Empty* request,
|
||||
::rdc::GetGroupAllIdsResponse* reply) {
|
||||
if (!reply || !request || !context) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_gpu_group_t group_id_list[RDC_MAX_NUM_GROUPS];
|
||||
uint32_t count = 0;
|
||||
rdc_status_t result = rdc_group_get_all_ids(
|
||||
rdc_handle_, group_id_list, &count);
|
||||
reply->set_status(result);
|
||||
if (result != RDC_ST_OK) {
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
for (uint32_t i = 0; i < count; i++) {
|
||||
reply->add_group_ids(group_id_list[i]);
|
||||
}
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::GetFieldGroupAllIds(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::Empty* request,
|
||||
::rdc::GetFieldGroupAllIdsResponse* reply) {
|
||||
if (!reply || !request || !context) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_field_grp_t field_group_id_list[RDC_MAX_NUM_FIELD_GROUPS];
|
||||
uint32_t count = 0;
|
||||
rdc_status_t result = rdc_group_field_get_all_ids(
|
||||
rdc_handle_, field_group_id_list, &count);
|
||||
reply->set_status(result);
|
||||
if (result != RDC_ST_OK) {
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
for (uint32_t i = 0; i < count; i++) {
|
||||
reply->add_field_group_ids(field_group_id_list[i]);
|
||||
}
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::DestroyGpuGroup(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::DestroyGpuGroupRequest* request,
|
||||
::rdc::DestroyGpuGroupResponse* reply) {
|
||||
(void)(context);
|
||||
if (!reply || !request) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_status_t result = rdc_group_gpu_destroy(
|
||||
rdc_handle_, request->group_id());
|
||||
reply->set_status(result);
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::CreateFieldGroup(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::CreateFieldGroupRequest* request,
|
||||
::rdc::CreateFieldGroupResponse* reply) {
|
||||
(void)(context);
|
||||
if (!reply || !request) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_field_grp_t field_group_id;
|
||||
uint32_t field_ids[RDC_MAX_FIELD_IDS_PER_FIELD_GROUP];
|
||||
for (int i = 0; i < request->field_ids_size(); i++) {
|
||||
field_ids[i] = request->field_ids(i);
|
||||
}
|
||||
rdc_status_t result = rdc_group_field_create(
|
||||
rdc_handle_, request->field_ids_size() , &field_ids[0],
|
||||
request->field_group_name().c_str(), &field_group_id);
|
||||
reply->set_status(result);
|
||||
if (result != RDC_ST_OK) {
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
reply->set_field_group_id(field_group_id);
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::GetFieldGroupInfo(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::GetFieldGroupInfoRequest* request,
|
||||
::rdc::GetFieldGroupInfoResponse* reply) {
|
||||
(void)(context);
|
||||
if (!reply || !request) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_field_group_info_t field_info;
|
||||
rdc_status_t result = rdc_group_field_get_info(
|
||||
rdc_handle_, request->field_group_id(), &field_info);
|
||||
reply->set_status(result);
|
||||
if (result != RDC_ST_OK) {
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
reply->set_filed_group_name(field_info.group_name);
|
||||
for (uint32_t i = 0; i < field_info.count; i++) {
|
||||
reply->add_field_ids(field_info.field_ids[i]);
|
||||
}
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::DestroyFieldGroup(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::DestroyFieldGroupRequest* request,
|
||||
::rdc::DestroyFieldGroupResponse* reply) {
|
||||
(void)(context);
|
||||
if (!reply || !request) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_status_t result = rdc_group_field_destroy(
|
||||
rdc_handle_, request->field_group_id());
|
||||
reply->set_status(result);
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::WatchFields(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::WatchFieldsRequest* request,
|
||||
::rdc::WatchFieldsResponse* reply) {
|
||||
(void)(context);
|
||||
if (!reply || !request) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_status_t result = rdc_field_watch(
|
||||
rdc_handle_, request->group_id(), request->field_group_id(),
|
||||
request->update_freq(), request->max_keep_age(),
|
||||
request->max_keep_samples());
|
||||
reply->set_status(result);
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::GetLatestFieldValue(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::GetLatestFieldValueRequest* request,
|
||||
::rdc::GetLatestFieldValueResponse* reply) {
|
||||
(void)(context);
|
||||
if (!reply || !request) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_field_value value;
|
||||
rdc_status_t result = rdc_field_get_latest_value(
|
||||
rdc_handle_, request->gpu_index(), request->field_id(), &value);
|
||||
reply->set_status(result);
|
||||
if (result != RDC_ST_OK) {
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
reply->set_field_id(value.field_id);
|
||||
reply->set_rdc_status(value.status);
|
||||
reply->set_ts(value.ts);
|
||||
reply->set_type(static_cast<::rdc::GetLatestFieldValueResponse_FieldType>
|
||||
(value.type));
|
||||
if (value.type == INTEGER) {
|
||||
reply->set_l_int(value.value.l_int);
|
||||
} else if (value.type == DOUBLE) {
|
||||
reply->set_dbl(value.value.dbl);
|
||||
} else if (value.type == STRING || value.type == BLOB) {
|
||||
reply->set_str(value.value.str);
|
||||
}
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::GetFieldSince(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::GetFieldSinceRequest* request,
|
||||
::rdc::GetFieldSinceResponse* reply) {
|
||||
(void)(context);
|
||||
if (!reply || !request) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_field_value value;
|
||||
uint64_t next_timestamp;
|
||||
rdc_status_t result = rdc_field_get_value_since(
|
||||
rdc_handle_, request->gpu_index(), request->field_id(),
|
||||
request->since_time_stamp(), &next_timestamp, &value);
|
||||
reply->set_status(result);
|
||||
if (result != RDC_ST_OK) {
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
reply->set_next_since_time_stamp(next_timestamp);
|
||||
reply->set_field_id(value.field_id);
|
||||
reply->set_rdc_status(value.status);
|
||||
reply->set_ts(value.ts);
|
||||
reply->set_type(static_cast<::rdc::GetFieldSinceResponse_FieldType>
|
||||
(value.type));
|
||||
if (value.type == INTEGER) {
|
||||
reply->set_l_int(value.value.l_int);
|
||||
} else if (value.type == DOUBLE) {
|
||||
reply->set_dbl(value.value.dbl);
|
||||
} else if (value.type == STRING || value.type == BLOB) {
|
||||
reply->set_str(value.value.str);
|
||||
}
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::UnWatchFields(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::UnWatchFieldsRequest* request,
|
||||
::rdc::UnWatchFieldsResponse* reply) {
|
||||
(void)(context);
|
||||
if (!reply || !request) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_status_t result = rdc_field_unwatch(
|
||||
rdc_handle_, request->group_id(), request->field_group_id());
|
||||
reply->set_status(result);
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
::grpc::Status RdcAPIServiceImpl::UpdateAllFields(
|
||||
::grpc::ServerContext* context,
|
||||
const ::rdc::UpdateAllFieldsRequest* request,
|
||||
::rdc::UpdateAllFieldsResponse* reply) {
|
||||
(void)(context);
|
||||
if (!reply || !request) {
|
||||
return ::grpc::Status(::grpc::StatusCode::INTERNAL, "Empty contents");
|
||||
}
|
||||
|
||||
rdc_status_t result = rdc_field_update_all(
|
||||
rdc_handle_, request->wait_for_update());
|
||||
reply->set_status(result);
|
||||
|
||||
return ::grpc::Status::OK;
|
||||
}
|
||||
|
||||
|
||||
} // namespace rdc
|
||||
} // namespace amd
|
||||
|
||||
|
||||
새 이슈에서 참조
사용자 차단