Fix client to await the driver pod (#325)

kimoonkim · erikerlandson · commit bb1b234084c6 · 2017-06-02T16:08:10.000-07:00
diff --git a/resource-managers/kubernetes/core/src/main/scala/org/apache/spark/deploy/kubernetes/submit/LoggingPodStatusWatcher.scala b/resource-managers/kubernetes/core/src/main/scala/org/apache/spark/deploy/kubernetes/submit/LoggingPodStatusWatcher.scala
@@ -137,7 +137,7 @@ private[kubernetes] class LoggingPodStatusWatcherImpl(
   }
 
   override def awaitCompletion(): Unit = {
-    podCompletedFuture.countDown()
+    podCompletedFuture.await()
     logInfo(pod.map { p =>
       s"Container final statuses:\n\n${containersDescription(p)}"
     }.getOrElse("No containers were found in the driver pod."))

Original file line number	Diff line number	Diff line change
`@@ -137,7 +137,7 @@ private[kubernetes] class LoggingPodStatusWatcherImpl(`
`137`	`137`	`}`
`138`	`138`
`139`	`139`	`override def awaitCompletion(): Unit = {`
`140`		`- podCompletedFuture.countDown()`
	`140`	`+ podCompletedFuture.await()`
`141`	`141`	`logInfo(pod.map { p =>`
`142`	`142`	`s"Container final statuses:\n\n${containersDescription(p)}"`
`143`	`143`	`}.getOrElse("No containers were found in the driver pod."))`