From 60efa73998aad749447271cb98b267b297cc5b51 Mon Sep 17 00:00:00 2001
From: Cedric Beust <cedric@beust.com>
Date: Sat, 24 Oct 2015 15:47:26 -0700
Subject: [PATCH] Shut down the executor in case of an exception.

---
 .../kotlin/com/beust/kobalt/internal/DynamicGraph.kt  | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/main/kotlin/com/beust/kobalt/internal/DynamicGraph.kt b/src/main/kotlin/com/beust/kobalt/internal/DynamicGraph.kt
index 563a9ae0..a98283c2 100644
--- a/src/main/kotlin/com/beust/kobalt/internal/DynamicGraph.kt
+++ b/src/main/kotlin/com/beust/kobalt/internal/DynamicGraph.kt
@@ -1,9 +1,10 @@
 package com.beust.kobalt.internal
 
+import com.beust.kobalt.misc.KobaltLogger
 import com.beust.kobalt.misc.NamedThreadFactory
 import com.beust.kobalt.misc.ToString
 import com.beust.kobalt.misc.log
-import com.google.common.collect.ArrayListMultimap
+import com.beust.kobalt.misc.KobaltLogger.*
 import com.google.common.collect.HashMultimap
 import java.util.concurrent.*
 
@@ -45,7 +46,8 @@ public class DynamicGraphExecutor<T>(val graph: DynamicGraph<T>,
      */
     public fun run() : Int {
         var lastResult = TaskResult()
-        while (graph.freeNodes.size > 0) {
+        var gotError = false
+        while (graph.freeNodes.size > 0 && ! gotError) {
             log(3, "Current count: ${graph.nodeCount}")
             synchronized(graph) {
                 val freeNodes = graph.freeNodes
@@ -56,7 +58,7 @@ public class DynamicGraphExecutor<T>(val graph: DynamicGraph<T>,
                 var n = callables.size
 
                 // When a callable ends, see if it freed a node. If not, keep looping
-                while (n > 0 && graph.freeNodes.size == 0) {
+                while (n > 0 && graph.freeNodes.size == 0 && ! gotError) {
                     try {
                         val future = completion.take()
                         val taskResult = future.get(2, TimeUnit.SECONDS)
@@ -71,6 +73,9 @@ public class DynamicGraphExecutor<T>(val graph: DynamicGraph<T>,
                             })
                     } catch(ex: TimeoutException) {
                         log(2, "Time out")
+                    } catch(ex: Exception) {
+                        log(1, "Tests failed: ${ex.message}")
+                        gotError = true
                     }
                 }
             }