如何使用sparkListener对不同的数据帧写操作进行记录计数？

import org.apache.spark.scheduler._

class CustomSparkListener extends SparkListener {
  override def onOtherEvent(event: SparkListenerEvent): Unit = {
    event match {
      case writerEvent: DataFrameWriterEvent =>
        val dataFrameName = writerEvent.dataFrameName
        val dataSourceType = writerEvent.dataSourceType
        // 记录数据帧写操作信息，可以根据需要进行日志记录、存储等操作
        println(s"DataFrame $dataFrameName is being written to $dataSourceType")
      case _ =>
      // 其他事件类型，可以根据需要进行处理
    }
  }
}

// 创建SparkSession
val spark = SparkSession.builder()
  .appName("SparkListenerExample")
  .master("local[*]")
  .getOrCreate()

// 注册自定义的SparkListener
spark.sparkContext.addSparkListener(new CustomSparkListener())

// 执行Spark应用程序，包括数据帧的写操作
val data = Seq(("Alice", 25), ("Bob", 30), ("Charlie", 35))
val df = spark.createDataFrame(data).toDF("name", "age")
df.write.format("parquet").save("output.parquet")