enum spark UDT

SparkEnumUDT.scala

package org.apache.spark.sql.types

import xxx.Condition
import xxx.Department


class ConditionUDT extends UserDefinedType[Condition.Value] {
  override def sqlType: DataType = IntegerType

  override def serialize(obj: Condition.Value): Int = obj.id

  override def deserialize(datum: Any): Condition.Value = Condition(datum.asInstanceOf[Int])

  override def userClass: Class[Condition.Value] = classOf[Condition.Value]
}

object EnumUDTRegister {

  def register(): Unit = {
    val clazz = this.getClass.getClassLoader.loadClass("org.apache.spark.sql.types.UDTRegistration")
    val m = clazz.getMethods.filter(_.getName == "register").head
    m.setAccessible(true)

    m.invoke(clazz, classOf[Condition.Value].getCanonicalName, "org.apache.spark.sql.types.ConditionUDT")
  }

}