На простом законченном примере использования агрегации окон в Spark 2.31 (HDP 3.0) я вижу, что Spark создает интервалы, выровненные по некоторому целому числу. Например, здесь я указываю 60 секунд windowDuration, и Spark начинает первый интервал с ближайшей минуты:
>>> from pyspark.sql import functions as F
>>> df = spark.createDataFrame([("2016-03-11 09:00:07", 1),("2016-03-11 09:00:08", 1)]).toDF("date", "val")
>>> w = df.groupBy(F.window("date", "60 seconds")).agg(F.sum("val").alias("sum"))
>>> w.select(w.window.start.cast("string").alias("start"),w.window.end.cast("string").alias("end"), "sum").collect()
[Row(start='2016-03-11 09:00:00', end='2016-03-11 09:01:00', sum=2)]
Есть ли способ начать интервал при первом сообщении, т.е. в моем случае я хотел бы иметь:
[Row(start='2016-03-11 09:00:07', end='2016-03-11 09:01:07', sum=2)]
Вот так =>
from pyspark.sql import functions as F
from datetime import datetime
df = spark.createDataFrame([("2016-03-11 09:00:07", 1),("2016-03-11 09:00:08", 1)]).toDF("date", "val")
startSecond = datetime.strptime(df.head()[0], '%Y-%m-%d %H:%M:%S').second
w = df.groupBy(F.window("date", "60 seconds", "60 seconds", str(startSecond) + " seconds")).agg(F.sum("val").alias("sum"))
w.select(w.window.start.cast("string").alias("start"),w.window.end.cast("string").alias("end"), "sum").collect()
Единственное, как получить startSecond в потоковом режиме: startSecond = datetime.strptime(st.head()[0], '%Y-%m-%d %H:%M:%S').second ... pyspark.sql.utils.AnalysisException: 'Queries with streaming sources must be executed with writeStream.start();;\nkafka'
Спасибо! Читая документы, я подумал, что это может работать таким образом, но попробовал неправильные вещи.