Spark Scala将数据帧写入MongoDB

import org.apache.spark.sql.{SparkSession, SaveMode}
import org.mongodb.scala._
import org.mongodb.scala.bson.collection.immutable.Document

val spark = SparkSession.builder()
  .appName("Write DataFrame to MongoDB")
  .master("local")
  .getOrCreate()

val data = Seq(("Alice", 25), ("Bob", 30), ("Charlie", 35))
val df = spark.createDataFrame(data).toDF("name", "age")

val documents = df.collect().map { row =>
  Document("name" -> row.getString(0), "age" -> row.getInt(1))
}

val mongoClient = MongoClient()
val database = mongoClient.getDatabase("mydb")
val collection = database.getCollection("mycollection")

collection.insertMany(documents)