jffwang
/
hetu-core
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

support time column and testcase for all data types

This commit is contained in:
aloknath 2021-11-08 18:24:42 +05:30
parent c4d2858d33
commit bea799f5f2
2 changed files with 80 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
presto-mysql/src/main/java/io/prestosql/plugin/mysql/MySqlClient.java
View File

@ -88,6 +88,7 @@ import static io.prestosql.plugin.jdbc.JdbcErrorCode.JDBC_ERROR;
import static io.prestosql.plugin.jdbc.JdbcErrorCode.JDBC_QUERY_GENERATOR_FAILURE;
import static io.prestosql.plugin.jdbc.JdbcErrorCode.JDBC_UNSUPPORTED_EXPRESSION;
import static io.prestosql.plugin.jdbc.StandardColumnMappings.realWriteFunction;
import static io.prestosql.plugin.jdbc.StandardColumnMappings.timeWriteFunction;
import static io.prestosql.plugin.jdbc.StandardColumnMappings.timestampWriteFunctionUsingSqlTimestamp;
import static io.prestosql.plugin.jdbc.StandardColumnMappings.varbinaryWriteFunction;
import static io.prestosql.plugin.jdbc.StandardColumnMappings.varcharWriteFunction;
@ -98,6 +99,7 @@ import static io.prestosql.spi.StandardErrorCode.INVALID_FUNCTION_ARGUMENT;
import static io.prestosql.spi.StandardErrorCode.NOT_SUPPORTED;
import static io.prestosql.spi.type.Decimals.MAX_PRECISION;
import static io.prestosql.spi.type.RealType.REAL;
import static io.prestosql.spi.type.TimeType.TIME;
import static io.prestosql.spi.type.TimeWithTimeZoneType.TIME_WITH_TIME_ZONE;
import static io.prestosql.spi.type.TimestampType.TIMESTAMP;
import static io.prestosql.spi.type.TimestampWithTimeZoneType.TIMESTAMP_WITH_TIME_ZONE;
@ -203,6 +205,9 @@ public class MySqlClient
if (REAL.equals(type)) {
return WriteMapping.longMapping("float", realWriteFunction());
}
if (TIME.equals(type)) {
return WriteMapping.longMapping("time", timeWriteFunction());
}
if (TIME_WITH_TIME_ZONE.equals(type) || TIMESTAMP_WITH_TIME_ZONE.equals(type)) {
throw new PrestoException(NOT_SUPPORTED, "Unsupported column type: " + type.getDisplayName());
}

75
presto-mysql/src/test/java/io/prestosql/plugin/mysql/TestMySqlIntegrationSmokeTest.java
View File

@ -111,6 +111,81 @@ public class TestMySqlIntegrationSmokeTest
assertUpdate("DROP TABLE test_insert");
}
@Test
public void testDateTime()
throws Exception
{
Session session = testSessionBuilder()
.setCatalog("mysql")
.setSchema("test_database")
.build();
assertUpdate("CREATE TABLE mysql.test_database.testDateTime(a date,b time,c timestamp)");
assertUpdate("INSERT INTO mysql.test_database.testDateTime VALUES (date '2001-08-22', TIME '23:00:01.000',timestamp '2001-08-22 03:04:05')", 1);
assertQuery("SELECT * FROM mysql.test_database.testDateTime", "SELECT '2001-08-22' a,'23:00:01.000' b, '2001-08-22 03:04:05' c");
assertUpdate("DROP TABLE mysql.test_database.testDateTime");
}
@Test
public void integerColumn()
throws Exception
{
Session session = testSessionBuilder()
.setCatalog("mysql")
.setSchema("test_database")
.build();
assertUpdate("CREATE TABLE mysql.test_database.testInteger(a tinyint, b smallint, c integer, d bigint)");
assertUpdate("INSERT INTO mysql.test_database.testInteger VALUES (tinyint '-2', smallint '1', 5, 158)", 1);
assertQuery("SELECT * FROM mysql.test_database.testInteger", "SELECT -2 a, 1 b, 5 c, 158 d");
assertUpdate("DROP TABLE mysql.test_database.testInteger");
}
@Test
public void floatingPointColumn()
throws Exception
{
Session session = testSessionBuilder()
.setCatalog("mysql")
.setSchema("test_database")
.build();
assertUpdate("CREATE TABLE mysql.test_database.testFloating(a real, b double)");
assertUpdate("INSERT INTO mysql.test_database.testFloating VALUES (1.987, 3.1487596)", 1);
assertQuery("SELECT * FROM mysql.test_database.testFloating", "SELECT 1.987 a, 3.1487596 b");
assertUpdate("DROP TABLE mysql.test_database.testFloating");
}
@Test
public void fixedPrecisionColumn()
throws Exception
{
Session session = testSessionBuilder()
.setCatalog("mysql")
.setSchema("test_database")
.build();
assertUpdate("CREATE TABLE mysql.test_database.testFixed(a decimal(10,3))");
assertUpdate("INSERT INTO mysql.test_database.testFixed VALUES (1234567.123)", 1);
assertQuery("SELECT * FROM mysql.test_database.testFixed", "SELECT 1234567.123 a");
assertUpdate("DROP TABLE mysql.test_database.testFixed");
}
@Test
public void stringColumn()
throws Exception
{
Session session = testSessionBuilder()
.setCatalog("mysql")
.setSchema("test_database")
.build();
assertUpdate("CREATE TABLE mysql.test_database.testString(a string, b varchar(5), c char(5), d json)");
assertUpdate("INSERT INTO mysql.test_database.testString VALUES ('olk', 'abcd', 'ftr', json '[{\"street\": \"street address\", \"city\": \"Berlin\"}]')", 1);
assertQuery("SELECT * FROM mysql.test_database.testString", "SELECT 'olk' a, 'abcd' b, 'ftr' c , '[{\"street\": \"street address\", \"city\": \"Berlin\"}]' d");
assertUpdate("DROP TABLE mysql.test_database.testString");
}
@Test
public void testNameEscaping()
{