Search in sources :

Example 1 with TimeZoneKey

use of io.prestosql.spi.type.TimeZoneKey in project hetu-core by openlookeng.

the class MaterializedResult method writeValue.

private static void writeValue(Type type, BlockBuilder blockBuilder, Object value) {
    if (value == null) {
        blockBuilder.appendNull();
    } else if (BIGINT.equals(type)) {
        type.writeLong(blockBuilder, ((Number) value).longValue());
    } else if (INTEGER.equals(type)) {
        type.writeLong(blockBuilder, ((Number) value).intValue());
    } else if (SMALLINT.equals(type)) {
        type.writeLong(blockBuilder, ((Number) value).shortValue());
    } else if (TINYINT.equals(type)) {
        type.writeLong(blockBuilder, ((Number) value).byteValue());
    } else if (REAL.equals(type)) {
        type.writeLong(blockBuilder, (long) floatToRawIntBits(((Number) value).floatValue()));
    } else if (DOUBLE.equals(type)) {
        type.writeDouble(blockBuilder, ((Number) value).doubleValue());
    } else if (BOOLEAN.equals(type)) {
        type.writeBoolean(blockBuilder, (Boolean) value);
    } else if (JSON.equals(type)) {
        type.writeSlice(blockBuilder, Slices.utf8Slice((String) value));
    } else if (type instanceof VarcharType) {
        type.writeSlice(blockBuilder, Slices.utf8Slice((String) value));
    } else if (type instanceof CharType) {
        type.writeSlice(blockBuilder, Slices.utf8Slice((String) value));
    } else if (VARBINARY.equals(type)) {
        type.writeSlice(blockBuilder, Slices.wrappedBuffer((byte[]) value));
    } else if (DATE.equals(type)) {
        int days = ((SqlDate) value).getDays();
        type.writeLong(blockBuilder, days);
    } else if (TIME.equals(type)) {
        SqlTime time = (SqlTime) value;
        if (time.isLegacyTimestamp()) {
            type.writeLong(blockBuilder, time.getMillisUtc());
        } else {
            type.writeLong(blockBuilder, time.getMillis());
        }
    } else if (TIME_WITH_TIME_ZONE.equals(type)) {
        long millisUtc = ((SqlTimeWithTimeZone) value).getMillisUtc();
        TimeZoneKey timeZoneKey = ((SqlTimeWithTimeZone) value).getTimeZoneKey();
        type.writeLong(blockBuilder, packDateTimeWithZone(millisUtc, timeZoneKey));
    } else if (TIMESTAMP.equals(type)) {
        long millisUtc = ((SqlTimestamp) value).getMillis();
        type.writeLong(blockBuilder, millisUtc);
    } else if (TIMESTAMP_WITH_TIME_ZONE.equals(type)) {
        long millisUtc = ((SqlTimestampWithTimeZone) value).getMillisUtc();
        TimeZoneKey timeZoneKey = ((SqlTimestampWithTimeZone) value).getTimeZoneKey();
        type.writeLong(blockBuilder, packDateTimeWithZone(millisUtc, timeZoneKey));
    } else if (ARRAY.equals(type.getTypeSignature().getBase())) {
        List<Object> list = (List<Object>) value;
        Type elementType = ((ArrayType) type).getElementType();
        BlockBuilder arrayBlockBuilder = blockBuilder.beginBlockEntry();
        for (Object element : list) {
            writeValue(elementType, arrayBlockBuilder, element);
        }
        blockBuilder.closeEntry();
    } else if (MAP.equals(type.getTypeSignature().getBase())) {
        Map<Object, Object> map = (Map<Object, Object>) value;
        Type keyType = ((MapType) type).getKeyType();
        Type valueType = ((MapType) type).getValueType();
        BlockBuilder mapBlockBuilder = blockBuilder.beginBlockEntry();
        for (Entry<Object, Object> entry : map.entrySet()) {
            writeValue(keyType, mapBlockBuilder, entry.getKey());
            writeValue(valueType, mapBlockBuilder, entry.getValue());
        }
        blockBuilder.closeEntry();
    } else if (type instanceof RowType) {
        List<Object> row = (List<Object>) value;
        List<Type> fieldTypes = type.getTypeParameters();
        BlockBuilder rowBlockBuilder = blockBuilder.beginBlockEntry();
        for (int field = 0; field < row.size(); field++) {
            writeValue(fieldTypes.get(field), rowBlockBuilder, row.get(field));
        }
        blockBuilder.closeEntry();
    } else {
        throw new IllegalArgumentException("Unsupported type " + type);
    }
}
Also used : VarcharType(io.prestosql.spi.type.VarcharType) SqlTime(io.prestosql.spi.type.SqlTime) SqlTimeWithTimeZone(io.prestosql.spi.type.SqlTimeWithTimeZone) RowType(io.prestosql.spi.type.RowType) MapType(io.prestosql.spi.type.MapType) RowType(io.prestosql.spi.type.RowType) Type(io.prestosql.spi.type.Type) ArrayType(io.prestosql.spi.type.ArrayType) CharType(io.prestosql.spi.type.CharType) MapType(io.prestosql.spi.type.MapType) VarcharType(io.prestosql.spi.type.VarcharType) SqlTimestampWithTimeZone(io.prestosql.spi.type.SqlTimestampWithTimeZone) SqlDate(io.prestosql.spi.type.SqlDate) ImmutableList.toImmutableList(com.google.common.collect.ImmutableList.toImmutableList) List(java.util.List) ArrayList(java.util.ArrayList) ImmutableList(com.google.common.collect.ImmutableList) CharType(io.prestosql.spi.type.CharType) TimeZoneKey(io.prestosql.spi.type.TimeZoneKey) Map(java.util.Map) ImmutableMap(com.google.common.collect.ImmutableMap) BlockBuilder(io.prestosql.spi.block.BlockBuilder)

