Skip to content

Commit d697e8d

Browse files
authored
fix: Correctly Guard List Access (#8563)
1 parent 8a23c8c commit d697e8d

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

src/sagemaker_server.cc

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -182,10 +182,13 @@ SagemakerAPIServer::Handle(evhtp_request_t* req)
182182
if (action == "/invoke") {
183183
LOG_VERBOSE(1) << "SageMaker request: INVOKE MODEL";
184184

185-
if (sagemaker_models_list_.find(multi_model_name.c_str()) ==
186-
sagemaker_models_list_.end()) {
187-
evhtp_send_reply(req, EVHTP_RES_NOTFOUND); /* 404*/
188-
return;
185+
{
186+
std::lock_guard<std::mutex> lock(models_list_mutex_);
187+
if (sagemaker_models_list_.find(multi_model_name.c_str()) ==
188+
sagemaker_models_list_.end()) {
189+
evhtp_send_reply(req, EVHTP_RES_NOTFOUND); /* 404*/
190+
return;
191+
}
189192
}
190193
LOG_VERBOSE(1) << "SageMaker MME Custom Invoke Model Path";
191194

0 commit comments

Comments
 (0)