YarnClientSchedulerBack.asyncMonitorApplication should be common with

Client.monitorApplication
apache · Apr 1, 2015 · 568f46f · 568f46f
1 parent 305abe1
commit 568f46f
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/yarn/src/main/scala/org/apache/spark/scheduler/cluster/YarnClientSchedulerBackend.scala b/yarn/src/main/scala/org/apache/spark/scheduler/cluster/YarnClientSchedulerBackend.scala
@@ -128,22 +128,14 @@ private[spark] class YarnClientSchedulerBackend(
     val t = new Thread {
       override def run() {
         while (!stopping) {
-          var state: YarnApplicationState = null
-          try {
-            val report = client.getApplicationReport(appId)
-            state = report.getYarnApplicationState()
-          } catch {
-            case e: ApplicationNotFoundException =>
-              state = YarnApplicationState.KILLED
-          }
+          val (state, _) = client.monitorApplication(appId, logApplicationReport = false)
           if (state == YarnApplicationState.FINISHED ||
             state == YarnApplicationState.KILLED ||
             state == YarnApplicationState.FAILED) {
             logError(s"Yarn application has already exited with state $state!")
             sc.stop()
             stopping = true
           }
-          Thread.sleep(1000L)
         }
         Thread.currentThread().interrupt()
       }