有没有办法在Kafka流中的`foreachBatch`函数中传入一个额外的/额外的参数？

# 额外的参数
extra_param = "额外参数"

def process_batch(batch_df, batch_id):
    # 在这里使用额外的参数
    print("处理批次", batch_id, "，额外参数为", extra_param)
    # 其他处理逻辑

# 应用`foreachBatch`函数
kafka_stream.writeStream.foreachBatch(process_batch).start()

def process_batch(iterator):
    # 获取额外的参数
    extra_param = iterator.__next__()
    # 处理每个分区的数据
    for record in iterator:
        # 处理逻辑
        pass

# 为数据集添加额外的参数
extra_params = ["额外参数1", "额外参数2"]
stream_with_params = kafka_stream.select(F.lit(extra_params).alias("extra_params"), F.struct("*"))

# 应用`mapPartitions`转换器
stream_with_params.rdd.mapPartitions(process_batch).foreach(lambda _: None)