public RelDataType inferReturnType(SqlOperatorBinding opBinding) {
RelDataType type1 = opBinding.getOperandType(0);
RelDataType type2 = opBinding.getOperandType(1);
if (SqlTypeUtil.isExactNumeric(type1) && SqlTypeUtil.isExactNumeric(type2)) {
if (SqlTypeUtil.isDecimal(type1) || SqlTypeUtil.isDecimal(type2)) {
int p1 = type1.getPrecision();
int p2 = type2.getPrecision();
int s1 = type1.getScale();
int s2 = type2.getScale();
int scale = Math.max(s1, s2);
assert scale <= SqlTypeName.MAX_NUMERIC_SCALE;
int precision = Math.max(p1 - s1, p2 - s2) + scale + 1;
precision = Math.min(precision, SqlTypeName.MAX_NUMERIC_PRECISION);
assert precision > 0;
RelDataType ret;
ret = opBinding.getTypeFactory().createSqlType(SqlTypeName.DECIMAL, precision, scale);
return ret;
}
}
return null;
}
/**
* Determines whether a {@link RexCall} requires decimal expansion. It usually requires expansion
* if it has decimal operands.
*
* <p>Exceptions to this rule are:
*
* <ul>
* <li>isNull doesn't require expansion
* <li>It's okay to cast decimals to and from char types
* <li>It's okay to cast nulls as decimals
* <li>Casts require expansion if their return type is decimal
* <li>Reinterpret casts can handle a decimal operand
* </ul>
*
* @param expr expression possibly in need of expansion
* @param recurse whether to check nested calls
* @return whether the expression requires expansion
*/
public static boolean requiresDecimalExpansion(RexNode expr, boolean recurse) {
if (!(expr instanceof RexCall)) {
return false;
}
RexCall call = (RexCall) expr;
boolean localCheck = true;
switch (call.getKind()) {
case Reinterpret:
case IsNull:
localCheck = false;
break;
case Cast:
RelDataType lhsType = call.getType();
RelDataType rhsType = call.operands[0].getType();
if (rhsType.getSqlTypeName() == SqlTypeName.NULL) {
return false;
}
if (SqlTypeUtil.inCharFamily(lhsType) || SqlTypeUtil.inCharFamily(rhsType)) {
localCheck = false;
} else if (SqlTypeUtil.isDecimal(lhsType) && (lhsType != rhsType)) {
return true;
}
break;
default:
localCheck = call.getOperator().requiresDecimalExpansion();
}
if (localCheck) {
if (SqlTypeUtil.isDecimal(call.getType())) {
// NOTE jvs 27-Mar-2007: Depending on the type factory, the
// result of a division may be decimal, even though both inputs
// are integer.
return true;
}
for (int i = 0; i < call.operands.length; i++) {
if (SqlTypeUtil.isDecimal(call.operands[i].getType())) {
return true;
}
}
}
return (recurse && requiresDecimalExpansion(call.operands, recurse));
}
public RelDataType inferReturnType(SqlOperatorBinding opBinding) {
RelDataType type1 = opBinding.getOperandType(0);
if (SqlTypeUtil.isDecimal(type1)) {
if (type1.getScale() == 0) {
return type1;
} else {
int p = type1.getPrecision();
RelDataType ret;
ret = opBinding.getTypeFactory().createSqlType(SqlTypeName.DECIMAL, p, 0);
if (type1.isNullable()) {
ret = opBinding.getTypeFactory().createTypeWithNullability(ret, true);
}
return ret;
}
}
return null;
}
private Expression castToAssignableValueImpl() {
if (requiresSpecializedCast()) {
if (rhsType.isNullable() && (!SqlTypeUtil.isDecimal(rhsType))) {
rhsExp = getValue(rhsType, rhsExp);
}
addStatement(
new ExpressionStatement(
new MethodCall(
lhsExp,
"cast",
new ExpressionList(rhsExp, Literal.makeLiteral(lhsType.getPrecision())))));
} else {
// Set current_date for casting time to timestamp. If
// rhsType is null then we may have to be ready for anything.
// But it will be null even for current_timestamp, so the
// condition below seems a bit excessive.
if ((lhsType.getSqlTypeName() == SqlTypeName.TIMESTAMP)
&& ((rhsType == null) || (rhsType.getSqlTypeName() == SqlTypeName.TIME))) {
addStatement(
new ExpressionStatement(
new MethodCall(lhsExp, "setCurrentDate", new ExpressionList(getCurrentDate()))));
}
addStatement(
new ExpressionStatement(
new MethodCall(
lhsExp, AssignableValue.ASSIGNMENT_METHOD_NAME, new ExpressionList(rhsExp))));
}
// Trim precision of datetime values.
//
if (((lhsType.getSqlTypeName() == SqlTypeName.TIMESTAMP)
|| (lhsType.getSqlTypeName() == SqlTypeName.TIME))) {
if ((rhsType != null)
// FIXME: JavaType(java.sql.Time) and
// JavaType(java.sql.Timestamp) say they support precision
// but do not.
&& !rhsType.toString().startsWith("JavaType(")
&& rhsType.getSqlTypeName().allowsPrec()
&& (lhsType.getPrecision() < rhsType.getPrecision())) {
int lhsPrecision = lhsType.getPrecision();
if (lhsPrecision == -1) {
lhsPrecision = 0;
}
addStatement(
new ExpressionStatement(
new MethodCall(
lhsExp,
SqlDateTimeWithoutTZ.ADJUST_PRECISION_METHOD_NAME,
new ExpressionList(Literal.makeLiteral(lhsPrecision)))));
}
}
boolean mayNeedPadOrTruncate = false;
if (SqlTypeUtil.inCharOrBinaryFamilies(lhsType) && !SqlTypeUtil.isLob(lhsType)) {
mayNeedPadOrTruncate = true;
}
if (mayNeedPadOrTruncate) {
// check overflow if it is datetime.
// TODO: should check it at the run time.
// so, it should be in the
// cast(SqlDateTimeWithTZ, int precision);
if ((rhsType != null) && (rhsType.getSqlTypeName() != null)) {
SqlTypeName typeName = rhsType.getSqlTypeName();
int precision = 0;
switch (typeName) {
case DATE:
precision = 10;
break;
case TIME:
precision = 8;
break;
case TIMESTAMP:
precision = 19;
break;
}
if ((precision != 0) && (precision > lhsType.getPrecision())) {
addStatement(
new IfStatement(
new BinaryExpression(
Literal.makeLiteral(precision),
BinaryExpression.GREATER,
Literal.makeLiteral(lhsType.getPrecision())),
getThrowStmtList()));
}
}
if ((rhsType != null)
&& (rhsType.getFamily() == lhsType.getFamily())
&& !SqlTypeUtil.isLob(rhsType)) {
// we may be able to skip pad/truncate based on
// known facts about source and target precisions
if (SqlTypeUtil.isBoundedVariableWidth(lhsType)) {
if (lhsType.getPrecision() >= rhsType.getPrecision()) {
// target precision is greater than source
// precision, so truncation is impossible
// and we can skip adjustment
return lhsExp;
}
} else {
if ((lhsType.getPrecision() == rhsType.getPrecision())
&& !SqlTypeUtil.isBoundedVariableWidth(rhsType)) {
// source and target are both fixed-width, and
// precisions are the same, so there's no adjustment
// needed
return lhsExp;
}
}
}
// determine target precision
Expression precisionExp = Literal.makeLiteral(lhsType.getPrecision());
// need to pad only for fixed width
Expression needPadExp = Literal.makeLiteral(!SqlTypeUtil.isBoundedVariableWidth(lhsType));
// pad character is 0 for binary, space for character
Expression padByteExp;
if (!SqlTypeUtil.inCharFamily(lhsType)) {
padByteExp = new CastExpression(OJSystem.BYTE, Literal.makeLiteral(0));
} else {
padByteExp = new CastExpression(OJSystem.BYTE, Literal.makeLiteral(' '));
}
// generate the call to do the job
addStatement(
new ExpressionStatement(
new MethodCall(
lhsExp,
BytePointer.ENFORCE_PRECISION_METHOD_NAME,
new ExpressionList(precisionExp, needPadExp, padByteExp))));
}
return lhsExp;
}