io.deephaven.engine.table.impl.sources.regioned.RegionedColumnSourceZonedDateTime Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of deephaven-engine-table Show documentation
Engine Table: Implementation and closely-coupled utilities
There is a newer version: 0.37.1
Show newest version
package io.deephaven.engine.table.impl.sources.regioned;

import io.deephaven.chunk.Chunk;
import io.deephaven.chunk.LongChunk;
import io.deephaven.chunk.WritableChunk;
import io.deephaven.chunk.WritableObjectChunk;
import io.deephaven.chunk.attributes.Values;
import io.deephaven.engine.rowset.RowSequence;
import io.deephaven.engine.table.ColumnSource;
import io.deephaven.engine.table.impl.ColumnSourceGetDefaults;
import io.deephaven.engine.table.impl.sources.*;
import io.deephaven.time.DateTimeUtils;
import org.jetbrains.annotations.NotNull;

import java.time.Instant;
import java.time.LocalDate;
import java.time.LocalTime;
import java.time.ZoneId;
import java.time.ZonedDateTime;

/**
 * Regioned column source implementation for columns of {@link ZonedDateTime}s.
 */
final class RegionedColumnSourceZonedDateTime
        extends RegionedColumnSourceReferencing>
        implements ColumnSourceGetDefaults.ForObject, ConvertibleTimeSource,
        ConvertibleTimeSource.Zoned {
    private final ZoneId zone;

    public static ColumnSource asLocalDate(ZoneId zone, RegionedColumnSourceLong inner) {
        return new LongAsLocalDateColumnSource(inner, zone);
    }

    public static ColumnSource asLocalTime(ZoneId zone, RegionedColumnSourceLong inner) {
        return new LongAsLocalTimeColumnSource(inner, zone);
    }

    public RegionedColumnSourceZonedDateTime(
            @NotNull final ZoneId zone,
            @NotNull final RegionedColumnSourceLong inner) {
        super(ColumnRegionLong.createNull(PARAMETERS.regionMask), ZonedDateTime.class, inner);
        this.zone = zone;
    }

    @Override
    public void convertRegion(
            @NotNull final WritableChunk destination,
            @NotNull final Chunk source,
            @NotNull final RowSequence rowSequence) {
        final WritableObjectChunk objectChunk = destination.asWritableObjectChunk();
        final LongChunk longChunk = source.asLongChunk();

        final int size = objectChunk.size();
        final int length = longChunk.size();

        for (int i = 0; i < length; ++i) {
            objectChunk.set(size + i, DateTimeUtils.epochNanosToZonedDateTime(longChunk.get(i), zone));
        }
        objectChunk.setSize(size + length);
    }

    @Override
    public ZonedDateTime get(final long rowKey) {
        return rowKey == RowSequence.NULL_ROW_KEY ? null
                : DateTimeUtils.epochNanosToZonedDateTime(getNativeSource().lookupRegion(rowKey).getLong(rowKey),
                        zone);
    }

    @Override
    public ColumnSource toInstant() {
        return new RegionedColumnSourceInstant((RegionedColumnSourceLong) getNativeSource());
    }

    @Override
    public boolean supportsTimeConversion() {
        return true;
    }

    @Override
    public ColumnSource toZonedDateTime(@NotNull final ZoneId zone) {
        if (this.zone.equals(zone)) {
            return this;
        }
        return new RegionedColumnSourceZonedDateTime(zone, (RegionedColumnSourceLong) getNativeSource());
    }

    @Override
    public ColumnSource toLocalDate(ZoneId zone) {
        return RegionedColumnSourceZonedDateTime.asLocalDate(zone,
                (RegionedColumnSourceLong) getNativeSource());
    }

    @Override
    public ColumnSource toLocalTime(ZoneId zone) {
        return RegionedColumnSourceZonedDateTime.asLocalTime(zone,
                (RegionedColumnSourceLong) getNativeSource());
    }

    @Override
    public ColumnSource toEpochNano() {
        return getNativeSource();
    }

    @Override
    public ZoneId getZone() {
        return zone;
    }
}