Example 2 with TimeZoneKey

use of io.prestosql.spi.type.TimeZoneKey in project hetu-core by openlookeng.

the class TimestampOperators method castToTimestampWithTimeZone.

@ScalarOperator(CAST)
@SqlType(StandardTypes.TIMESTAMP_WITH_TIME_ZONE)
public static long castToTimestampWithTimeZone(ConnectorSession session, @SqlType(StandardTypes.TIMESTAMP) long value) {
    TimeZoneKey timeZoneKey = session.getTimeZoneKey();
    ISOChronology localChronology = getChronology(timeZoneKey);
    // its UTC representation we need to shift the value by the offset of TZ.
    return packDateTimeWithZone(localChronology.getZone().convertLocalToUTC(value, false), timeZoneKey);
}
Also used : ISOChronology(org.joda.time.chrono.ISOChronology) TimeZoneKey(io.prestosql.spi.type.TimeZoneKey) ScalarOperator(io.prestosql.spi.function.ScalarOperator) SqlType(io.prestosql.spi.function.SqlType)

Example 3 with TimeZoneKey

use of io.prestosql.spi.type.TimeZoneKey in project hetu-core by openlookeng.

the class TestPrestoDriver method testSetTimeZoneId.

@Test
public void testSetTimeZoneId() throws Exception {
    TimeZoneKey defaultZoneKey = TimeZoneKey.getTimeZoneKey(TimeZone.getDefault().getID());
    DateTimeZone defaultZone = DateTimeZone.forTimeZone(TimeZone.getDefault());
    String sql = "SELECT current_timezone() zone, TIMESTAMP '2001-02-03 3:04:05' ts";
    try (Connection connection = createConnection()) {
        try (Statement statement = connection.createStatement();
            ResultSet rs = statement.executeQuery(sql)) {
            assertTrue(rs.next());
            assertEquals(rs.getString("zone"), defaultZoneKey.getId());
            assertEquals(rs.getTimestamp("ts"), new Timestamp(new DateTime(2001, 2, 3, 3, 4, 5, defaultZone).getMillis()));
        }
        connection.unwrap(PrestoConnection.class).setTimeZoneId("UTC");
        try (Statement statement = connection.createStatement();
            ResultSet rs = statement.executeQuery(sql)) {
            assertTrue(rs.next());
            assertEquals(rs.getString("zone"), "UTC");
            assertEquals(rs.getTimestamp("ts"), new Timestamp(new DateTime(2001, 2, 3, 3, 4, 5, DateTimeZone.UTC).getMillis()));
        }
    }
}
Also used : Statement(java.sql.Statement) Connection(java.sql.Connection) ResultSet(java.sql.ResultSet) TimeZoneKey(io.prestosql.spi.type.TimeZoneKey) Timestamp(java.sql.Timestamp) DateTimeZone(org.joda.time.DateTimeZone) DateTime(org.joda.time.DateTime) LocalDateTime(org.joda.time.LocalDateTime) Test(org.testng.annotations.Test)

