Set e as the cause of FetchFailedException

apache · Nov 2, 2014 · d51b0b6 · d51b0b6
1 parent b88c919
commit d51b0b6
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/core/src/main/scala/org/apache/spark/shuffle/FetchFailedException.scala b/core/src/main/scala/org/apache/spark/shuffle/FetchFailedException.scala
@@ -33,16 +33,18 @@ private[spark] class FetchFailedException(
     mapId: Int,
     reduceId: Int,
     message: String)
-  extends Exception {
+  extends Exception(message) {
 
   def this(bmAddress: BlockManagerId, shuffleId: Int, mapId: Int, reduceId: Int, e: Throwable) {
-    this(bmAddress, shuffleId, mapId, reduceId, Utils.exceptionString(e))
+    this(bmAddress, shuffleId, mapId, reduceId, e.getMessage)
+    initCause(e)
   }
 
   override def getMessage: String =
     "Fetch failed: %s %d %d %d".format(bmAddress, shuffleId, mapId, reduceId)
 
-  def toTaskEndReason: TaskEndReason = FetchFailed(bmAddress, shuffleId, mapId, reduceId, message)
+  def toTaskEndReason: TaskEndReason = FetchFailed(bmAddress, shuffleId, mapId, reduceId,
+    Utils.exceptionString(this))
 }
 
 /**