[SPARK-4787] Stop context properly if an exception occurs during DAGS…

…cheduler initialization.
markhamstra · Dec 26, 2014 · 2172578 · 2172578
1 parent 29fabb1
commit 2172578
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/core/src/main/scala/org/apache/spark/SparkContext.scala b/core/src/main/scala/org/apache/spark/SparkContext.scala
@@ -329,8 +329,11 @@ class SparkContext(config: SparkConf) extends Logging with ExecutorAllocationCli
   try {
     dagScheduler = new DAGScheduler(this)
   } catch {
-    case e: Exception => throw
-      new SparkException("DAGScheduler cannot be initialized due to %s".format(e.getMessage))
+    case e: Exception => {
+      stop()
+      throw
+        new SparkException("DAGScheduler cannot be initialized due to %s".format(e.getMessage))
+    }
   }
 
   // start TaskScheduler after taskScheduler sets DAGScheduler reference in DAGScheduler's