Spark Streaming保证特定的启动窗口时间

时间:2017-09-20 18:40:46

标签: scala apache-spark spark-streaming amazon-kinesis

我使用Spark Streaming使用Structured Streaming框架从Kinesis读取数据,我的连接如下

val kinesis = spark
  .readStream
  .format("kinesis")
  .option("streams", streamName)
  .option("endpointUrl", endpointUrl)
  .option("initialPositionInStream", "earliest")
  .option("format", "json")
  .schema(<my-schema>)
  .load

数据来自几个具有唯一ID的IoT设备,我需要通过此ID和时间戳字段上的翻滚窗口聚合数据,如下所示:

val aggregateData = kinesis
    .groupBy($"uid", window($"timestamp", "15 minute", "15 minute"))
    .agg(...)

我遇到的问题是我需要保证每个窗口都是在圆形时间开始(例如00:00,00:00:00:00等),我还需要保证只有包含完整15分钟长窗口的行将输出到我的接收器,我目前正在做的是

val query = aggregateData
    .writeStream
      .foreach(postgreSQLWriter)
      .outputMode("update")
      .start()
      .awaitTermination()

postgreSQLWriter是我创建的StreamWriter,用于将每一行插入到PostgreSQL SGBD中。如何强制我的窗口长度为15分钟,开始时间为每个设备唯一ID的15分钟时间戳值?

1 个答案:

答案 0 :(得分:2)

问题1: 为了在特定的时间开始,还有一个参数火花分组功能,其中&#34;偏移&#34;。 通过指定它将在一小时的指定时间后开始 示例:

dataframe.groupBy($"Column1",window($"TimeStamp","22 minute","1 minute","15 minute"))

所以上面的语法将按列1分组并创建22分钟持续时间的窗口,滑动窗口大小为1分钟,偏移量为15分钟

例如它从:

开始
window1: 8:15(8:00 add 15 minute offset) to 8:37 (8:15 add 22 minutes)
window2: 8:16(previous window start + 1 minute) to 8:38 ( 22 minute size again)

问题2: 仅推送那些具有完整15分钟大小的窗口,创建一个计数列,用于计算该窗口中的事件数。一旦达到15,使用过滤器命令将其推送到任何地方

计算数:

dataframe.groupBy($"Column1",window($"TimeStamp","22 minute","1 minute","15 minute")).agg(count*$"Column2").as("count"))

仅包含计数15的写入流过滤器:

aggregateddata.filter($"count"===15).writeStream.format(....).outputMode("complete").start()