TEZ-3980: ShuffleRunner: the wake loop needs to check for shutdown (Gopal V, reviewed by Gunther Hagleitner)
Project: http://git-wip-us.apache.org/repos/asf/tez/repo Commit: http://git-wip-us.apache.org/repos/asf/tez/commit/a37a367b Tree: http://git-wip-us.apache.org/repos/asf/tez/tree/a37a367b Diff: http://git-wip-us.apache.org/repos/asf/tez/diff/a37a367b Branch: refs/heads/master Commit: a37a367b400dd5b1cab22b35a47fb664055f3df6 Parents: 5949b0c Author: Gopal V <[email protected]> Authored: Mon Aug 27 20:37:27 2018 -0700 Committer: Gopal V <[email protected]> Committed: Mon Aug 27 20:37:27 2018 -0700 ---------------------------------------------------------------------- .../tez/runtime/library/common/shuffle/impl/ShuffleManager.java | 3 +++ 1 file changed, 3 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/tez/blob/a37a367b/tez-runtime-library/src/main/java/org/apache/tez/runtime/library/common/shuffle/impl/ShuffleManager.java ---------------------------------------------------------------------- diff --git a/tez-runtime-library/src/main/java/org/apache/tez/runtime/library/common/shuffle/impl/ShuffleManager.java b/tez-runtime-library/src/main/java/org/apache/tez/runtime/library/common/shuffle/impl/ShuffleManager.java index 0a0286e..5f3693f 100644 --- a/tez-runtime-library/src/main/java/org/apache/tez/runtime/library/common/shuffle/impl/ShuffleManager.java +++ b/tez-runtime-library/src/main/java/org/apache/tez/runtime/library/common/shuffle/impl/ShuffleManager.java @@ -325,6 +325,9 @@ public class ShuffleManager implements FetcherCallback { && numCompletedInputs.get() < numInputs) { inputContext.notifyProgress(); boolean ret = wakeLoop.await(1000, TimeUnit.MILLISECONDS); + if (isShutdown.get()) { + break; + } } } finally { lock.unlock();
