bio.ferlab.datalake.spark3.transformation.ToUtcTimestamps.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of datalake-spark3_2.12 Show documentation

Library built on top of Apache Spark to speed-up data lakes development..

There is a newer version: 14.8.0

package bio.ferlab.datalake.spark3.transformation

import org.apache.spark.sql.DataFrame
import org.apache.spark.sql.functions.{col, to_utc_timestamp}
import org.apache.spark.sql.types.TimestampType

case class ToUtcTimestamps(format: String, columns: String*) extends Transformation {
  override def transform: DataFrame => DataFrame = {df =>
    val cols = if (columns.isEmpty) {
      df.schema.fields.filter(_.dataType == TimestampType).map(_.name).toSeq
    } else {
      columns.toSeq
    }
    cols.foldLeft(df)((d, c) =>  d.withColumn(c, to_utc_timestamp(col(c), format)))
  }
}