Browse Source

serving pylint and cpplint 1202-2

tags/v1.1.0
xuyongfei 5 years ago
parent
commit
64ef9edea1
4 changed files with 16 additions and 16 deletions
  1. +1
    -1
      mindspore_serving/ccsrc/common/tensor.cc
  2. +8
    -8
      mindspore_serving/ccsrc/worker/worker.cc
  3. +1
    -1
      mindspore_serving/ccsrc/worker/worker.h
  4. +6
    -6
      mindspore_serving/client/cpp/client.cc

+ 1
- 1
mindspore_serving/ccsrc/common/tensor.cc View File

@@ -14,10 +14,10 @@
* limitations under the License.
*/
#include "common/tensor.h"
#include <securec.h>
#include <functional>
#include <utility>
#include "common/log.h"
#include "securec.h"

namespace mindspore::serving {



+ 8
- 8
mindspore_serving/ccsrc/worker/worker.cc View File

@@ -152,7 +152,7 @@ Status Worker::Run(const RequestSpec &request_spec, const std::vector<serving::I
auto result = result_pair.second;
while (result->HasNext()) {
serving::Instance instance;
result->GetNext(instance);
result->GetNext(&instance);
outputs->push_back(instance);
}
return SUCCESS;
@@ -395,8 +395,7 @@ void Worker::GetVersions(const LoadServableSpec &servable_spec, std::vector<uint
if (dir == "__pycache__") continue;
auto version_parse = trans_to_integer(dir);
if (version_parse == 0) {
if (ignore_dir.count(servable_spec.servable_directory + dir) == 0) {
ignore_dir.emplace(servable_spec.servable_directory + dir);
if (ignore_dir.emplace(servable_spec.servable_directory + dir).second) {
MSI_LOG_INFO << "Ignore directory " << dir << ", model_directory " << servable_spec.servable_directory
<< ", model_name " << servable_spec.servable_name;
}
@@ -488,7 +487,8 @@ AsyncResult::AsyncResult(size_t size) : result_(size), next_index_(0) {}

bool AsyncResult::HasNext() { return next_index_ < future_list_.size(); }

Status AsyncResult::GetNext(Instance &instance_result) {
Status AsyncResult::GetNext(Instance *instance_result) {
MSI_EXCEPTION_IF_NULL(instance_result);
if (next_index_ >= future_list_.size()) {
MSI_LOG_ERROR << "GetNext failed, index greater than instance count " << future_list_.size();
return FAILED;
@@ -497,14 +497,14 @@ Status AsyncResult::GetNext(Instance &instance_result) {
next_index_++;
auto &future = future_list_[index];
if (!future.valid()) {
instance_result.error_msg = result_[index].error_msg;
instance_result->error_msg = result_[index].error_msg;
return FAILED;
}
const int kWaitMaxHundredMs = 100;
int i;
for (i = 0; i < kWaitMaxHundredMs; i++) { //
if (Worker::GetInstance().HasCleared()) {
instance_result.error_msg = Status(FAILED, "Servable stopped");
instance_result->error_msg = Status(FAILED, "Servable stopped");
return FAILED;
}
if (future.wait_for(std::chrono::milliseconds(100)) == std::future_status::ready) {
@@ -513,12 +513,12 @@ Status AsyncResult::GetNext(Instance &instance_result) {
}
if (i >= kWaitMaxHundredMs) {
MSI_LOG_ERROR << "GetNext failed, wait time out, index " << index << ", total count " << future_list_.size();
instance_result.error_msg = Status(FAILED, "Time out");
instance_result->error_msg = Status(FAILED, "Time out");
return FAILED;
}

future.get();
instance_result = result_[index];
*instance_result = result_[index];
return SUCCESS;
}



+ 1
- 1
mindspore_serving/ccsrc/worker/worker.h View File

@@ -38,7 +38,7 @@ class AsyncResult {
explicit AsyncResult(size_t size);

bool HasNext();
Status GetNext(Instance &instance_result);
Status GetNext(Instance *instance_result);

private:
std::vector<std::future<void>> future_list_;


+ 6
- 6
mindspore_serving/client/cpp/client.cc View File

@@ -420,11 +420,14 @@ class ClientImpl {
auto channel = grpc::CreateChannel(target_str, grpc::InsecureChannelCredentials());
stub_ = proto::MSService::NewStub(channel);
}
Status Predict(const proto::PredictRequest &request, proto::PredictReply &reply) {
Status Predict(const proto::PredictRequest &request, proto::PredictReply *reply) {
if (reply == nullptr) {
return Status(SYSTEM_ERROR, "ClientImpl::Predict input reply cannot be nullptr");
}
grpc::ClientContext context;

// The actual RPC.
grpc::Status status = stub_->Predict(&context, request, &reply);
grpc::Status status = stub_->Predict(&context, request, reply);
if (status.ok()) {
return SUCCESS;
} else {
@@ -457,10 +460,7 @@ Status Client::SendRequest(const InstancesRequest &request, InstancesReply *repl
servable_spec->set_method_name(method_name_);
servable_spec->set_version_number(version_number_);

Status result = impl_->Predict(*proto_request, *proto_reply);
// std::string str;
// google::protobuf::TextFormat::PrintToString(*proto_reply, &str);
// std::cout << str << std::endl;
Status result = impl_->Predict(*proto_request, proto_reply);
return result;
}



Loading…
Cancel
Save