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

!3448 Fix the case cache server is stuck on an infinite loop when we get out of memory error 

Merge pull request !3448 from JesseKLee/infinite_loop
This commit is contained in:
mindspore-ci-bot 2020-07-25 03:00:07 +08:00 committed by Gitee
parent 06ed9ffd6a 73d4b00343
commit d8fcf269d6
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mindspore/ccsrc/minddata/dataset/util/service.cc
View File

@ -34,7 +34,18 @@ Status Service::ServiceStart() {
state_ = STATE::kStartInProg;
// At this point, we will let go of the lock. This allow others to proceed.
lck.Unlock();
RETURN_IF_NOT_OK(DoServiceStart());
// Call the real implementation from the derived class.
Status rc = DoServiceStart();
// If we hit any error, change the state back into the initial state.
// It is possible that the user may want to drive a clean up by calling
// ServiceStop but if it will end up in a loop because of the state is still
// kStartInProg.
if (rc.IsError()) {
lck.Lock();
state_ = STATE::kStopped;
lck.Unlock();
return rc;
}
// Lock again to change state.
lck.Lock();
state_ = STATE::kRunning;