如何检查spark sql映射类型中是否存在key

import org.apache.spark.sql.types._

val schema = new StructType().add("map", MapType(StringType, StringType))

val fields = schema.fields

val mapField = fields.find(_.name == "map")

mapField match {
  case Some(field) =>
    if (field.dataType.isInstanceOf[MapType]) {
      val mapType = field.dataType.asInstanceOf[MapType]
      val hasKey = mapType.keyType match {
        case StringType => // 如果key的类型是String
          val map = Map("key" -> "value")
          map.contains("key")
        case _ => false // 其他类型的key
      }
      println(s"映射类型中是否存在key：$hasKey")
    }
  case None =>
    println("找不到对应的字段")
}