From 9bffc662796734791c0323a21f9ec6ea84236ef9 Mon Sep 17 00:00:00 2001
From: xuyongfei <xuyongfei@huawei.com>
Date: Tue, 20 Apr 2021 10:27:46 +0800
Subject: [PATCH] Serving, fix bug, in some cases, failed to find the user's
 context information, resulting in no response

---
 mindspore_serving/ccsrc/worker/work_executor.cc | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mindspore_serving/ccsrc/worker/work_executor.cc b/mindspore_serving/ccsrc/worker/work_executor.cc
index 3b36b6a..b74b19d 100644
--- a/mindspore_serving/ccsrc/worker/work_executor.cc
+++ b/mindspore_serving/ccsrc/worker/work_executor.cc
@@ -251,7 +251,10 @@ Status WorkExecutor::Work(const RequestSpec &request_spec, const std::vector<Ins
     context.user_context = user_context;
   }
   infer_session.instances = instances;
-  infer_session_map_[user_id] = infer_session;
+  {
+    std::unique_lock<std::mutex> lock(infer_session_map_mutex_);
+    infer_session_map_[user_id] = infer_session;
+  }
 
   if (!method_def.preprocess_name.empty()) {
     OnReceivePreprocessInputs(instances);