xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

serving support acl mutex

This commit is contained in:
xuyongfei 2020-09-29 18:56:02 +08:00
parent 8f55187492
commit 89c40ddc4c
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
serving/core/server.cc
View File

@ -77,6 +77,7 @@ grpc::Status CreatGRPCStatus(const Status &status) {
// Service Implement // Service Implement
class MSServiceImpl final : public MSService::Service { class MSServiceImpl final : public MSService::Service {
grpc::Status Predict(grpc::ServerContext *context, const PredictRequest *request, PredictReply *reply) override { grpc::Status Predict(grpc::ServerContext *context, const PredictRequest *request, PredictReply *reply) override {
std::lock_guard<std::mutex> lock(mutex_);
MSI_TIME_STAMP_START(Predict) MSI_TIME_STAMP_START(Predict)
auto res = Session::Instance().Predict(*request, *reply); auto res = Session::Instance().Predict(*request, *reply);
MSI_TIME_STAMP_END(Predict) MSI_TIME_STAMP_END(Predict)
@ -91,6 +92,7 @@ class MSServiceImpl final : public MSService::Service {
MSI_LOG(INFO) << "TestService call"; MSI_LOG(INFO) << "TestService call";
return grpc::Status::OK; return grpc::Status::OK;
} }
std::mutex mutex_;
}; };
static std::pair<struct evhttp *, struct event_base *> NewHttpServer() { static std::pair<struct evhttp *, struct event_base *> NewHttpServer() {