put it in a seperate case

WangTaoTheTonic · WangTaoTheTonic · commit 27203d8a5d6a · 2016-03-14T22:07:52.000+08:00
diff --git a/yarn/src/main/scala/org/apache/spark/deploy/yarn/Client.scala b/yarn/src/main/scala/org/apache/spark/deploy/yarn/Client.scala
@@ -975,12 +975,11 @@ private[spark] class Client(
           case e: ApplicationNotFoundException =>
             logError(s"Application $appId not found.")
             return (YarnApplicationState.KILLED, FinalApplicationStatus.KILLED)
+          case e: Exception if (e.isInstanceOf[InterruptedException]
+            || e.getCause.isInstanceOf[InterruptedException]) =>
+            logInfo("The reporter thread is interrupted, we assume app is finished.")
+            return (YarnApplicationState.FINISHED, FinalApplicationStatus.SUCCEEDED)
           case NonFatal(e) =>
-            if (e.isInstanceOf[InterruptedException]
-              || e.getCause.isInstanceOf[InterruptedException]) {
-              logInfo("The reporter thread is interrupted, we assume app is finished.")
-              return (YarnApplicationState.FINISHED, FinalApplicationStatus.SUCCEEDED)
-            }
             logError(s"Failed to contact YARN for application $appId.", e)
             return (YarnApplicationState.FAILED, FinalApplicationStatus.FAILED)
         }