Example 4 with TimeZoneKey

use of io.prestosql.spi.type.TimeZoneKey in project hetu-core by openlookeng.

the class TestTimeBase method testCastToTimeWithTimeZoneWithTZWithRulesChanged.

@Test
public void testCastToTimeWithTimeZoneWithTZWithRulesChanged() {
    TimeZoneKey timeZoneThatChangedSince1970 = getTimeZoneKey("Asia/Kathmandu");
    DateTimeZone dateTimeZoneThatChangedSince1970 = getDateTimeZone(timeZoneThatChangedSince1970);
    Session session = Session.builder(this.session).setTimeZoneKey(timeZoneThatChangedSince1970).build();
    try (FunctionAssertions localAssertions = new FunctionAssertions(session)) {
        localAssertions.assertFunction("cast(TIME '03:04:05.321' as time with time zone)", TIME_WITH_TIME_ZONE, new SqlTimeWithTimeZone(new DateTime(1970, 1, 1, 3, 4, 5, 321, dateTimeZoneThatChangedSince1970).getMillis(), dateTimeZoneThatChangedSince1970.toTimeZone()));
    }
}
Also used : SqlTimeWithTimeZone(io.prestosql.spi.type.SqlTimeWithTimeZone) FunctionAssertions(io.prestosql.operator.scalar.FunctionAssertions) TimeZoneKey(io.prestosql.spi.type.TimeZoneKey) TimeZoneKey.getTimeZoneKey(io.prestosql.spi.type.TimeZoneKey.getTimeZoneKey) DateTimeZone(org.joda.time.DateTimeZone) DateTimeZoneIndex.getDateTimeZone(io.prestosql.spi.util.DateTimeZoneIndex.getDateTimeZone) DateTime(org.joda.time.DateTime) TestingSession(io.prestosql.testing.TestingSession) Session(io.prestosql.Session) Test(org.testng.annotations.Test)

Aggregations

TimeZoneKey (io.prestosql.spi.type.TimeZoneKey)4 SqlTimeWithTimeZone (io.prestosql.spi.type.SqlTimeWithTimeZone)2 DateTime (org.joda.time.DateTime)2 DateTimeZone (org.joda.time.DateTimeZone)2 Test (org.testng.annotations.Test)2 ImmutableList (com.google.common.collect.ImmutableList)1 ImmutableList.toImmutableList (com.google.common.collect.ImmutableList.toImmutableList)1 ImmutableMap (com.google.common.collect.ImmutableMap)1 Session (io.prestosql.Session)1 FunctionAssertions (io.prestosql.operator.scalar.FunctionAssertions)1 BlockBuilder (io.prestosql.spi.block.BlockBuilder)1 ScalarOperator (io.prestosql.spi.function.ScalarOperator)1 SqlType (io.prestosql.spi.function.SqlType)1 ArrayType (io.prestosql.spi.type.ArrayType)1 CharType (io.prestosql.spi.type.CharType)1 MapType (io.prestosql.spi.type.MapType)1 RowType (io.prestosql.spi.type.RowType)1 SqlDate (io.prestosql.spi.type.SqlDate)1 SqlTime (io.prestosql.spi.type.SqlTime)1 SqlTimestampWithTimeZone (io.prestosql.spi.type.SqlTimestampWithTimeZone)1