org.embulk.input.tsurugidb.getter.TimeColumnGetter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of embulk-input-tsurugidb Show documentation

Selects records from a table.

The newest version!

package org.embulk.input.tsurugidb.getter;

import java.io.IOException;
import java.time.Instant;
import java.time.LocalDate;
import java.time.LocalTime;
import java.time.ZoneId;

import org.embulk.spi.PageBuilder;
import org.embulk.spi.type.Type;
import org.embulk.spi.type.Types;
import org.embulk.util.timestamp.TimestampFormatter;

import com.tsurugidb.tsubakuro.exception.ServerException;
import com.tsurugidb.tsubakuro.kvs.Record;
import com.tsurugidb.tsubakuro.sql.ResultSet;

// https://github.com/embulk/embulk-input-jdbc/blob/master/embulk-input-jdbc/src/main/java/org/embulk/input/jdbc/getter/TimeColumnGetter.java
public class TimeColumnGetter extends AbstractTimestampColumnGetter {
    static final String DEFAULT_FORMAT = "%H:%M:%S";

    public TimeColumnGetter(PageBuilder to, Type toType, TimestampFormatter timestampFormatter, ZoneId zoneId) {
        super(to, toType, timestampFormatter, zoneId);
    }

    @Override
    protected void fetch(ResultSet from) throws IOException, ServerException, InterruptedException {
        LocalTime time = from.fetchTimeOfDayValue();
        setValue(time);
    }

    @Override
    protected void fetch(Record from, String name) {
        LocalTime time = from.getTimeOfDay(name);
        setValue(time);
    }

    protected void setValue(LocalTime time) {
        long epochSecond = time.toEpochSecond(LocalDate.EPOCH, zoneOffset);
        value = Instant.ofEpochSecond(epochSecond, time.getNano());
    }

    @Override
    @SuppressWarnings("deprecation")
    protected Type getDefaultToType() {
        return Types.TIMESTAMP.withFormat(DEFAULT_FORMAT);
    }
}