为scala dataframe中的每一行添加唯一ID以进行多次插入

import org.apache.spark.sql.functions.monotonically_increasing_id

val dfWithId = df.withColumn("id", monotonically_increasing_id())

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

object AddUniqueIdToDataFrame {
  def main(args: Array[String]): Unit = {
    // 创建SparkSession
    val spark = SparkSession.builder()
      .appName("Add Unique ID to DataFrame")
      .master("local")
      .getOrCreate()

    // 创建示例DataFrame
    val data = Seq(("John", 25), ("Alice", 30), ("Bob", 35))
    val df = spark.createDataFrame(data).toDF("name", "age")

    // 使用monotonically_increasing_id函数为每一行添加唯一ID
    val dfWithId = df.withColumn("id", monotonically_increasing_id())

    // 显示DataFrame
    dfWithId.show()
  }
}

+-----+---+---+
| name|age| id|
+-----+---+---+
| John| 25|  0|
|Alice| 30|  1|
|  Bob| 35|  2|
+-----+---+---+