Skip to content

Commit 14a7bbd

Browse files
alex268alex268
committed
Added checks of invalid values of Decimal
1 parent 96ccf40 commit 14a7bbd

File tree

4 files changed

+94
-35
lines changed

4 files changed

+94
-35
lines changed

‎jdbc/src/main/java/tech/ydb/jdbc/YdbConst.java‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,7 @@ public final class YdbConst {
103103
public static final String UNABLE_TO_CONVERT = "Cannot cast [%s] with value [%s] to [%s]";
104104
public static final String UNABLE_TO_CONVERT_AS_URL = "Cannot cast as URL: ";
105105
public static final String UNABLE_TO_CAST_TO_CLASS = "Cannot cast [%s] to class [%s]";
106+
public static final String UNABLE_TO_CAST_TO_DECIMAL = "Cannot cast to decimal type %s: [%s] is %s";
106107

107108
public static final String MISSING_VALUE_FOR_PARAMETER = "Missing value for parameter: ";
108109
public static final String MISSING_REQUIRED_VALUE = "Missing required value for parameter: ";

‎jdbc/src/main/java/tech/ydb/jdbc/common/MappingGetters.java‎

Lines changed: 1 addition & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -30,32 +30,6 @@
3030
import tech.ydb.table.values.Type;
3131
import tech.ydb.table.values.Value;
3232

33-
import static tech.ydb.table.values.PrimitiveType.Bool;
34-
import static tech.ydb.table.values.PrimitiveType.Bytes;
35-
import static tech.ydb.table.values.PrimitiveType.Date;
36-
import static tech.ydb.table.values.PrimitiveType.Datetime;
37-
import static tech.ydb.table.values.PrimitiveType.Double;
38-
import static tech.ydb.table.values.PrimitiveType.Float;
39-
import static tech.ydb.table.values.PrimitiveType.Int16;
40-
import static tech.ydb.table.values.PrimitiveType.Int32;
41-
import static tech.ydb.table.values.PrimitiveType.Int64;
42-
import static tech.ydb.table.values.PrimitiveType.Int8;
43-
import static tech.ydb.table.values.PrimitiveType.Interval;
44-
import static tech.ydb.table.values.PrimitiveType.Json;
45-
import static tech.ydb.table.values.PrimitiveType.JsonDocument;
46-
import static tech.ydb.table.values.PrimitiveType.Text;
47-
import static tech.ydb.table.values.PrimitiveType.Timestamp;
48-
import static tech.ydb.table.values.PrimitiveType.TzDate;
49-
import static tech.ydb.table.values.PrimitiveType.TzDatetime;
50-
import static tech.ydb.table.values.PrimitiveType.TzTimestamp;
51-
import static tech.ydb.table.values.PrimitiveType.Uint16;
52-
import static tech.ydb.table.values.PrimitiveType.Uint32;
53-
import static tech.ydb.table.values.PrimitiveType.Uint64;
54-
import static tech.ydb.table.values.PrimitiveType.Uint8;
55-
import static tech.ydb.table.values.PrimitiveType.Uuid;
56-
import static tech.ydb.table.values.PrimitiveType.Yson;
57-
import static tech.ydb.table.values.Type.Kind.PRIMITIVE;
58-
5933
public class MappingGetters {
6034
private MappingGetters() { }
6135

@@ -95,7 +69,7 @@ static Getters buildGetters(Type type) {
9569
value -> value.getDecimal().toBigDecimal().floatValue(),
9670
value -> value.getDecimal().toBigDecimal().doubleValue(),
9771
castToBytesNotSupported(clazz),
98-
PrimitiveReader::getDecimal,
72+
value -> value.getDecimal().toBigDecimal(),
9973
castToClassNotSupported(clazz),
10074
castToInstantNotSupported(clazz),
10175
castToNStringNotSupported(clazz),

‎jdbc/src/main/java/tech/ydb/jdbc/common/MappingSetters.java‎

Lines changed: 21 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -466,23 +466,36 @@ private static PrimitiveValue castToTimestamp(PrimitiveType type, Object x) thro
466466
throw castNotSupported(type, x);
467467
}
468468

469+
private static DecimalValue validateValue(DecimalType type, DecimalValue value, Object x) throws SQLException {
470+
if (value.isInf()) {
471+
throw new SQLException(String.format(YdbConst.UNABLE_TO_CAST_TO_DECIMAL, type, toString(x), "Infinite"));
472+
}
473+
if (value.isNegativeInf()) {
474+
throw new SQLException(String.format(YdbConst.UNABLE_TO_CAST_TO_DECIMAL, type, toString(x), "-Infinite"));
475+
}
476+
if (value.isNan()) {
477+
throw new SQLException(String.format(YdbConst.UNABLE_TO_CAST_TO_DECIMAL, type, toString(x), "NaN"));
478+
}
479+
return value;
480+
}
481+
469482
private static DecimalValue castToDecimalValue(DecimalType type, Object x) throws SQLException {
470483
if (x instanceof DecimalValue) {
471-
return (DecimalValue) x;
484+
return validateValue(type, (DecimalValue) x, x);
472485
} else if (x instanceof BigDecimal) {
473-
return type.newValue((BigDecimal) x);
486+
return validateValue(type, type.newValue((BigDecimal) x), x);
474487
} else if (x instanceof BigInteger) {
475-
return type.newValue((BigInteger) x);
488+
return validateValue(type, type.newValue((BigInteger) x), x);
476489
} else if (x instanceof Long) {
477-
return type.newValue((Long) x);
490+
return validateValue(type, type.newValue((Long) x), x);
478491
} else if (x instanceof Integer) {
479-
return type.newValue((Integer) x);
492+
return validateValue(type, type.newValue((Integer) x), x);
480493
} else if (x instanceof Short) {
481-
return type.newValue((Short) x);
494+
return validateValue(type, type.newValue((Short) x), x);
482495
} else if (x instanceof Byte) {
483-
return type.newValue((Byte) x);
496+
return validateValue(type, type.newValue((Byte) x), x);
484497
} else if (x instanceof String) {
485-
return type.newValue((String) x);
498+
return validateValue(type, type.newValue((String) x), x);
486499
}
487500
throw castNotSupported(type.getKind(), x);
488501
}

‎jdbc/src/test/java/tech/ydb/jdbc/impl/YdbPreparedStatementTest.java‎

Lines changed: 71 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package tech.ydb.jdbc.impl;
22

33
import java.math.BigDecimal;
4+
import java.math.BigInteger;
45
import java.sql.Date;
56
import java.sql.PreparedStatement;
67
import java.sql.ResultSet;
@@ -982,4 +983,74 @@ private void assertNextDate(ResultSet rs, int key, LocalDate ld) throws SQLExcep
982983
Assertions.assertEquals(ld.atStartOfDay(), rs.getObject("c_Date", LocalDateTime.class));
983984
Assertions.assertEquals(ld.atStartOfDay(ZoneId.systemDefault()).toInstant(), rs.getObject("c_Date", Instant.class));
984985
}
986+
987+
@ParameterizedTest(name = "with {0}")
988+
@EnumSource(SqlQueries.JdbcQuery.class)
989+
public void decimalTest(SqlQueries.JdbcQuery query) throws SQLException {
990+
String upsert = TEST_TABLE.upsertOne(query, "c_Decimal", "Decimal(22, 9)");
991+
992+
// YDB partially ignores Decimal(22, 9) limit, but have hard limit to 35 digits
993+
String maxValue = "9999999999" + "9999999999" + "9999999999" + "99999";
994+
BigDecimal closeToInf = new BigDecimal(new BigInteger(maxValue), 9);
995+
BigDecimal closeToNegInf = new BigDecimal(new BigInteger(maxValue).negate(), 9);
996+
try (PreparedStatement ps = jdbc.connection().prepareStatement(upsert)) {
997+
ps.setInt(1, 1);
998+
ps.setBigDecimal(2, BigDecimal.valueOf(1.5d));
999+
ps.execute();
1000+
1001+
ps.setInt(1, 2);
1002+
ps.setBigDecimal(2, BigDecimal.valueOf(-12345, 10)); // will be rounded to -0.000001234
1003+
ps.execute();
1004+
1005+
ps.setInt(1, 3);
1006+
ps.setBigDecimal(2, closeToInf);
1007+
ps.execute();
1008+
1009+
ps.setInt(1, 4);
1010+
ps.setBigDecimal(2, closeToNegInf);
1011+
ps.execute();
1012+
1013+
ps.setInt(1, 5);
1014+
ExceptionAssert.sqlException(""
1015+
+ "Cannot cast to decimal type Decimal(22, 9): "
1016+
+ "[class java.math.BigDecimal: 100000000000000000000000000.000000000] is Infinite",
1017+
() -> ps.setBigDecimal(2, closeToInf.add(BigDecimal.valueOf(1, 9)))
1018+
);
1019+
1020+
ExceptionAssert.sqlException(""
1021+
+ "Cannot cast to decimal type Decimal(22, 9): "
1022+
+ "[class java.math.BigDecimal: -100000000000000000000000000.000000000] is -Infinite",
1023+
() -> ps.setBigDecimal(2, closeToNegInf.subtract(BigDecimal.valueOf(1, 9)))
1024+
);
1025+
1026+
ExceptionAssert.sqlException(""
1027+
+ "Cannot cast to decimal type Decimal(22, 9): "
1028+
+ "[class java.math.BigDecimal: 100000000000000000000000000.000000001] is NaN",
1029+
() -> ps.setBigDecimal(2, closeToInf.add(BigDecimal.valueOf(2, 9)))
1030+
);
1031+
}
1032+
1033+
try (Statement st = jdbc.connection().createStatement()) {
1034+
try (ResultSet rs = st.executeQuery(TEST_TABLE.selectColumn("c_Decimal"))) {
1035+
assertNextDecimal(rs, 1, BigDecimal.valueOf(1.5d).setScale(9));
1036+
assertNextDecimal(rs, 2, BigDecimal.valueOf(-1234, 9));
1037+
assertNextDecimal(rs, 3, closeToInf);
1038+
assertNextDecimal(rs, 4, closeToNegInf);
1039+
1040+
Assertions.assertFalse(rs.next());
1041+
}
1042+
}
1043+
};
1044+
1045+
private void assertNextDecimal(ResultSet rs, int key, BigDecimal bg) throws SQLException {
1046+
Assertions.assertTrue(rs.next());
1047+
Assertions.assertEquals(key, rs.getInt("key"));
1048+
1049+
Object obj = rs.getObject("c_Decimal");
1050+
Assertions.assertTrue(obj instanceof BigDecimal);
1051+
Assertions.assertEquals(bg, obj);
1052+
1053+
BigDecimal decimal = rs.getBigDecimal("c_Decimal");
1054+
Assertions.assertEquals(bg, decimal);
1055+
}
9851056
}

0 commit comments

Comments
 (0)