| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.apache.cassandra.cql3.functions; |
| |
| import java.math.BigDecimal; |
| import java.math.BigInteger; |
| import java.util.Date; |
| |
| import org.apache.cassandra.cql3.CQLTester; |
| import org.apache.cassandra.serializers.SimpleDateSerializer; |
| import org.apache.cassandra.utils.UUIDGen; |
| import org.joda.time.DateTime; |
| import org.joda.time.DateTimeZone; |
| import org.joda.time.format.DateTimeFormat; |
| import org.junit.Test; |
| |
| public class CastFctsTest extends CQLTester |
| { |
| @Test |
| public void testInvalidQueries() throws Throwable |
| { |
| createTable("CREATE TABLE %s (a int primary key, b text, c double)"); |
| |
| assertInvalidSyntaxMessage("no viable alternative at input '(' (... b, c) VALUES ([CAST](...)", |
| "INSERT INTO %s (a, b, c) VALUES (CAST(? AS int), ?, ?)", 1.6, "test", 6.3); |
| |
| assertInvalidSyntaxMessage("no viable alternative at input '(' (..." + KEYSPACE + "." + currentTable() |
| + " SET c = [cast](...)", |
| "UPDATE %s SET c = cast(? as double) WHERE a = ?", 1, 1); |
| |
| assertInvalidSyntaxMessage("no viable alternative at input '(' (...= ? WHERE a = [CAST] (...)", |
| "UPDATE %s SET c = ? WHERE a = CAST (? AS INT)", 1, 2.0); |
| |
| assertInvalidSyntaxMessage("no viable alternative at input '(' (..." + KEYSPACE + "." + currentTable() |
| + " WHERE a = [CAST] (...)", |
| "DELETE FROM %s WHERE a = CAST (? AS INT)", 1, 2.0); |
| |
| assertInvalidSyntaxMessage("no viable alternative at input '(' (..." + KEYSPACE + "." + currentTable() |
| + " WHERE a = [CAST] (...)", |
| "SELECT * FROM %s WHERE a = CAST (? AS INT)", 1, 2.0); |
| |
| assertInvalidMessage("a cannot be cast to boolean", "SELECT CAST(a AS boolean) FROM %s"); |
| } |
| |
| @Test |
| public void testNumericCastsInSelectionClause() throws Throwable |
| { |
| createTable("CREATE TABLE %s (a tinyint primary key," |
| + " b smallint," |
| + " c int," |
| + " d bigint," |
| + " e float," |
| + " f double," |
| + " g decimal," |
| + " h varint," |
| + " i int)"); |
| |
| execute("INSERT INTO %s (a, b, c, d, e, f, g, h) VALUES (?, ?, ?, ?, ?, ?, ?, ?)", |
| (byte) 1, (short) 2, 3, 4L, 5.2F, 6.3, BigDecimal.valueOf(6.3), BigInteger.valueOf(4)); |
| |
| assertColumnNames(execute("SELECT CAST(b AS int), CAST(c AS int), CAST(d AS double) FROM %s"), |
| "cast(b as int)", |
| "c", |
| "cast(d as double)"); |
| |
| assertRows(execute("SELECT CAST(a AS tinyint), " + |
| "CAST(b AS tinyint), " + |
| "CAST(c AS tinyint), " + |
| "CAST(d AS tinyint), " + |
| "CAST(e AS tinyint), " + |
| "CAST(f AS tinyint), " + |
| "CAST(g AS tinyint), " + |
| "CAST(h AS tinyint), " + |
| "CAST(i AS tinyint) FROM %s"), |
| row((byte) 1, (byte) 2, (byte) 3, (byte) 4L, (byte) 5, (byte) 6, (byte) 6, (byte) 4, null)); |
| |
| assertRows(execute("SELECT CAST(a AS smallint), " + |
| "CAST(b AS smallint), " + |
| "CAST(c AS smallint), " + |
| "CAST(d AS smallint), " + |
| "CAST(e AS smallint), " + |
| "CAST(f AS smallint), " + |
| "CAST(g AS smallint), " + |
| "CAST(h AS smallint), " + |
| "CAST(i AS smallint) FROM %s"), |
| row((short) 1, (short) 2, (short) 3, (short) 4L, (short) 5, (short) 6, (short) 6, (short) 4, null)); |
| |
| assertRows(execute("SELECT CAST(a AS int), " + |
| "CAST(b AS int), " + |
| "CAST(c AS int), " + |
| "CAST(d AS int), " + |
| "CAST(e AS int), " + |
| "CAST(f AS int), " + |
| "CAST(g AS int), " + |
| "CAST(h AS int), " + |
| "CAST(i AS int) FROM %s"), |
| row(1, 2, 3, 4, 5, 6, 6, 4, null)); |
| |
| assertRows(execute("SELECT CAST(a AS bigint), " + |
| "CAST(b AS bigint), " + |
| "CAST(c AS bigint), " + |
| "CAST(d AS bigint), " + |
| "CAST(e AS bigint), " + |
| "CAST(f AS bigint), " + |
| "CAST(g AS bigint), " + |
| "CAST(h AS bigint), " + |
| "CAST(i AS bigint) FROM %s"), |
| row(1L, 2L, 3L, 4L, 5L, 6L, 6L, 4L, null)); |
| |
| assertRows(execute("SELECT CAST(a AS float), " + |
| "CAST(b AS float), " + |
| "CAST(c AS float), " + |
| "CAST(d AS float), " + |
| "CAST(e AS float), " + |
| "CAST(f AS float), " + |
| "CAST(g AS float), " + |
| "CAST(h AS float), " + |
| "CAST(i AS float) FROM %s"), |
| row(1.0F, 2.0F, 3.0F, 4.0F, 5.2F, 6.3F, 6.3F, 4.0F, null)); |
| |
| assertRows(execute("SELECT CAST(a AS double), " + |
| "CAST(b AS double), " + |
| "CAST(c AS double), " + |
| "CAST(d AS double), " + |
| "CAST(e AS double), " + |
| "CAST(f AS double), " + |
| "CAST(g AS double), " + |
| "CAST(h AS double), " + |
| "CAST(i AS double) FROM %s"), |
| row(1.0, 2.0, 3.0, 4.0, (double) 5.2F, 6.3, 6.3, 4.0, null)); |
| |
| assertRows(execute("SELECT CAST(a AS decimal), " + |
| "CAST(b AS decimal), " + |
| "CAST(c AS decimal), " + |
| "CAST(d AS decimal), " + |
| "CAST(e AS decimal), " + |
| "CAST(f AS decimal), " + |
| "CAST(g AS decimal), " + |
| "CAST(h AS decimal), " + |
| "CAST(i AS decimal) FROM %s"), |
| row(BigDecimal.valueOf(1), |
| BigDecimal.valueOf(2), |
| BigDecimal.valueOf(3), |
| BigDecimal.valueOf(4), |
| BigDecimal.valueOf(5.2F), |
| BigDecimal.valueOf(6.3), |
| BigDecimal.valueOf(6.3), |
| BigDecimal.valueOf(4), |
| null)); |
| |
| assertRows(execute("SELECT CAST(a AS ascii), " + |
| "CAST(b AS ascii), " + |
| "CAST(c AS ascii), " + |
| "CAST(d AS ascii), " + |
| "CAST(e AS ascii), " + |
| "CAST(f AS ascii), " + |
| "CAST(g AS ascii), " + |
| "CAST(h AS ascii), " + |
| "CAST(i AS ascii) FROM %s"), |
| row("1", |
| "2", |
| "3", |
| "4", |
| "5.2", |
| "6.3", |
| "6.3", |
| "4", |
| null)); |
| |
| assertRows(execute("SELECT CAST(a AS text), " + |
| "CAST(b AS text), " + |
| "CAST(c AS text), " + |
| "CAST(d AS text), " + |
| "CAST(e AS text), " + |
| "CAST(f AS text), " + |
| "CAST(g AS text), " + |
| "CAST(h AS text), " + |
| "CAST(i AS text) FROM %s"), |
| row("1", |
| "2", |
| "3", |
| "4", |
| "5.2", |
| "6.3", |
| "6.3", |
| "4", |
| null)); |
| } |
| |
| @Test |
| public void testNoLossOfPrecisionForCastToDecimal() throws Throwable |
| { |
| createTable("CREATE TABLE %s (k int PRIMARY KEY, bigint_clmn bigint, varint_clmn varint)"); |
| execute("INSERT INTO %s(k, bigint_clmn, varint_clmn) VALUES(2, 9223372036854775807, 1234567890123456789)"); |
| |
| assertRows(execute("SELECT CAST(bigint_clmn AS decimal), CAST(varint_clmn AS decimal) FROM %s"), |
| row(BigDecimal.valueOf(9223372036854775807L), BigDecimal.valueOf(1234567890123456789L))); |
| } |
| |
| @Test |
| public void testTimeCastsInSelectionClause() throws Throwable |
| { |
| createTable("CREATE TABLE %s (a timeuuid primary key, b timestamp, c date, d time)"); |
| |
| DateTime dateTime = DateTimeFormat.forPattern("yyyy-MM-dd hh:mm:ss") |
| .withZone(DateTimeZone.UTC) |
| .parseDateTime("2015-05-21 11:03:02"); |
| |
| DateTime date = DateTimeFormat.forPattern("yyyy-MM-dd") |
| .withZone(DateTimeZone.UTC) |
| .parseDateTime("2015-05-21"); |
| |
| long timeInMillis = dateTime.getMillis(); |
| |
| execute("INSERT INTO %s (a, b, c, d) VALUES (?, '2015-05-21 11:03:02+00', '2015-05-21', '11:03:02')", |
| UUIDGen.getTimeUUID(timeInMillis)); |
| |
| assertRows(execute("SELECT CAST(a AS timestamp), " + |
| "CAST(b AS timestamp), " + |
| "CAST(c AS timestamp) FROM %s"), |
| row(new Date(dateTime.getMillis()), new Date(dateTime.getMillis()), new Date(date.getMillis()))); |
| |
| int timeInMillisToDay = SimpleDateSerializer.timeInMillisToDay(date.getMillis()); |
| assertRows(execute("SELECT CAST(a AS date), " + |
| "CAST(b AS date), " + |
| "CAST(c AS date) FROM %s"), |
| row(timeInMillisToDay, timeInMillisToDay, timeInMillisToDay)); |
| |
| assertRows(execute("SELECT CAST(b AS text), " + |
| "CAST(c AS text), " + |
| "CAST(d AS text) FROM %s"), |
| row("2015-05-21T11:03:02.000Z", "2015-05-21", "11:03:02.000000000")); |
| } |
| |
| @Test |
| public void testOtherTypeCastsInSelectionClause() throws Throwable |
| { |
| createTable("CREATE TABLE %s (a ascii primary key," |
| + " b inet," |
| + " c boolean)"); |
| |
| execute("INSERT INTO %s (a, b, c) VALUES (?, '127.0.0.1', ?)", |
| "test", true); |
| |
| assertRows(execute("SELECT CAST(a AS text), " + |
| "CAST(b AS text), " + |
| "CAST(c AS text) FROM %s"), |
| row("test", "127.0.0.1", "true")); |
| } |
| |
| @Test |
| public void testCastsWithReverseOrder() throws Throwable |
| { |
| createTable("CREATE TABLE %s (a int," |
| + " b smallint," |
| + " c double," |
| + " primary key (a, b)) WITH CLUSTERING ORDER BY (b DESC);"); |
| |
| execute("INSERT INTO %s (a, b, c) VALUES (?, ?, ?)", |
| 1, (short) 2, 6.3); |
| |
| assertRows(execute("SELECT CAST(a AS tinyint), " + |
| "CAST(b AS tinyint), " + |
| "CAST(c AS tinyint) FROM %s"), |
| row((byte) 1, (byte) 2, (byte) 6)); |
| |
| assertRows(execute("SELECT CAST(CAST(a AS tinyint) AS smallint), " + |
| "CAST(CAST(b AS tinyint) AS smallint), " + |
| "CAST(CAST(c AS tinyint) AS smallint) FROM %s"), |
| row((short) 1, (short) 2, (short) 6)); |
| |
| assertRows(execute("SELECT CAST(CAST(CAST(a AS tinyint) AS double) AS text), " + |
| "CAST(CAST(CAST(b AS tinyint) AS double) AS text), " + |
| "CAST(CAST(CAST(c AS tinyint) AS double) AS text) FROM %s"), |
| row("1.0", "2.0", "6.0")); |
| |
| String f = createFunction(KEYSPACE, "int", |
| "CREATE FUNCTION %s(val int) " + |
| "RETURNS NULL ON NULL INPUT " + |
| "RETURNS double " + |
| "LANGUAGE java " + |
| "AS 'return (double)val;'"); |
| |
| assertRows(execute("SELECT " + f + "(CAST(b AS int)) FROM %s"), |
| row((double) 2)); |
| |
| assertRows(execute("SELECT CAST(" + f + "(CAST(b AS int)) AS text) FROM %s"), |
| row("2.0")); |
| } |
| |
| @Test |
| public void testCounterCastsInSelectionClause() throws Throwable |
| { |
| createTable("CREATE TABLE %s (a int primary key, b counter)"); |
| |
| execute("UPDATE %s SET b = b + 2 WHERE a = 1"); |
| |
| assertRows(execute("SELECT CAST(b AS tinyint), " + |
| "CAST(b AS smallint), " + |
| "CAST(b AS int), " + |
| "CAST(b AS bigint), " + |
| "CAST(b AS float), " + |
| "CAST(b AS double), " + |
| "CAST(b AS decimal), " + |
| "CAST(b AS ascii), " + |
| "CAST(b AS text) FROM %s"), |
| row((byte) 2, (short) 2, 2, 2L, 2.0F, 2.0, BigDecimal.valueOf(2), "2", "2")); |
| } |
| } |
