复制Spark Dataset中的行N次

import org.apache.spark.sql.{SparkSession, Dataset}

object Main {
  def main(args: Array[String]): Unit = {
    // 创建SparkSession
    val spark = SparkSession.builder()
      .appName("Copy Rows N Times")
      .master("local")
      .getOrCreate()

    // 创建示例数据集
    import spark.implicits._
    val dataset = Seq(("row1"), ("row2"), ("row3")).toDS()

    // 定义复制次数
    val n = 3

    // 使用flatMap函数复制行
    val copiedDataset = dataset.flatMap(row => Seq.fill(n)(row))

    // 打印结果
    copiedDataset.show()

    // 停止SparkSession
    spark.stop()
  }
}