Spark Scala -确定多行日期之间的差距

import org.apache.spark.sql.SparkSession
import org.apache.spark.sql.functions._

// 创建SparkSession
val spark = SparkSession.builder()
  .appName("Date Difference")
  .getOrCreate()

// 创建示例数据集
val data = Seq(
  ("2022-01-01", "2022-01-03"),
  ("2022-02-05", "2022-02-08"),
  ("2022-03-10", "2022-03-15")
)

// 创建DataFrame
val df = spark.createDataFrame(data).toDF("start_date", "end_date")

// 计算日期差距
val dateDiff = datediff(to_date($"end_date"), to_date($"start_date"))

// 显示结果
df.withColumn("date_diff", dateDiff).show()