commit a37a367b400dd5b1cab22b35a47fb664055f3df6
author Gopal V <gopalv@apache.org> Mon Aug 27 20:37:27 2018 -0700
committer Gopal V <gopalv@apache.org> Mon Aug 27 20:37:27 2018 -0700
tree 5ccd857d9619bcfcefba40f407991f047cc65ed7
parent 5949b0cef6e9b23c1ac5c75461c027b40cafc91d

TEZ-3980: ShuffleRunner: the wake loop needs to check for shutdown (Gopal V, reviewed by Gunther Hagleitner)

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 @@
               && numCompletedInputs.get() < numInputs) {
             inputContext.notifyProgress();
             boolean ret = wakeLoop.await(1000, TimeUnit.MILLISECONDS);
+            if (isShutdown.get()) {
+              break;
+            }
           }
         } finally {
           lock.unlock();