如何从包含枚举的案例类创建Spark Dataset或Dataframe

object Main extends App {
  // 定义枚举
  object Color extends Enumeration {
    type Color = Value
    val Red, Green, Blue = Value
  }

  // 定义案例类
  case class Person(name: String, age: Int, color: Color.Color)

  // 创建Spark Session
  val spark = SparkSession.builder()
    .appName("Spark Example")
    .master("local")
    .getOrCreate()

  // 导入隐式转换
  import spark.implicits._

  // 创建包含枚举的案例类的数据集
  val data = Seq(
    Person("Alice", 25, Color.Red),
    Person("Bob", 30, Color.Green),
    Person("Charlie", 35, Color.Blue)
  )
  val dataset = spark.createDataset(data)

  // 显示数据集内容
  dataset.show()

  // 将数据集转换为Dataframe
  val dataframe = dataset.toDF()

  // 显示Dataframe内容
  dataframe.show()

  // 停止Spark Session
  spark.stop()
}