scalikejdbc.TimeZoneConverter.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalikejdbc-core_2.12 Show documentation

scalikejdbc-core

There is a newer version: 4.3.2

Show newest version

package scalikejdbc

import java.sql.Timestamp
import java.util.{ Calendar, TimeZone }

import scala.collection.concurrent.TrieMap

/**
 * TimeZone converter for SQL Timestamp
 */
class TimeZoneConverter(fromTimeZone: TimeZone, toTimeZone: TimeZone) {

  def convert(timestamp: Timestamp): Timestamp =
    if (timestamp != null) convertInternal(timestamp) else null

  private[this] def convertInternal(source: Timestamp): Timestamp = {
    val fromCal = Calendar.getInstance(fromTimeZone)
    fromCal.setTime(source)
    val fromOffset =
      fromCal.get(Calendar.ZONE_OFFSET) + fromCal.get(Calendar.DST_OFFSET)

    val toCal = Calendar.getInstance(toTimeZone)
    toCal.setTime(source)
    val toOffset =
      toCal.get(Calendar.ZONE_OFFSET) + toCal.get(Calendar.DST_OFFSET)

    val delta = toOffset - fromOffset

    val millis = source.getTime + delta
    val converted = new Timestamp(millis)
    converted.setNanos(source.getNanos)
    converted
  }
}

object TimeZoneConverter {
  class TimeZoneConverterBuilder(fromTimeZone: TimeZone) {
    private[this] val converterCache: TrieMap[TimeZone, TimeZoneConverter] =
      TrieMap.empty

    def to(toTimeZone: TimeZone): TimeZoneConverter =
      converterCache.getOrElseUpdate(
        toTimeZone,
        new TimeZoneConverter(fromTimeZone, toTimeZone)
      )
  }

  private[this] val builderCache: TrieMap[TimeZone, TimeZoneConverterBuilder] =
    TrieMap.empty

  def from(timeZone: TimeZone): TimeZoneConverterBuilder =
    builderCache.getOrElseUpdate(
      timeZone,
      new TimeZoneConverterBuilder(timeZone)
    )
}