org.sfm.jdbc.impl.getter.time.JavaZonedDateTimeResultSetGetter Maven / Gradle / Ivy

Go to download
package org.sfm.jdbc.impl.getter.time;

import org.sfm.jdbc.JdbcColumnKey;
import org.sfm.reflect.Getter;

import java.sql.ResultSet;
import java.time.*;
import java.time.temporal.TemporalAccessor;
import java.util.Date;


public class JavaZonedDateTimeResultSetGetter implements Getter {
    private final int index;
    private final ZoneId zone;

    public JavaZonedDateTimeResultSetGetter(JdbcColumnKey key, ZoneId zoneId) {
        this.index = key.getIndex();
        this.zone = zoneId;
    }

    @Override
    public ZonedDateTime get(ResultSet target) throws Exception {
        Object o = target.getObject(index);

        if (o == null) {
            return null;
        }

        if (o instanceof Date) {
            return Instant.ofEpochMilli(((Date) o).getTime()).atZone(zone);
        }

        if (o instanceof ZonedDateTime) {
            return (ZonedDateTime) o;
        }

        if (o instanceof LocalDateTime) {
            return ((LocalDateTime)o).atZone(zone);
        }

        if (o instanceof OffsetDateTime) {
            return ((OffsetDateTime)o).toZonedDateTime();
        }

        if (o instanceof TemporalAccessor) {
            return ZonedDateTime.from((TemporalAccessor) o);
        }

        throw new IllegalArgumentException("Cannot convert " + o + " to ZonedDateTime");
    }

    @Override
    public String toString() {
        return "JavaZonedDateTimeResultSetGetter{" +
                "column=" + index +
                '}';
    }
}