SQL: Consolidate JDBC type handling (elastic/x-pack-elasticsearch#3761)

Consolidates handling of JDBC types conversion into a single file that should simplify maintaining consistency between type handling. Also separates the types that are handled as part of Elasticsearch output and types that are handled as user-supplied parameters.

relates elastic/x-pack-elasticsearch#3556

Original commit: elastic/x-pack-elasticsearch@d251fce66b
This commit is contained in:
Igor Motov 2018-01-29 16:35:42 -05:00 committed by GitHub
parent 07658cc04f
commit 1627ec1378
12 changed files with 308 additions and 337 deletions

View File

@ -22,6 +22,9 @@ import java.sql.SQLFeatureNotSupportedException;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
import static java.sql.JDBCType.INTEGER;
import static java.sql.JDBCType.SMALLINT;
/** /**
* Implementation of {@link DatabaseMetaData} for Elasticsearch. Draws inspiration * Implementation of {@link DatabaseMetaData} for Elasticsearch. Draws inspiration
* from <a href="https://www.postgresql.org/docs/9.0/static/information-schema.html"> * from <a href="https://www.postgresql.org/docs/9.0/static/information-schema.html">
@ -640,11 +643,11 @@ class JdbcDatabaseMetaData implements DatabaseMetaData, JdbcWrapper {
"PROCEDURE_CAT", "PROCEDURE_CAT",
"PROCEDURE_SCHEM", "PROCEDURE_SCHEM",
"PROCEDURE_NAME", "PROCEDURE_NAME",
"NUM_INPUT_PARAMS", int.class, "NUM_INPUT_PARAMS", INTEGER,
"NUM_OUTPUT_PARAMS", int.class, "NUM_OUTPUT_PARAMS", INTEGER,
"NUM_RESULT_SETS", int.class, "NUM_RESULT_SETS", INTEGER,
"REMARKS", "REMARKS",
"PROCEDURE_TYPE", short.class, "PROCEDURE_TYPE", SMALLINT,
"SPECIFIC_NAME"); "SPECIFIC_NAME");
} }
@ -657,20 +660,20 @@ class JdbcDatabaseMetaData implements DatabaseMetaData, JdbcWrapper {
"PROCEDURE_SCHEM", "PROCEDURE_SCHEM",
"PROCEDURE_NAME", "PROCEDURE_NAME",
"COLUMN_NAME", "COLUMN_NAME",
"COLUMN_TYPE", short.class, "COLUMN_TYPE", SMALLINT,
"DATA_TYPE", int.class, "DATA_TYPE", INTEGER,
"TYPE_NAME", "TYPE_NAME",
"PRECISION", int.class, "PRECISION", INTEGER,
"LENGTH", int.class, "LENGTH", INTEGER,
"SCALE", short.class, "SCALE", SMALLINT,
"RADIX", short.class, "RADIX", SMALLINT,
"NULLABLE", short.class, "NULLABLE", SMALLINT,
"REMARKS", "REMARKS",
"COLUMN_DEF", "COLUMN_DEF",
"SQL_DATA_TYPE", int.class, "SQL_DATA_TYPE", INTEGER,
"SQL_DATETIME_SUB", int.class, "SQL_DATETIME_SUB", INTEGER,
"CHAR_OCTET_LENGTH", int.class, "CHAR_OCTET_LENGTH", INTEGER,
"ORDINAL_POSITION", int.class, "ORDINAL_POSITION", INTEGER,
"IS_NULLABLE", "IS_NULLABLE",
"SPECIFIC_NAME"); "SPECIFIC_NAME");
} }
@ -739,7 +742,6 @@ class JdbcDatabaseMetaData implements DatabaseMetaData, JdbcWrapper {
@Override @Override
public ResultSet getColumns(String catalog, String schemaPattern, String tableNamePattern, String columnNamePattern) public ResultSet getColumns(String catalog, String schemaPattern, String tableNamePattern, String columnNamePattern)
throws SQLException { throws SQLException {
PreparedStatement ps = con.prepareStatement("SYS COLUMNS TABLES LIKE ? LIKE ?"); PreparedStatement ps = con.prepareStatement("SYS COLUMNS TABLES LIKE ? LIKE ?");
ps.setString(1, tableNamePattern != null ? tableNamePattern.trim() : "%"); ps.setString(1, tableNamePattern != null ? tableNamePattern.trim() : "%");
ps.setString(2, columnNamePattern != null ? columnNamePattern.trim() : "%"); ps.setString(2, columnNamePattern != null ? columnNamePattern.trim() : "%");
@ -865,9 +867,9 @@ class JdbcDatabaseMetaData implements DatabaseMetaData, JdbcWrapper {
"TYPE_SCHEM", "TYPE_SCHEM",
"TYPE_NAME", "TYPE_NAME",
"CLASS_NAME", "CLASS_NAME",
"DATA_TYPE", int.class, "DATA_TYPE", INTEGER,
"REMARKS", "REMARKS",
"BASE_TYPE", short.class); "BASE_TYPE", SMALLINT);
} }
@Override @Override
@ -926,23 +928,23 @@ class JdbcDatabaseMetaData implements DatabaseMetaData, JdbcWrapper {
"TYPE_SCHEM", "TYPE_SCHEM",
"TYPE_NAME", "TYPE_NAME",
"ATTR_NAME", "ATTR_NAME",
"DATA_TYPE", int.class, "DATA_TYPE", INTEGER,
"ATTR_TYPE_NAME", "ATTR_TYPE_NAME",
"ATTR_SIZE", int.class, "ATTR_SIZE", INTEGER,
"DECIMAL_DIGITS", int.class, "DECIMAL_DIGITS", INTEGER,
"NUM_PREC_RADIX", int.class, "NUM_PREC_RADIX", INTEGER,
"NULLABLE", int.class, "NULLABLE", INTEGER,
"REMARKS", "REMARKS",
"ATTR_DEF", "ATTR_DEF",
"SQL_DATA_TYPE", int.class, "SQL_DATA_TYPE", INTEGER,
"SQL_DATETIME_SUB", int.class, "SQL_DATETIME_SUB", INTEGER,
"CHAR_OCTET_LENGTH", int.class, "CHAR_OCTET_LENGTH", INTEGER,
"ORDINAL_POSITION", int.class, "ORDINAL_POSITION", INTEGER,
"IS_NULLABLE", "IS_NULLABLE",
"SCOPE_CATALOG", "SCOPE_CATALOG",
"SCOPE_SCHEMA", "SCOPE_SCHEMA",
"SCOPE_TABLE", "SCOPE_TABLE",
"SOURCE_DATA_TYPE", short.class); "SOURCE_DATA_TYPE", SMALLINT);
} }
@Override @Override
@ -1018,7 +1020,7 @@ class JdbcDatabaseMetaData implements DatabaseMetaData, JdbcWrapper {
"FUNCTION_SCHEM", "FUNCTION_SCHEM",
"FUNCTION_NAME", "FUNCTION_NAME",
"REMARKS", "REMARKS",
"FUNCTION_TYPE", short.class, "FUNCTION_TYPE", SMALLINT,
"SPECIFIC_NAME"); "SPECIFIC_NAME");
} }
@ -1031,16 +1033,16 @@ class JdbcDatabaseMetaData implements DatabaseMetaData, JdbcWrapper {
"FUNCTION_SCHEM", "FUNCTION_SCHEM",
"FUNCTION_NAME", "FUNCTION_NAME",
"COLUMN_NAME", "COLUMN_NAME",
"DATA_TYPE", int.class, "DATA_TYPE", INTEGER,
"TYPE_NAME", "TYPE_NAME",
"PRECISION", int.class, "PRECISION", INTEGER,
"LENGTH", int.class, "LENGTH", INTEGER,
"SCALE", short.class, "SCALE", SMALLINT,
"RADIX", short.class, "RADIX", SMALLINT,
"NULLABLE", short.class, "NULLABLE", SMALLINT,
"REMARKS", "REMARKS",
"CHAR_OCTET_LENGTH", int.class, "CHAR_OCTET_LENGTH", INTEGER,
"ORDINAL_POSITION", int.class, "ORDINAL_POSITION", INTEGER,
"IS_NULLABLE", "IS_NULLABLE",
"SPECIFIC_NAME"); "SPECIFIC_NAME");
} }
@ -1054,10 +1056,10 @@ class JdbcDatabaseMetaData implements DatabaseMetaData, JdbcWrapper {
"TABLE_SCHEM", "TABLE_SCHEM",
"TABLE_NAME", "TABLE_NAME",
"COLUMN_NAME", "COLUMN_NAME",
"DATA_TYPE", int.class, "DATA_TYPE", INTEGER,
"COLUMN_SIZE", int.class, "COLUMN_SIZE", INTEGER,
"DECIMAL_DIGITS", int.class, "DECIMAL_DIGITS", INTEGER,
"NUM_PREC_RADIX", int.class, "NUM_PREC_RADIX", INTEGER,
"REMARKS", "REMARKS",
"COLUMN_USAGE", "COLUMN_USAGE",
"IS_NULLABLE"); "IS_NULLABLE");
@ -1078,8 +1080,8 @@ class JdbcDatabaseMetaData implements DatabaseMetaData, JdbcWrapper {
JDBCType type = JDBCType.VARCHAR; JDBCType type = JDBCType.VARCHAR;
if (i + 1 < cols.length) { if (i + 1 < cols.length) {
// check if the next item it's a type // check if the next item it's a type
if (cols[i + 1] instanceof Class) { if (cols[i + 1] instanceof JDBCType) {
type = JDBCType.valueOf(JdbcUtils.fromClass((Class<?>) cols[i + 1])); type = (JDBCType) cols[i + 1];
i++; i++;
} }
// it's not, use the default and move on // it's not, use the default and move on

View File

@ -32,7 +32,7 @@ class JdbcParameterMetaData implements ParameterMetaData, JdbcWrapper {
@Override @Override
public boolean isSigned(int param) throws SQLException { public boolean isSigned(int param) throws SQLException {
return JdbcUtils.isSigned(paramInfo(param).type.getVendorTypeNumber().intValue()); return TypeConverter.isSigned(paramInfo(param).type);
} }
@Override @Override
@ -49,7 +49,7 @@ class JdbcParameterMetaData implements ParameterMetaData, JdbcWrapper {
@Override @Override
public int getParameterType(int param) throws SQLException { public int getParameterType(int param) throws SQLException {
return paramInfo(param).type.getVendorTypeNumber().intValue(); return paramInfo(param).type.getVendorTypeNumber();
} }
@Override @Override
@ -59,7 +59,7 @@ class JdbcParameterMetaData implements ParameterMetaData, JdbcWrapper {
@Override @Override
public String getParameterClassName(int param) throws SQLException { public String getParameterClassName(int param) throws SQLException {
return JdbcUtils.classOf(paramInfo(param).type.getVendorTypeNumber()).getName(); return TypeConverter.classNameOf(paramInfo(param).type);
} }
@Override @Override

View File

@ -369,12 +369,7 @@ class JdbcResultSet implements ResultSet, JdbcWrapper {
JDBCType columnType = cursor.columns().get(columnIndex - 1).type; JDBCType columnType = cursor.columns().get(columnIndex - 1).type;
T t = TypeConverter.convert(val, columnType, type); return TypeConverter.convert(val, columnType, type);
if (t != null || type == null) {
return t;
}
throw new SQLException("Conversion from type [" + columnType + "] to [" + type.getName() + "] not supported");
} }
@Override @Override

View File

@ -62,7 +62,7 @@ class JdbcResultSetMetaData implements ResultSetMetaData, JdbcWrapper {
@Override @Override
public boolean isSigned(int column) throws SQLException { public boolean isSigned(int column) throws SQLException {
return JdbcUtils.isSigned(getColumnType(column)); return TypeConverter.isSigned(column(column).type);
} }
@Override @Override
@ -137,7 +137,7 @@ class JdbcResultSetMetaData implements ResultSetMetaData, JdbcWrapper {
@Override @Override
public String getColumnClassName(int column) throws SQLException { public String getColumnClassName(int column) throws SQLException {
return JdbcUtils.classOf(column(column).type.getVendorTypeNumber()).getName(); return TypeConverter.classNameOf(column(column).type);
} }
private void checkOpen() throws SQLException { private void checkOpen() throws SQLException {

View File

@ -1,169 +0,0 @@
/*
* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
* or more contributor license agreements. Licensed under the Elastic License;
* you may not use this file except in compliance with the Elastic License.
*/
package org.elasticsearch.xpack.sql.jdbc.jdbc;
import org.elasticsearch.xpack.sql.jdbc.JdbcSQLException;
import java.math.BigDecimal;
import java.sql.Blob;
import java.sql.Clob;
import java.sql.Date;
import java.sql.JDBCType;
import java.sql.Time;
import java.sql.Timestamp;
import static java.sql.Types.BIGINT;
import static java.sql.Types.BINARY;
import static java.sql.Types.BIT;
import static java.sql.Types.BLOB;
import static java.sql.Types.BOOLEAN;
import static java.sql.Types.CHAR;
import static java.sql.Types.CLOB;
import static java.sql.Types.DATE;
import static java.sql.Types.DECIMAL;
import static java.sql.Types.DOUBLE;
import static java.sql.Types.FLOAT;
import static java.sql.Types.INTEGER;
import static java.sql.Types.LONGVARBINARY;
import static java.sql.Types.LONGVARCHAR;
import static java.sql.Types.NULL;
import static java.sql.Types.NUMERIC;
import static java.sql.Types.REAL;
import static java.sql.Types.SMALLINT;
import static java.sql.Types.TIME;
import static java.sql.Types.TIMESTAMP;
import static java.sql.Types.TIMESTAMP_WITH_TIMEZONE;
import static java.sql.Types.TINYINT;
import static java.sql.Types.VARBINARY;
import static java.sql.Types.VARCHAR;
public abstract class JdbcUtils {
public static int fromClass(Class<?> clazz) throws JdbcSQLException {
if (clazz == null) {
return NULL;
}
if (clazz == String.class) {
return VARCHAR;
}
if (clazz == Boolean.class || clazz == boolean.class) {
return BOOLEAN;
}
if (clazz == Byte.class || clazz == byte.class) {
return TINYINT;
}
if (clazz == Short.class || clazz == short.class) {
return SMALLINT;
}
if (clazz == Integer.class || clazz == int.class) {
return INTEGER;
}
if (clazz == Long.class || clazz == long.class) {
return BIGINT;
}
if (clazz == Float.class || clazz == float.class) {
return REAL;
}
if (clazz == Double.class || clazz == double.class) {
return DOUBLE;
}
if (clazz == Void.class || clazz == void.class) {
return NULL;
}
if (clazz == byte[].class) {
return VARBINARY;
}
if (clazz == Date.class) {
return DATE;
}
if (clazz == Time.class) {
return TIME;
}
if (clazz == Timestamp.class) {
return TIMESTAMP;
}
if (clazz == Blob.class) {
return BLOB;
}
if (clazz == Clob.class) {
return CLOB;
}
if (clazz == BigDecimal.class) {
return DECIMAL;
}
throw new JdbcSQLException("Unrecognized class [" + clazz + "]");
}
// see javax.sql.rowset.RowSetMetaDataImpl
// and https://db.apache.org/derby/docs/10.5/ref/rrefjdbc20377.html
public static Class<?> classOf(int jdbcType) throws JdbcSQLException {
switch (jdbcType) {
case NUMERIC:
case DECIMAL:
return BigDecimal.class;
case BOOLEAN:
case BIT:
return Boolean.class;
case TINYINT:
return Byte.class;
case SMALLINT:
return Short.class;
case INTEGER:
return Integer.class;
case BIGINT:
return Long.class;
case REAL:
return Float.class;
case FLOAT:
case DOUBLE:
return Double.class;
case BINARY:
case VARBINARY:
case LONGVARBINARY:
return byte[].class;
case CHAR:
case VARCHAR:
case LONGVARCHAR:
return String.class;
case DATE:
return Date.class;
case TIME:
return Time.class;
case TIMESTAMP:
return Timestamp.class;
case BLOB:
return Blob.class;
case CLOB:
return Clob.class;
case TIMESTAMP_WITH_TIMEZONE:
return Long.class;
default:
throw new JdbcSQLException("Unsupported JDBC type " + jdbcType + ", " + type(jdbcType).getName() + "");
}
}
static boolean isSigned(int type) {
switch (type) {
case BIGINT:
case DECIMAL:
case DOUBLE:
case FLOAT:
case INTEGER:
case SMALLINT:
case REAL:
case NUMERIC:
return true;
default:
return false;
}
}
static JDBCType type(int jdbcType) {
return JDBCType.valueOf(jdbcType);
}
}

View File

@ -5,6 +5,11 @@
*/ */
package org.elasticsearch.xpack.sql.jdbc.jdbc; package org.elasticsearch.xpack.sql.jdbc.jdbc;
import org.elasticsearch.xpack.sql.jdbc.JdbcSQLException;
import java.math.BigDecimal;
import java.sql.Blob;
import java.sql.Clob;
import java.sql.Date; import java.sql.Date;
import java.sql.JDBCType; import java.sql.JDBCType;
import java.sql.SQLException; import java.sql.SQLException;
@ -30,10 +35,28 @@ import static java.util.Calendar.MONTH;
import static java.util.Calendar.SECOND; import static java.util.Calendar.SECOND;
import static java.util.Calendar.YEAR; import static java.util.Calendar.YEAR;
abstract class TypeConverter { /**
* Conversion utilities for conversion of JDBC types to Java type and back
* <p>
* The following JDBC types are supported as part of Elasticsearch Response. See org.elasticsearch.xpack.sql.type.DataType for details.
* <p>
* NULL, BOOLEAN, TINYINT, SMALLINT, INTEGER, BIGINT, DOUBLE, REAL, FLOAT, VARCHAR, VARBINARY and TIMESTAMP
* <p>
* The following additional types are also supported as parameters:
* <p>
* NUMERIC, DECIMAL, BIT, BINARY, LONGVARBINARY, CHAR, LONGVARCHAR, DATE, TIME, BLOB, CLOB, TIMESTAMP_WITH_TIMEZONE
*/
final class TypeConverter {
private TypeConverter() {
}
private static final long DAY_IN_MILLIS = 60 * 60 * 24; private static final long DAY_IN_MILLIS = 60 * 60 * 24;
/**
* Converts millisecond after epoc to date
*/
static Date convertDate(Long millis, Calendar cal) { static Date convertDate(Long millis, Calendar cal) {
return dateTimeConvert(millis, cal, c -> { return dateTimeConvert(millis, cal, c -> {
c.set(HOUR_OF_DAY, 0); c.set(HOUR_OF_DAY, 0);
@ -44,6 +67,9 @@ abstract class TypeConverter {
}); });
} }
/**
* Converts millisecond after epoc to time
*/
static Time convertTime(Long millis, Calendar cal) { static Time convertTime(Long millis, Calendar cal) {
return dateTimeConvert(millis, cal, c -> { return dateTimeConvert(millis, cal, c -> {
c.set(ERA, GregorianCalendar.AD); c.set(ERA, GregorianCalendar.AD);
@ -54,10 +80,11 @@ abstract class TypeConverter {
}); });
} }
/**
* Converts millisecond after epoc to timestamp
*/
static Timestamp convertTimestamp(Long millis, Calendar cal) { static Timestamp convertTimestamp(Long millis, Calendar cal) {
return dateTimeConvert(millis, cal, c -> { return dateTimeConvert(millis, cal, c -> new Timestamp(c.getTimeInMillis()));
return new Timestamp(c.getTimeInMillis());
});
} }
private static <T> T dateTimeConvert(Long millis, Calendar c, Function<Calendar, T> creator) { private static <T> T dateTimeConvert(Long millis, Calendar c, Function<Calendar, T> creator) {
@ -66,20 +93,23 @@ abstract class TypeConverter {
} }
long initial = c.getTimeInMillis(); long initial = c.getTimeInMillis();
try { try {
c.setTimeInMillis(millis.longValue()); c.setTimeInMillis(millis);
return creator.apply(c); return creator.apply(c);
} finally { } finally {
c.setTimeInMillis(initial); c.setTimeInMillis(initial);
} }
} }
/**
* Converts object val from columnType to type
*/
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
static <T> T convert(Object val, JDBCType columnType, Class<T> type) throws SQLException { static <T> T convert(Object val, JDBCType columnType, Class<T> type) throws SQLException {
if (type == null) { if (type == null) {
return (T) asNative(val, columnType); return (T) convert(val, columnType);
} }
if (type == String.class) { if (type == String.class) {
return (T) asString(asNative(val, columnType)); return (T) asString(convert(val, columnType));
} }
if (type == Boolean.class) { if (type == Boolean.class) {
return (T) asBoolean(val, columnType); return (T) asBoolean(val, columnType);
@ -132,43 +162,145 @@ abstract class TypeConverter {
if (type == OffsetDateTime.class) { if (type == OffsetDateTime.class) {
return (T) asOffsetDateTime(val, columnType); return (T) asOffsetDateTime(val, columnType);
} }
return null; throw new SQLException("Conversion from type [" + columnType + "] to [" + type.getName() + "] not supported");
} }
// keep in check with JdbcUtils#columnType /**
static Object asNative(Object v, JDBCType columnType) { * Translates numeric JDBC type into corresponding Java class
switch (columnType) { * <p>
* See {@link javax.sql.rowset.RowSetMetaDataImpl#getColumnClassName} and
* https://db.apache.org/derby/docs/10.5/ref/rrefjdbc20377.html
*/
public static String classNameOf(JDBCType jdbcType) throws JdbcSQLException {
switch (jdbcType) {
// ES - supported types
case BOOLEAN:
return Boolean.class.getName();
case TINYINT: // BYTE DataType
return Byte.class.getName();
case SMALLINT: // SHORT DataType
return Short.class.getName();
case INTEGER:
return Integer.class.getName();
case BIGINT: // LONG DataType
return Long.class.getName();
case DOUBLE:
return Double.class.getName();
case REAL: // FLOAT DataType
return Float.class.getName();
case FLOAT: // HALF_FLOAT DataType
return Double.class.getName(); // TODO: Is this correct?
case VARCHAR: // KEYWORD or TEXT DataType
return String.class.getName();
case VARBINARY: // BINARY DataType
return byte[].class.getName();
case TIMESTAMP: // DATE DataType
return Timestamp.class.getName();
// Parameters data types that cannot be returned by ES but can appear in client - supplied parameters
case NUMERIC:
case DECIMAL:
return BigDecimal.class.getName();
case BIT: case BIT:
return Boolean.class.getName();
case BINARY:
case LONGVARBINARY:
return byte[].class.getName();
case CHAR:
case LONGVARCHAR:
return String.class.getName();
case DATE:
return Date.class.getName();
case TIME:
return Time.class.getName();
case BLOB:
return Blob.class.getName();
case CLOB:
return Clob.class.getName();
case TIMESTAMP_WITH_TIMEZONE:
return Long.class.getName();
default:
throw new JdbcSQLException("Unsupported JDBC type [" + jdbcType + "]");
}
}
/**
* Converts the object from JSON representation to the specified JDBCType
* <p>
* The returned types needs to correspond to ES-portion of classes returned by {@link TypeConverter#classNameOf}
*/
static Object convert(Object v, JDBCType columnType) throws SQLException {
switch (columnType) {
case NULL:
return null;
case BOOLEAN: case BOOLEAN:
case BINARY: case BINARY:
case VARBINARY: case VARBINARY:
case LONGVARBINARY:
case CHAR:
case VARCHAR: case VARCHAR:
case LONGVARCHAR: return v; // These types are already represented correctly in JSON
return v;
case TINYINT: case TINYINT:
return ((Number) v).byteValue(); return ((Number) v).byteValue(); // Parser might return it as integer or long - need to update to the correct type
case SMALLINT: case SMALLINT:
return ((Number) v).shortValue(); return ((Number) v).shortValue(); // Parser might return it as integer or long - need to update to the correct type
case INTEGER: case INTEGER:
return ((Number) v).intValue(); return ((Number) v).intValue();
case BIGINT: case BIGINT:
return ((Number) v).longValue(); return ((Number) v).longValue();
case FLOAT: case FLOAT:
case DOUBLE: case DOUBLE:
return doubleValue(v); return doubleValue(v); // Double might be represented as string for infinity and NaN values
case REAL: case REAL:
return floatValue(v); return floatValue(v); // Float might be represented as string for infinity and NaN values
case TIMESTAMP: case TIMESTAMP:
return ((Number) v).longValue(); return ((Number) v).longValue();
// since the date is already in UTC_CALENDAR just do calendar math
case DATE:
return new Date(utcMillisRemoveTime(((Number) v).longValue()));
case TIME:
return new Time(utcMillisRemoveDate(((Number) v).longValue()));
default: default:
return null; throw new SQLException("Unexpected column type [" + columnType.getName() + "]");
}
}
/**
* Returns true if the type represents a signed number, false otherwise
* <p>
* It needs to support both params and column types
*/
static boolean isSigned(JDBCType type) throws SQLException {
switch (type) {
// ES Supported types
case BIGINT:
case DOUBLE:
case FLOAT:
case INTEGER:
case TINYINT:
case SMALLINT:
return true;
case NULL:
case BOOLEAN:
case VARCHAR:
case VARBINARY:
case TIMESTAMP:
return false;
// Parameter types
case REAL:
case DECIMAL:
case NUMERIC:
return true;
case BIT:
case BINARY:
case LONGVARBINARY:
case CHAR:
case LONGVARCHAR:
case DATE:
case TIME:
case BLOB:
case CLOB:
case TIMESTAMP_WITH_TIMEZONE:
return false;
default:
throw new SQLException("Unexpected column or parameter type [" + type + "]");
} }
} }
@ -208,9 +340,8 @@ abstract class TypeConverter {
return nativeValue == null ? null : String.valueOf(nativeValue); return nativeValue == null ? null : String.valueOf(nativeValue);
} }
private static Boolean asBoolean(Object val, JDBCType columnType) { private static Boolean asBoolean(Object val, JDBCType columnType) throws SQLException {
switch (columnType) { switch (columnType) {
case BIT:
case BOOLEAN: case BOOLEAN:
case TINYINT: case TINYINT:
case SMALLINT: case SMALLINT:
@ -221,13 +352,13 @@ abstract class TypeConverter {
case DOUBLE: case DOUBLE:
return Boolean.valueOf(Integer.signum(((Number) val).intValue()) == 0); return Boolean.valueOf(Integer.signum(((Number) val).intValue()) == 0);
default: default:
return null; throw new SQLException("Conversion from type [" + columnType + "] to [Boolean] not supported");
} }
} }
private static Byte asByte(Object val, JDBCType columnType) throws SQLException { private static Byte asByte(Object val, JDBCType columnType) throws SQLException {
switch (columnType) { switch (columnType) {
case BIT:
case BOOLEAN: case BOOLEAN:
return Byte.valueOf(((Boolean) val).booleanValue() ? (byte) 1 : (byte) 0); return Byte.valueOf(((Boolean) val).booleanValue() ? (byte) 1 : (byte) 0);
case TINYINT: case TINYINT:
@ -242,12 +373,11 @@ abstract class TypeConverter {
default: default:
} }
return null; throw new SQLException("Conversion from type [" + columnType + "] to [Byte] not supported");
} }
private static Short asShort(Object val, JDBCType columnType) throws SQLException { private static Short asShort(Object val, JDBCType columnType) throws SQLException {
switch (columnType) { switch (columnType) {
case BIT:
case BOOLEAN: case BOOLEAN:
return Short.valueOf(((Boolean) val).booleanValue() ? (short) 1 : (short) 0); return Short.valueOf(((Boolean) val).booleanValue() ? (short) 1 : (short) 0);
case TINYINT: case TINYINT:
@ -262,12 +392,11 @@ abstract class TypeConverter {
default: default:
} }
return null; throw new SQLException("Conversion from type [" + columnType + "] to [Short] not supported");
} }
private static Integer asInteger(Object val, JDBCType columnType) throws SQLException { private static Integer asInteger(Object val, JDBCType columnType) throws SQLException {
switch (columnType) { switch (columnType) {
case BIT:
case BOOLEAN: case BOOLEAN:
return Integer.valueOf(((Boolean) val).booleanValue() ? 1 : 0); return Integer.valueOf(((Boolean) val).booleanValue() ? 1 : 0);
case TINYINT: case TINYINT:
@ -282,12 +411,11 @@ abstract class TypeConverter {
default: default:
} }
return null; throw new SQLException("Conversion from type [" + columnType + "] to [Integer] not supported");
} }
private static Long asLong(Object val, JDBCType columnType) throws SQLException { private static Long asLong(Object val, JDBCType columnType) throws SQLException {
switch (columnType) { switch (columnType) {
case BIT:
case BOOLEAN: case BOOLEAN:
return Long.valueOf(((Boolean) val).booleanValue() ? 1 : 0); return Long.valueOf(((Boolean) val).booleanValue() ? 1 : 0);
case TINYINT: case TINYINT:
@ -309,12 +437,11 @@ abstract class TypeConverter {
default: default:
} }
return null; throw new SQLException("Conversion from type [" + columnType + "] to [Long] not supported");
} }
private static Float asFloat(Object val, JDBCType columnType) throws SQLException { private static Float asFloat(Object val, JDBCType columnType) throws SQLException {
switch (columnType) { switch (columnType) {
case BIT:
case BOOLEAN: case BOOLEAN:
return Float.valueOf(((Boolean) val).booleanValue() ? 1 : 0); return Float.valueOf(((Boolean) val).booleanValue() ? 1 : 0);
case TINYINT: case TINYINT:
@ -329,12 +456,11 @@ abstract class TypeConverter {
default: default:
} }
return null; throw new SQLException("Conversion from type [" + columnType + "] to [Float] not supported");
} }
private static Double asDouble(Object val, JDBCType columnType) throws SQLException { private static Double asDouble(Object val, JDBCType columnType) throws SQLException {
switch (columnType) { switch (columnType) {
case BIT:
case BOOLEAN: case BOOLEAN:
return Double.valueOf(((Boolean) val).booleanValue() ? 1 : 0); return Double.valueOf(((Boolean) val).booleanValue() ? 1 : 0);
case TINYINT: case TINYINT:
@ -349,7 +475,7 @@ abstract class TypeConverter {
default: default:
} }
return null; throw new SQLException("Conversion from type [" + columnType + "] to [Double] not supported");
} }
private static Date asDate(Object val, JDBCType columnType) throws SQLException { private static Date asDate(Object val, JDBCType columnType) throws SQLException {
@ -364,7 +490,7 @@ abstract class TypeConverter {
default: default:
} }
return null; throw new SQLException("Conversion from type [" + columnType + "] to [Date] not supported");
} }
private static Time asTime(Object val, JDBCType columnType) throws SQLException { private static Time asTime(Object val, JDBCType columnType) throws SQLException {
@ -379,7 +505,7 @@ abstract class TypeConverter {
default: default:
} }
return null; throw new SQLException("Conversion from type [" + columnType + "] to [Time] not supported");
} }
private static Timestamp asTimestamp(Object val, JDBCType columnType) throws SQLException { private static Timestamp asTimestamp(Object val, JDBCType columnType) throws SQLException {
@ -394,12 +520,13 @@ abstract class TypeConverter {
default: default:
} }
return null; throw new SQLException("Conversion from type [" + columnType + "] to [Timestamp] not supported");
} }
private static byte[] asByteArray(Object val, JDBCType columnType) { private static byte[] asByteArray(Object val, JDBCType columnType) {
throw new UnsupportedOperationException(); throw new UnsupportedOperationException();
} }
private static LocalDate asLocalDate(Object val, JDBCType columnType) { private static LocalDate asLocalDate(Object val, JDBCType columnType) {
throw new UnsupportedOperationException(); throw new UnsupportedOperationException();
} }

View File

@ -13,7 +13,6 @@ import org.elasticsearch.xpack.sql.plugin.AbstractSqlRequest;
import org.elasticsearch.xpack.sql.plugin.SqlQueryResponse; import org.elasticsearch.xpack.sql.plugin.SqlQueryResponse;
import org.joda.time.DateTime; import org.joda.time.DateTime;
import java.io.IOException;
import java.sql.JDBCType; import java.sql.JDBCType;
import static org.hamcrest.Matchers.instanceOf; import static org.hamcrest.Matchers.instanceOf;
@ -22,7 +21,7 @@ import static org.hamcrest.Matchers.instanceOf;
public class TypeConverterTests extends ESTestCase { public class TypeConverterTests extends ESTestCase {
public void testFloatAsNative() throws IOException { public void testFloatAsNative() throws Exception {
assertThat(convertAsNative(42.0f, JDBCType.REAL), instanceOf(Float.class)); assertThat(convertAsNative(42.0f, JDBCType.REAL), instanceOf(Float.class));
assertThat(convertAsNative(42.0, JDBCType.REAL), instanceOf(Float.class)); assertThat(convertAsNative(42.0, JDBCType.REAL), instanceOf(Float.class));
assertEquals(42.0f, (float) convertAsNative(42.0, JDBCType.REAL), 0.001f); assertEquals(42.0f, (float) convertAsNative(42.0, JDBCType.REAL), 0.001f);
@ -31,7 +30,7 @@ public class TypeConverterTests extends ESTestCase {
assertEquals(Float.POSITIVE_INFINITY, convertAsNative(Float.POSITIVE_INFINITY, JDBCType.REAL)); assertEquals(Float.POSITIVE_INFINITY, convertAsNative(Float.POSITIVE_INFINITY, JDBCType.REAL));
} }
public void testDoubleAsNative() throws IOException { public void testDoubleAsNative() throws Exception {
JDBCType type = randomFrom(JDBCType.FLOAT, JDBCType.DOUBLE); JDBCType type = randomFrom(JDBCType.FLOAT, JDBCType.DOUBLE);
assertThat(convertAsNative(42.0, type), instanceOf(Double.class)); assertThat(convertAsNative(42.0, type), instanceOf(Double.class));
assertEquals(42.0f, (double) convertAsNative(42.0, type), 0.001f); assertEquals(42.0f, (double) convertAsNative(42.0, type), 0.001f);
@ -40,13 +39,13 @@ public class TypeConverterTests extends ESTestCase {
assertEquals(Double.POSITIVE_INFINITY, convertAsNative(Double.POSITIVE_INFINITY, type)); assertEquals(Double.POSITIVE_INFINITY, convertAsNative(Double.POSITIVE_INFINITY, type));
} }
public void testTimestampAsNative() throws IOException { public void testTimestampAsNative() throws Exception {
DateTime now = DateTime.now(); DateTime now = DateTime.now();
assertThat(convertAsNative(now, JDBCType.TIMESTAMP), instanceOf(Long.class)); assertThat(convertAsNative(now, JDBCType.TIMESTAMP), instanceOf(Long.class));
assertEquals(now.getMillis(), convertAsNative(now, JDBCType.TIMESTAMP)); assertEquals(now.getMillis(), convertAsNative(now, JDBCType.TIMESTAMP));
} }
private Object convertAsNative(Object value, JDBCType type) throws IOException { private Object convertAsNative(Object value, JDBCType type) throws Exception {
// Simulate sending over XContent // Simulate sending over XContent
XContentBuilder builder = JsonXContent.contentBuilder(); XContentBuilder builder = JsonXContent.contentBuilder();
builder.startObject(); builder.startObject();
@ -55,7 +54,7 @@ public class TypeConverterTests extends ESTestCase {
builder.endObject(); builder.endObject();
builder.close(); builder.close();
Object copy = XContentHelper.convertToMap(builder.bytes(), false, builder.contentType()).v2().get("value"); Object copy = XContentHelper.convertToMap(builder.bytes(), false, builder.contentType()).v2().get("value");
return TypeConverter.asNative(copy, type); return TypeConverter.convert(copy, type);
} }
} }

View File

@ -132,7 +132,10 @@ public class SysColumns extends Command {
type.size, type.size,
// no DECIMAL support // no DECIMAL support
null, null,
// RADIX // RADIX - Determines how numbers returned by COLUMN_SIZE and DECIMAL_DIGITS should be interpreted.
// 10 means they represent the number of decimal digits allowed for the column.
// 2 means they represent the number of bits allowed for the column.
// null means radix is not applicable for the given type.
type.isInteger ? Integer.valueOf(10) : type.isRational ? Integer.valueOf(2) : null, type.isInteger ? Integer.valueOf(10) : type.isRational ? Integer.valueOf(2) : null,
// everything is nullable // everything is nullable
DatabaseMetaData.columnNullable, DatabaseMetaData.columnNullable,

View File

@ -81,7 +81,7 @@ public class SysTypes extends Command {
// everything is searchable, // everything is searchable,
DatabaseMetaData.typeSearchable, DatabaseMetaData.typeSearchable,
// only numerics are signed // only numerics are signed
t.isNumeric() ? !t.isSigned : null, t.isSigned(),
//no fixed precision scale SQL_FALSE //no fixed precision scale SQL_FALSE
0, 0,
null, null,

View File

@ -13,23 +13,23 @@ import java.util.Locale;
*/ */
public enum DataType { public enum DataType {
// @formatter:off // @formatter:off
// jdbc type, size, defPrecision, dispSize, sig, int, rat, docvals // jdbc type, size, defPrecision, dispSize, int, rat, docvals
NULL( JDBCType.NULL, 0, 0, 0), NULL( JDBCType.NULL, 0, 0, 0),
UNSUPPORTED( JDBCType.OTHER, 0, 0, 0), UNSUPPORTED( JDBCType.OTHER, 0, 0, 0),
BOOLEAN( JDBCType.BOOLEAN, 1, 1, 1), BOOLEAN( JDBCType.BOOLEAN, 1, 1, 1),
BYTE( JDBCType.TINYINT, Byte.BYTES, 3, 5, true, true, false, true), BYTE( JDBCType.TINYINT, Byte.BYTES, 3, 5, true, false, true),
SHORT( JDBCType.SMALLINT, Short.BYTES, 5, 6, true, true, false, true), SHORT( JDBCType.SMALLINT, Short.BYTES, 5, 6, true, false, true),
INTEGER( JDBCType.INTEGER, Integer.BYTES, 10, 11, true, true, false, true), INTEGER( JDBCType.INTEGER, Integer.BYTES, 10, 11, true, false, true),
LONG( JDBCType.BIGINT, Long.BYTES, 19, 20, true, true, false, true), LONG( JDBCType.BIGINT, Long.BYTES, 19, 20, true, false, true),
// 53 bits defaultPrecision ~ 16(15.95) decimal digits (53log10(2)), // 53 bits defaultPrecision ~ 16(15.95) decimal digits (53log10(2)),
DOUBLE( JDBCType.DOUBLE, Double.BYTES, 16, 25, true, false, true, true), DOUBLE( JDBCType.DOUBLE, Double.BYTES, 16, 25, false, true, true),
// 24 bits defaultPrecision - 24*log10(2) =~ 7 (7.22) // 24 bits defaultPrecision - 24*log10(2) =~ 7 (7.22)
FLOAT( JDBCType.REAL, Float.BYTES, 7, 15, true, false, true, true), FLOAT( JDBCType.REAL, Float.BYTES, 7, 15, false, true, true),
HALF_FLOAT( JDBCType.FLOAT, Double.BYTES, 16, 25, true, false, true, true), HALF_FLOAT( JDBCType.FLOAT, Double.BYTES, 16, 25, false, true, true),
// precision is based on long // precision is based on long
SCALED_FLOAT(JDBCType.FLOAT, Double.BYTES, 19, 25, true, false, true, true), SCALED_FLOAT(JDBCType.FLOAT, Double.BYTES, 19, 25, false, true, true),
KEYWORD( JDBCType.VARCHAR, Integer.MAX_VALUE, 256, 0), KEYWORD( JDBCType.VARCHAR, Integer.MAX_VALUE, 256, 0),
TEXT( JDBCType.VARCHAR, Integer.MAX_VALUE, Integer.MAX_VALUE, 0, false, false, false, false), TEXT( JDBCType.VARCHAR, Integer.MAX_VALUE, Integer.MAX_VALUE, 0, false, false, false),
OBJECT( JDBCType.STRUCT, -1, 0, 0), OBJECT( JDBCType.STRUCT, -1, 0, 0),
NESTED( JDBCType.STRUCT, -1, 0, 0), NESTED( JDBCType.STRUCT, -1, 0, 0),
BINARY( JDBCType.VARBINARY, -1, Integer.MAX_VALUE, 0), BINARY( JDBCType.VARBINARY, -1, Integer.MAX_VALUE, 0),
@ -71,11 +71,6 @@ public enum DataType {
*/ */
public final int displaySize; public final int displaySize;
/**
* True if the type represents a signed number
*/
public final boolean isSigned;
/** /**
* True if the type represents an integer number * True if the type represents an integer number
*/ */
@ -91,21 +86,20 @@ public enum DataType {
*/ */
public final boolean defaultDocValues; public final boolean defaultDocValues;
DataType(JDBCType jdbcType, int size, int defaultPrecision, int displaySize, boolean isSigned, boolean isInteger, boolean isRational, DataType(JDBCType jdbcType, int size, int defaultPrecision, int displaySize, boolean isInteger, boolean isRational,
boolean defaultDocValues) { boolean defaultDocValues) {
this.esType = name().toLowerCase(Locale.ROOT); this.esType = name().toLowerCase(Locale.ROOT);
this.jdbcType = jdbcType; this.jdbcType = jdbcType;
this.size = size; this.size = size;
this.defaultPrecision = defaultPrecision; this.defaultPrecision = defaultPrecision;
this.displaySize = displaySize; this.displaySize = displaySize;
this.isSigned = isSigned;
this.isInteger = isInteger; this.isInteger = isInteger;
this.isRational = isRational; this.isRational = isRational;
this.defaultDocValues = defaultDocValues; this.defaultDocValues = defaultDocValues;
} }
DataType(JDBCType jdbcType, int size, int defaultPrecision, int displaySize) { DataType(JDBCType jdbcType, int size, int defaultPrecision, int displaySize) {
this(jdbcType, size, defaultPrecision, displaySize, false, false, false, true); this(jdbcType, size, defaultPrecision, displaySize, false, false, true);
} }
public String sqlName() { public String sqlName() {
@ -116,6 +110,14 @@ public enum DataType {
return isInteger || isRational; return isInteger || isRational;
} }
/**
* Returns true if value is signed, false if it is unsigned or null if the type doesn't represent a number
*/
public Boolean isSigned() {
// For now all numeric values that es supports are signed
return isNumeric() ? true : null;
}
public boolean isString() { public boolean isString() {
return this == KEYWORD || this == TEXT; return this == KEYWORD || this == TEXT;
} }

View File

@ -16,7 +16,8 @@ import java.util.function.Function;
import java.util.function.LongFunction; import java.util.function.LongFunction;
/** /**
* Conversions from one data type to another. * Conversions from one Elasticsearch data type to another Elasticsearch data types.
* <p>
* This class throws {@link SqlIllegalArgumentException} to differentiate between validation * This class throws {@link SqlIllegalArgumentException} to differentiate between validation
* errors inside SQL as oppose to the rest of ES. * errors inside SQL as oppose to the rest of ES.
*/ */
@ -24,6 +25,13 @@ public abstract class DataTypeConversion {
private static final DateTimeFormatter UTC_DATE_FORMATTER = ISODateTimeFormat.dateTimeNoMillis().withZoneUTC(); private static final DateTimeFormatter UTC_DATE_FORMATTER = ISODateTimeFormat.dateTimeNoMillis().withZoneUTC();
/**
* Returns the type compatible with both left and right types
* <p>
* If one of the types is null - returns another type
* If both types are numeric - returns type with the highest precision int < long < float < double
* If one of the types is string and another numeric - returns numeric
*/
public static DataType commonType(DataType left, DataType right) { public static DataType commonType(DataType left, DataType right) {
if (left == right) { if (left == right) {
return left; return left;
@ -65,46 +73,30 @@ public abstract class DataTypeConversion {
return null; return null;
} }
public static boolean canConvert(DataType from, DataType to) { // TODO it'd be cleaner and more right to fetch the conversion /**
* Returns true if the from type can be converted to the to type, false - otherwise
*/
public static boolean canConvert(DataType from, DataType to) {
// Special handling for nulls and if conversion is not requires
if (from == to || from == DataType.NULL) {
return true;
}
// only primitives are supported so far // only primitives are supported so far
if (!from.isPrimitive() || !to.isPrimitive()) { return from.isPrimitive() && to.isPrimitive() && conversion(from, to) != null;
return false;
}
if (from.getClass() == to.getClass()) {
return true;
}
if (from == DataType.NULL) {
return true;
}
// anything can be converted to String
if (to.isString()) {
return true;
}
// also anything can be converted into a bool
if (to == DataType.BOOLEAN) {
return true;
}
// numeric conversion
if ((from.isString() || from == DataType.BOOLEAN || from == DataType.DATE || from.isNumeric()) && to.isNumeric()) {
return true;
}
// date conversion
if ((from == DataType.DATE || from.isString() || from.isNumeric()) && to == DataType.DATE) {
return true;
}
return false;
} }
/** /**
* Get the conversion from one type to another. * Get the conversion from one type to another.
*/ */
public static Conversion conversionFor(DataType from, DataType to) { public static Conversion conversionFor(DataType from, DataType to) {
Conversion conversion = conversion(from, to);
if (conversion == null) {
throw new SqlIllegalArgumentException("cannot convert from [" + from + "] to [" + to + "]");
}
return conversion;
}
private static Conversion conversion(DataType from, DataType to) {
switch (to) { switch (to) {
case KEYWORD: case KEYWORD:
case TEXT: case TEXT:
@ -126,8 +118,9 @@ public abstract class DataTypeConversion {
case BOOLEAN: case BOOLEAN:
return conversionToBoolean(from); return conversionToBoolean(from);
default: default:
throw new SqlIllegalArgumentException("cannot convert from [" + from + "] to [" + to + "]"); return null;
} }
} }
private static Conversion conversionToString(DataType from) { private static Conversion conversionToString(DataType from) {
@ -150,7 +143,7 @@ public abstract class DataTypeConversion {
if (from.isString()) { if (from.isString()) {
return Conversion.STRING_TO_LONG; return Conversion.STRING_TO_LONG;
} }
throw new SqlIllegalArgumentException("cannot convert from [" + from + "] to [Long]"); return null;
} }
private static Conversion conversionToInt(DataType from) { private static Conversion conversionToInt(DataType from) {
@ -166,7 +159,7 @@ public abstract class DataTypeConversion {
if (from.isString()) { if (from.isString()) {
return Conversion.STRING_TO_INT; return Conversion.STRING_TO_INT;
} }
throw new SqlIllegalArgumentException("cannot convert from [" + from + "] to [Integer]"); return null;
} }
private static Conversion conversionToShort(DataType from) { private static Conversion conversionToShort(DataType from) {
@ -182,7 +175,7 @@ public abstract class DataTypeConversion {
if (from.isString()) { if (from.isString()) {
return Conversion.STRING_TO_SHORT; return Conversion.STRING_TO_SHORT;
} }
throw new SqlIllegalArgumentException("cannot convert [" + from + "] to [Short]"); return null;
} }
private static Conversion conversionToByte(DataType from) { private static Conversion conversionToByte(DataType from) {
@ -198,7 +191,7 @@ public abstract class DataTypeConversion {
if (from.isString()) { if (from.isString()) {
return Conversion.STRING_TO_BYTE; return Conversion.STRING_TO_BYTE;
} }
throw new SqlIllegalArgumentException("cannot convert [" + from + "] to [Byte]"); return null;
} }
private static Conversion conversionToFloat(DataType from) { private static Conversion conversionToFloat(DataType from) {
@ -214,7 +207,7 @@ public abstract class DataTypeConversion {
if (from.isString()) { if (from.isString()) {
return Conversion.STRING_TO_FLOAT; return Conversion.STRING_TO_FLOAT;
} }
throw new SqlIllegalArgumentException("cannot convert [" + from + "] to [Float]"); return null;
} }
private static Conversion conversionToDouble(DataType from) { private static Conversion conversionToDouble(DataType from) {
@ -230,7 +223,7 @@ public abstract class DataTypeConversion {
if (from.isString()) { if (from.isString()) {
return Conversion.STRING_TO_DOUBLE; return Conversion.STRING_TO_DOUBLE;
} }
throw new SqlIllegalArgumentException("cannot convert [" + from + "] to [Double]"); return null;
} }
private static Conversion conversionToDate(DataType from) { private static Conversion conversionToDate(DataType from) {
@ -246,7 +239,7 @@ public abstract class DataTypeConversion {
if (from.isString()) { if (from.isString()) {
return Conversion.STRING_TO_DATE; return Conversion.STRING_TO_DATE;
} }
throw new SqlIllegalArgumentException("cannot convert [" + from + "] to [Date]"); return null;
} }
private static Conversion conversionToBoolean(DataType from) { private static Conversion conversionToBoolean(DataType from) {
@ -256,7 +249,7 @@ public abstract class DataTypeConversion {
if (from.isString()) { if (from.isString()) {
return Conversion.STRING_TO_BOOLEAN; return Conversion.STRING_TO_BOOLEAN;
} }
throw new SqlIllegalArgumentException("cannot convert [" + from + "] to [Boolean]"); return null;
} }
public static byte safeToByte(long x) { public static byte safeToByte(long x) {
@ -295,9 +288,14 @@ public abstract class DataTypeConversion {
return Booleans.parseBoolean(lowVal); return Booleans.parseBoolean(lowVal);
} }
/**
* Converts arbitrary object to the desired data type.
* <p>
* Throws SqlIllegalArgumentException if such conversion is not possible
*/
public static Object convert(Object value, DataType dataType) { public static Object convert(Object value, DataType dataType) {
DataType detectedType = DataTypes.fromJava(value); DataType detectedType = DataTypes.fromJava(value);
if (detectedType.equals(dataType) || value == null) { if (detectedType == dataType || value == null) {
return value; return value;
} }
return conversionFor(detectedType, dataType).convert(value); return conversionFor(detectedType, dataType).convert(value);

View File

@ -177,4 +177,18 @@ public class DataTypeConversionTests extends ESTestCase {
assertEquals("[" + Short.MAX_VALUE + "] out of [Byte] range", e.getMessage()); assertEquals("[" + Short.MAX_VALUE + "] out of [Byte] range", e.getMessage());
} }
} }
public void testCommonType() {
assertEquals(DataType.BOOLEAN, DataTypeConversion.commonType(DataType.BOOLEAN, DataType.NULL));
assertEquals(DataType.BOOLEAN, DataTypeConversion.commonType(DataType.NULL, DataType.BOOLEAN));
assertEquals(DataType.BOOLEAN, DataTypeConversion.commonType(DataType.BOOLEAN, DataType.BOOLEAN));
assertEquals(DataType.NULL, DataTypeConversion.commonType(DataType.NULL, DataType.NULL));
assertEquals(DataType.INTEGER, DataTypeConversion.commonType(DataType.INTEGER, DataType.KEYWORD));
assertEquals(DataType.LONG, DataTypeConversion.commonType(DataType.TEXT, DataType.LONG));
assertEquals(null, DataTypeConversion.commonType(DataType.TEXT, DataType.KEYWORD));
assertEquals(DataType.SHORT, DataTypeConversion.commonType(DataType.SHORT, DataType.BYTE));
assertEquals(DataType.FLOAT, DataTypeConversion.commonType(DataType.BYTE, DataType.FLOAT));
assertEquals(DataType.FLOAT, DataTypeConversion.commonType(DataType.FLOAT, DataType.INTEGER));
assertEquals(DataType.DOUBLE, DataTypeConversion.commonType(DataType.DOUBLE, DataType.FLOAT));
}
} }