如何检查所需密钥是否包含在JSON spark Scala Dataframe中

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

val spark = SparkSession.builder()
  .appName("Key Check")
  .getOrCreate()

val jsonPath = "path/to/json/file.json"
val df = spark.read.json(jsonPath)

val requiredKeys = List("key1", "key2", "key3")

val columns = df.columns

val keyCheck = requiredKeys.forall(columns.contains)

if (keyCheck) {
  println("所有所需密钥都包含在DataFrame中。")
} else {
  println("有些所需密钥未包含在DataFrame中。")
}