[SPARK-17147][STREAMING][KAFKA] change another assert to NoSuchElement

apache · Feb 21, 2018 · 08f3570 · 08f3570
1 parent 2246750
commit 08f3570
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/external/kafka-0-10/src/main/scala/org/apache/spark/streaming/kafka010/KafkaRDD.scala b/external/kafka-0-10/src/main/scala/org/apache/spark/streaming/kafka010/KafkaRDD.scala
@@ -306,7 +306,9 @@ private class CompactedKafkaRDDIterator[K, V](
   override def hasNext(): Boolean = okNext
 
   override def next(): ConsumerRecord[K, V] = {
-    assert(hasNext, "Can't call getNext() once untilOffset has been reached")
+    if (!hasNext) {
+      throw new ju.NoSuchElementException("Can't call getNext() once untilOffset has been reached")
+    }
     val r = nextRecord
     if (r.offset + 1 >= part.untilOffset) {
       okNext = false