com.sparkutils.shim.expressions.Cast.scala Maven / Gradle / Ivy

Go to download
package com.sparkutils.shim.expressions

import org.apache.spark.sql.catalyst.expressions.{Add, Cast, EvalMode, Expression}
import org.apache.spark.sql.internal.SQLConf
import org.apache.spark.sql.types.DataType

object Cast2 {

  /**
   * Dbr 11.2 broke the contract for add and cast
   *
   * @param child
   * @param dataType
   * @return
   */
  def apply(child: Expression, dataType: DataType): Expression =
    Cast(child, dataType)

  def unapply(cast: Cast): Option[(Expression, DataType)] =
    cast match {
      case Cast(ch, dt, _, _) =>
        Some((ch, dt))
      case _ => None
    }
}

object Cast3 {

  /**
   * Dbr 11.2 broke the contract for add and cast
   *
   * @param child
   * @param dataType
   * @return
   */
  def apply(child: Expression, dataType: DataType, timeZoneId: Option[String] = None): Expression =
    Cast(child, dataType, timeZoneId)

  def unapply(cast: Cast): Option[(Expression, DataType, Option[String])] =
    cast match {
      case Cast(ch, dt, tz, _) =>
        Some((ch, dt, tz))
      case _ => None
    }
}

object Cast4 {

  /**
   * Dbr 11.2 broke the contract for add and cast
   *
   * @param child
   * @param dataType
   * @return
   */
  def apply(child: Expression, dataType: DataType, timeZoneId: Option[String] = None, evalMode: EvalMode.Value = EvalMode.fromSQLConf(SQLConf.get)): Expression =
    Cast(child, dataType, timeZoneId, evalMode)

  def unapply(cast: Cast): Option[(Expression, DataType, Option[String], EvalMode.Value)] =
    cast match {
      case Cast(ch, dt, tz, ev) =>
        Some((ch, dt, tz, ev))
      case _ => None
    }
}