From 8516dd28e5e54ef7df439856da49b2b0dcf29d3a Mon Sep 17 00:00:00 2001 From: Koert Kuipers Date: Tue, 13 May 2014 19:44:35 -0400 Subject: [PATCH] SPARK-1801. expose InterruptibleIterator and TaskKilledException in developer api --- .../main/scala/org/apache/spark/InterruptibleIterator.scala | 5 ++++- .../main/scala/org/apache/spark/TaskKilledException.scala | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/core/src/main/scala/org/apache/spark/InterruptibleIterator.scala b/core/src/main/scala/org/apache/spark/InterruptibleIterator.scala index ec11dbbffaaf8..4786bfadb8966 100644 --- a/core/src/main/scala/org/apache/spark/InterruptibleIterator.scala +++ b/core/src/main/scala/org/apache/spark/InterruptibleIterator.scala @@ -17,11 +17,14 @@ package org.apache.spark +import org.apache.spark.annotation.DeveloperApi + /** * An iterator that wraps around an existing iterator to provide task killing functionality. * It works by checking the interrupted flag in [[TaskContext]]. */ -private[spark] class InterruptibleIterator[+T](val context: TaskContext, val delegate: Iterator[T]) +@DeveloperApi +class InterruptibleIterator[+T](val context: TaskContext, val delegate: Iterator[T]) extends Iterator[T] { def hasNext: Boolean = { diff --git a/core/src/main/scala/org/apache/spark/TaskKilledException.scala b/core/src/main/scala/org/apache/spark/TaskKilledException.scala index cbd6b2866e4f9..47cac63cb88c4 100644 --- a/core/src/main/scala/org/apache/spark/TaskKilledException.scala +++ b/core/src/main/scala/org/apache/spark/TaskKilledException.scala @@ -17,7 +17,10 @@ package org.apache.spark +import org.apache.spark.annotation.DeveloperApi + /** * Exception for a task getting killed. */ -private[spark] class TaskKilledException extends RuntimeException +@DeveloperApi +class TaskKilledException extends RuntimeException