如何在PySpark 2.1.0中定义事件时间窗口上的UDAF

from pyspark.sql import SparkSession
from pyspark.sql.functions import udf
from pyspark.sql.types import IntegerType
from pyspark.sql.window import Window

spark = SparkSession.builder.appName("EventTimeWindowUDAF").getOrCreate()

class MyUDAF:
    def __init__(self):
        self.buffer = []

    def initialize(self):
        self.buffer = []

    def update(self, value):
        self.buffer.append(value)

    def merge(self, other):
        self.buffer.extend(other)

    def evaluate(self):
        return sum(self.buffer)

my_udaf = MyUDAF()
spark.udf.register("my_udaf", my_udaf)

df = spark.createDataFrame([(1, "2022-01-01 10:00:00", 10),
                            (2, "2022-01-01 10:01:00", 20),
                            (3, "2022-01-01 10:02:00", 30),
                            (4, "2022-01-01 10:03:00", 40)],
                           ["id", "event_time", "value"])

window = Window.orderBy("event_time").rangeBetween(-600, 0)

df.withColumn("sum_value", udf(lambda x: my_udaf.update(x), IntegerType())("value").over(window)) \
  .withColumn("result", udf(lambda x: my_udaf.evaluate(), IntegerType())("value").over(window)) \
  .show()