有没有一种方法可以将生成的groupby流重新加入kafka-spark结构化流中的原始流?

时间:2019-04-15 08:34:56

标签: pyspark apache-kafka pyspark-sql spark-structured-streaming

我正在阅读Kafka主题的信息流。我正在事件时间执行window groupBy操作。现在,我想将由此产生的流从groupBy加入到原始流中。

#I am reading from a Kafka topic. The following is a ping statement:
2019-04-15 13:32:33 | 64 bytes from google.com (X.X.X.X): icmp_seq=476 ttl=63 time=0.201ms
2019-04-15 13:32:34 | 64 bytes from google.com (X.X.X.X): icmp_seq=477 ttl=63 time=0.216ms
2019-04-15 13:32:35 | 64 bytes from google.com (X.X.X.X): icmp_seq=478 ttl=63 time=0.245ms
2019-04-15 13:32:36 | 64 bytes from google.com (X.X.X.X): icmp_seq=479 ttl=63 time=0.202ms
and so on..

root
|--key: binary
|--value: binary
|--topic: string
|--partition: integer
|--offset:long
|--timestamp:timestamp
|--timestampType:integer

#value contains the above ping statement so, I cast it as string.
words = lines.selectExpr("CAST(value AS STRING)")

#Now I split that line into columns with its values.
words = words.withColumn('string_val', F.regexp_replace(F.split(words.value, " ")[6], ":", "")) \
.withColumn('ping', F.regexp_replace(F.split(words.value, " ")[10], "time=", "").cast("double")) \
.withColumn('date', F.split(words.value, " ")[0]) \
.withColumn('time', F.regexp_replace(F.split(words.value, " ")[1], "|", ""))

words = words.withColumn('date_time', F.concat(F.col('date'), F.lit(" "), F.col('time')))
words = words.withColumn('date_format', F.col('date_time').cast('timestamp'))

#Now the schema becomes like this
root
|--value:string
|--string_val:string
|--ping:double
|--date:string
|--time:string
|--date_time:string
|--date_format:timestamp

#Now I have to perform a windowed groupBy operation with watermark
w = F.window('date_format', '30 seconds', '10 seconds')
words = words \
.withWatermark('date_format', '1 minutes') \
.groupBy(w).agg(F.mean('ping').alias('value'))

#Now my schema becomes like this
root
|--window:struct
|   |--start:timestamp
|   |--end:timestamp
|--value

有什么方法可以将此结果流重新加入其原始流?

1 个答案:

答案 0 :(得分:0)

可以使用spark 2.3中引入的“流到流连接”来实现 对于spark 2.3之前的任何版本,您必须将聚合保留在某个存储区(内存或磁盘)中,并使用该存储区(用于存储聚合状态)执行原始流的左外部联接。