YARN-5131. Distributed shell AM fails when extra container arrives du…

…ring finishing. Contributed by Wangda Tan.
lliiun-z · May 25, 2016 · 48c9313 · 48c9313
1 parent dcbb700
commit 48c9313
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...src/main/java/org/apache/hadoop/yarn/applications/distributedshell/ApplicationMaster.java b/...src/main/java/org/apache/hadoop/yarn/applications/distributedshell/ApplicationMaster.java
@@ -745,8 +745,8 @@ protected boolean finish() {
     FinalApplicationStatus appStatus;
     String appMessage = null;
     boolean success = true;
-    if (numFailedContainers.get() == 0 && 
-        numCompletedContainers.get() == numTotalContainers) {
+    if (numCompletedContainers.get() - numFailedContainers.get()
+        >= numTotalContainers) {
       appStatus = FinalApplicationStatus.SUCCEEDED;
     } else {
       appStatus = FinalApplicationStatus.FAILED;