Key not found exception: if receiver is slow to start, it is possible…

… that getReceivedBlockInfo will be called before compute has been called
pdeyhim · Jun 3, 2014 · 4609563 · 4609563
1 parent d967941
commit 4609563
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/streaming/src/main/scala/org/apache/spark/streaming/dstream/ReceiverInputDStream.scala b/streaming/src/main/scala/org/apache/spark/streaming/dstream/ReceiverInputDStream.scala
@@ -74,7 +74,7 @@ abstract class ReceiverInputDStream[T: ClassTag](@transient ssc_ : StreamingCont
 
   /** Get information on received blocks. */
   private[streaming] def getReceivedBlockInfo(time: Time) = {
-    receivedBlockInfo(time)
+     receivedBlockInfo.get(time).getOrElse(Array.empty[ReceivedBlockInfo])
   }
 
   /**