commit 8cd0f0c9b31944d892972178365871b8134b1619
author huafengw <fvunicorn@gmail.com> Fri Feb 24 19:37:32 2017 +0800
committer manuzhang <owenzhang1990@gmail.com> Fri Feb 24 19:37:49 2017 +0800
tree c17addaecda5a2d4b2591dc8a0d41ed384f8056c
parent 519a195004371d05a7b0a54e1e7b8291038c09f8

[GEARPUMP-282] Allow user to configure wait duration for a running app

Author: huafengw <fvunicorn@gmail.com>

Closes #160 from huafengw/fix282.

core/src/main/scala/org/apache/gearpump/cluster/client/RunningApplication.scala

diff --git a/core/src/main/scala/org/apache/gearpump/cluster/client/RunningApplication.scala b/core/src/main/scala/org/apache/gearpump/cluster/client/RunningApplication.scala
index d62356a..103df01 100644
--- a/core/src/main/scala/org/apache/gearpump/cluster/client/RunningApplication.scala
+++ b/core/src/main/scala/org/apache/gearpump/cluster/client/RunningApplication.scala
@@ -25,6 +25,8 @@
 import org.apache.gearpump.cluster.client.RunningApplication._
 import org.apache.gearpump.util.{ActorUtil, LogUtil}
 import org.slf4j.Logger
+import java.time.Duration
+import java.util.concurrent.TimeUnit
 
 import scala.concurrent.ExecutionContext.Implicits.global
 import scala.concurrent.Future
@@ -48,8 +50,12 @@
    * If failed, an exception will be thrown out
    */
   def waitUntilFinish(): Unit = {
+    this.waitUntilFinish(INF_DURATION)
+  }
+
+  def waitUntilFinish(duration: Duration): Unit = {
     val result = ActorUtil.askActor[ApplicationResult](master,
-      RegisterAppResultListener(appId), INF_TIMEOUT)
+      RegisterAppResultListener(appId), new Timeout(duration.getSeconds, TimeUnit.SECONDS))
     result match {
       case failed: ApplicationFailed =>
         throw failed.error
@@ -71,6 +77,6 @@
 object RunningApplication {
   private val LOG: Logger = LogUtil.getLogger(getClass)
   // This magic number is derived from Akka's configuration, which is the maximum delay
-  private val INF_TIMEOUT = new Timeout(2147482 seconds)
+  private val INF_DURATION = Duration.ofSeconds(2147482)
 }