Added an RDS that repeatedly returns the same input
This commit is contained in:
parent
650d11817e
commit
29bf44473c
|
@ -0,0 +1,18 @@
|
|||
package spark.streaming
|
||||
|
||||
import spark.RDD
|
||||
|
||||
/**
|
||||
* An input stream that always returns the same RDD on each timestep. Useful for testing.
|
||||
*/
|
||||
class ConstantInputRDS[T: ClassManifest](ssc: SparkStreamContext, rdd: RDD[T])
|
||||
extends InputRDS[T](ssc) {
|
||||
|
||||
override def start() {}
|
||||
|
||||
override def stop() {}
|
||||
|
||||
override def compute(validTime: Time): Option[RDD[T]] = {
|
||||
Some(rdd)
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue