Kinesin/daggy
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Unlocking unique lock to prevent one bad agent stalling the rest

Browse Source
This commit is contained in:
Ian Roddis
2022-02-11 13:06:20 -04:00
parent 6aab7be0a1
commit 443bc789cf
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libdaggy/src/executors/task/DaggyRunnerTaskExecutor.cpp
View File

@@ -171,6 +171,8 @@ TaskFuture DaggyRunnerTaskExecutor::execute(DAGRunID runID,
exe_capacity->cores -= taskUsed.cores; exe_capacity->cores -= taskUsed.cores;
exe_capacity->memoryMB -= taskUsed.memoryMB; exe_capacity->memoryMB -= taskUsed.memoryMB;
lock.unlock();
std::stringstream ss; std::stringstream ss;
ss << exe_runner << "/v1/task/" << runID << "/" << taskName; ss << exe_runner << "/v1/task/" << runID << "/" << taskName;
auto url = ss.str(); auto url = ss.str();
@@ -194,7 +196,7 @@ TaskFuture DaggyRunnerTaskExecutor::execute(DAGRunID runID,
auto fut = rt.fut; auto fut = rt.fut;
{ {
std::lock_guard<std::mutex> lock(rtGuard_); std::lock_guard<std::mutex> rtLock(rtGuard_);
runningTasks_.emplace(std::make_pair(runID, taskName), std::move(rt)); runningTasks_.emplace(std::make_pair(runID, taskName), std::move(rt));
} }
return fut; return fut;