spark-instrumented-optimizer/core/src/main/scala/spark/FetchFailedException.scala

package spark

class FetchFailedException(
    val serverUri: String,
    val shuffleId: Int,
    val mapId: Int,
    val reduceId: Int,
    cause: Throwable)
  extends Exception {
  
  override def getMessage(): String = 
    "Fetch failed: %s %d %d %d".format(serverUri, shuffleId, mapId, reduceId)

  override def getCause(): Throwable = cause

  def toTaskEndReason: TaskEndReason = FetchFailed(serverUri, shuffleId, mapId, reduceId)
}