Generate schema from case class in spark

SparkGenerateSchema.scala

import org.apache.spark.sql.types.{StructType, ArrayType}
import org.apache.spark.sql.catalyst.ScalaReflection

/** Simple schema */
case class A(key: String, time: java.sql.Timestamp, date: java.sql.Date, decimal: java.math.BigDecimal, map: Map[String, Int], nested: Seq[Map[String, Seq[Int]]])
val schema = ScalaReflection.schemaFor[A].dataType.asInstanceOf[StructType]
schema.printTreeString

/** Array schema */
val arrSchema = ScalaReflection.schemaFor[Seq[A]].dataType.asInstanceOf[ArrayType]

/** When aUDF is returning A */
val aUDF = udf(()=>{}, schema)

/** When arrUDF is returning Seq[A] */
val arrUDF = udf(()=>{}, arrSchema)