com.databricks.jdbc.api.impl.converters.DateConverter Maven / Gradle / Ivy

Go to download
package com.databricks.jdbc.api.impl.converters;

import com.databricks.jdbc.exception.DatabricksSQLException;
import java.math.BigInteger;
import java.sql.Date;
import java.sql.Timestamp;
import java.time.LocalDate;
import java.time.temporal.ChronoUnit;

public class DateConverter implements ObjectConverter {
  @Override
  public Date toDate(Object object) throws DatabricksSQLException {
    if (object instanceof String) {
      return Date.valueOf((String) object);
    } else if (object instanceof Date) {
      return (Date) object;
    } else {
      throw new DatabricksSQLException(
          "Unsupported type for DateObjectConverter: " + object.getClass());
    }
  }

  @Override
  public long toLong(Object object) throws DatabricksSQLException {
    LocalDate localStartDate = LocalDate.ofEpochDay(0);
    return ChronoUnit.DAYS.between(localStartDate, toDate(object).toLocalDate());
  }

  @Override
  public short toShort(Object object) throws DatabricksSQLException {
    long epochDays = toLong(object);
    if ((short) epochDays == epochDays) {
      return (short) epochDays;
    }
    throw new DatabricksSQLException("Invalid conversion: Date value out of short range");
  }

  @Override
  public int toInt(Object object) throws DatabricksSQLException {
    // the convertToLong will always be within integer limits
    return (int) toLong(object);
  }

  @Override
  public BigInteger toBigInteger(Object object) throws DatabricksSQLException {
    return BigInteger.valueOf(toLong(object));
  }

  @Override
  public LocalDate toLocalDate(Object object) throws DatabricksSQLException {
    return toDate(object).toLocalDate();
  }

  @Override
  public String toString(Object object) throws DatabricksSQLException {
    return toDate(object).toString();
  }

  @Override
  public Timestamp toTimestamp(Object object) throws DatabricksSQLException {
    return Timestamp.valueOf(toDate(object).toLocalDate().atStartOfDay());
  }
}