use of com.hazelcast.sql.impl.expression.datetime.ExtractField in project hazelcast by hazelcast.
the class RexToExpression method convertCall.
/**
* Converts a {@link RexCall} to {@link Expression}.
*
* @param call the call to convert.
* @return the resulting expression.
* @throws QueryException if the given {@link RexCall} can't be
* converted.
*/
@SuppressWarnings({ "checkstyle:CyclomaticComplexity", "checkstyle:MethodLength", "checkstyle:ReturnCount", "checkstyle:NPathComplexity", "checkstyle:MagicNumber" })
public static Expression<?> convertCall(RexCall call, Expression<?>[] operands) {
SqlOperator operator = call.getOperator();
QueryDataType resultType = HazelcastTypeUtils.toHazelcastType(call.getType());
switch(operator.getKind()) {
case DEFAULT:
return ConstantExpression.create(null, resultType);
case CAST:
if (operands[0].getType().equals(resultType)) {
// nullable ones after the conversion.
return operands[0];
}
return CastExpression.create(operands[0], resultType);
case AND:
return AndPredicate.create(operands);
case OR:
return OrPredicate.create(operands);
case NOT:
return NotPredicate.create(operands[0]);
case PLUS:
return PlusFunction.create(operands[0], operands[1], resultType);
case MINUS:
return MinusFunction.create(operands[0], operands[1], resultType);
case TIMES:
return MultiplyFunction.create(operands[0], operands[1], resultType);
case DIVIDE:
return DivideFunction.create(operands[0], operands[1], resultType);
case MOD:
return RemainderFunction.create(operands[0], operands[1], resultType);
case MINUS_PREFIX:
return UnaryMinusFunction.create(operands[0], resultType);
case PLUS_PREFIX:
return operands[0];
case FLOOR:
return FloorCeilFunction.create(operands[0], resultType, false);
case CEIL:
return FloorCeilFunction.create(operands[0], resultType, true);
case EQUALS:
return ComparisonPredicate.create(operands[0], operands[1], ComparisonMode.EQUALS);
case NOT_EQUALS:
return ComparisonPredicate.create(operands[0], operands[1], ComparisonMode.NOT_EQUALS);
case GREATER_THAN:
return ComparisonPredicate.create(operands[0], operands[1], ComparisonMode.GREATER_THAN);
case GREATER_THAN_OR_EQUAL:
return ComparisonPredicate.create(operands[0], operands[1], ComparisonMode.GREATER_THAN_OR_EQUAL);
case LESS_THAN:
return ComparisonPredicate.create(operands[0], operands[1], ComparisonMode.LESS_THAN);
case LESS_THAN_OR_EQUAL:
return ComparisonPredicate.create(operands[0], operands[1], ComparisonMode.LESS_THAN_OR_EQUAL);
case SEARCH:
return SearchPredicate.create(operands[0], operands[1]);
case IS_TRUE:
return IsTruePredicate.create(operands[0]);
case IS_NOT_TRUE:
return IsNotTruePredicate.create(operands[0]);
case IS_FALSE:
return IsFalsePredicate.create(operands[0]);
case IS_NOT_FALSE:
return IsNotFalsePredicate.create(operands[0]);
case IS_NULL:
return IsNullPredicate.create(operands[0]);
case IS_NOT_NULL:
return IsNotNullPredicate.create(operands[0]);
case LIKE:
boolean negated = ((HazelcastLikeOperator) operator).isNegated();
Expression<?> escape1 = operands.length == 2 ? null : operands[2];
return LikeFunction.create(operands[0], operands[1], escape1, negated);
case TRIM:
assert operands.length == 3;
assert operands[0] instanceof SymbolExpression;
SqlTrimFunction.Flag trimFlag = ((SymbolExpression) operands[0]).getSymbol();
return TrimFunction.create(operands[2], operands[1], trimFlag.getLeft() == 1, trimFlag.getRight() == 1);
case EXTRACT:
assert operands.length == 2;
assert operands[0] instanceof SymbolExpression;
TimeUnitRange field = ((SymbolExpression) operands[0]).getSymbol();
ExtractField extractField = convertField(field);
return ExtractFunction.create(operands[1], extractField);
case CASE:
return CaseExpression.create(operands);
case COALESCE:
return CaseExpression.coalesce(operands);
case NULLIF:
return CaseExpression.nullif(operands[0], operands[1]);
case OTHER:
if (operator == HazelcastSqlOperatorTable.CONCAT) {
assert operands.length == 2;
return ConcatFunction.create(operands[0], operands[1]);
}
if (operator == HazelcastSqlOperatorTable.NOT_LIKE) {
assert ((HazelcastLikeOperator) operator).isNegated();
Expression<?> escape2 = operands.length == 2 ? null : operands[2];
return LikeFunction.create(operands[0], operands[1], escape2, true);
}
break;
case POSITION:
case OTHER_FUNCTION:
SqlFunction function = (SqlFunction) operator;
// Math.
if (function == HazelcastSqlOperatorTable.POWER) {
assert operands.length == 2;
return DoubleBiFunction.create(operands[0], operands[1], DoubleBiFunction.POWER);
} else if (function == HazelcastSqlOperatorTable.SQUARE) {
return DoubleFunction.create(operands[0], DoubleFunction.SQUARE);
} else if (function == HazelcastSqlOperatorTable.SQRT) {
return DoubleFunction.create(operands[0], DoubleFunction.SQRT);
} else if (function == HazelcastSqlOperatorTable.CBRT) {
return DoubleFunction.create(operands[0], DoubleFunction.CBRT);
} else if (function == HazelcastSqlOperatorTable.COS) {
return DoubleFunction.create(operands[0], DoubleFunction.COS);
} else if (function == HazelcastSqlOperatorTable.SIN) {
return DoubleFunction.create(operands[0], DoubleFunction.SIN);
} else if (function == HazelcastSqlOperatorTable.TAN) {
return DoubleFunction.create(operands[0], DoubleFunction.TAN);
} else if (function == HazelcastSqlOperatorTable.COT) {
return DoubleFunction.create(operands[0], DoubleFunction.COT);
} else if (function == HazelcastSqlOperatorTable.ACOS) {
return DoubleFunction.create(operands[0], DoubleFunction.ACOS);
} else if (function == HazelcastSqlOperatorTable.ASIN) {
return DoubleFunction.create(operands[0], DoubleFunction.ASIN);
} else if (function == HazelcastSqlOperatorTable.ATAN) {
return DoubleFunction.create(operands[0], DoubleFunction.ATAN);
} else if (function == HazelcastSqlOperatorTable.ATAN2) {
assert operands.length == 2;
return DoubleBiFunction.create(operands[0], operands[1], DoubleBiFunction.ATAN2);
} else if (function == HazelcastSqlOperatorTable.EXP) {
return DoubleFunction.create(operands[0], DoubleFunction.EXP);
} else if (function == HazelcastSqlOperatorTable.LN) {
return DoubleFunction.create(operands[0], DoubleFunction.LN);
} else if (function == HazelcastSqlOperatorTable.LOG10) {
return DoubleFunction.create(operands[0], DoubleFunction.LOG10);
} else if (function == HazelcastSqlOperatorTable.RAND) {
return RandFunction.create(operands.length == 0 ? null : operands[0]);
} else if (function == HazelcastSqlOperatorTable.ABS) {
return AbsFunction.create(operands[0], resultType);
} else if (function == SqlStdOperatorTable.PI) {
return ConstantExpression.create(Math.PI, resultType);
} else if (function == HazelcastSqlOperatorTable.SIGN) {
return SignFunction.create(operands[0], resultType);
} else if (function == HazelcastSqlOperatorTable.DEGREES) {
return DoubleFunction.create(operands[0], DoubleFunction.DEGREES);
} else if (function == HazelcastSqlOperatorTable.RADIANS) {
return DoubleFunction.create(operands[0], DoubleFunction.RADIANS);
} else if (function == HazelcastSqlOperatorTable.ROUND) {
return RoundTruncateFunction.create(operands[0], operands.length == 1 ? null : operands[1], resultType, false);
} else if (function == HazelcastSqlOperatorTable.TRUNCATE) {
return RoundTruncateFunction.create(operands[0], operands.length == 1 ? null : operands[1], resultType, true);
}
if (function == CHAR_LENGTH || function == CHARACTER_LENGTH || function == LENGTH) {
return CharLengthFunction.create(operands[0]);
} else if (function == HazelcastSqlOperatorTable.UPPER) {
return UpperFunction.create(operands[0]);
} else if (function == HazelcastSqlOperatorTable.LOWER) {
return LowerFunction.create(operands[0]);
} else if (function == HazelcastSqlOperatorTable.INITCAP) {
return InitcapFunction.create(operands[0]);
} else if (function == HazelcastSqlOperatorTable.ASCII) {
return AsciiFunction.create(operands[0]);
} else if (function == HazelcastSqlOperatorTable.SUBSTRING) {
Expression<?> input = operands[0];
Expression<?> start = operands[1];
Expression<?> length = operands.length > 2 ? operands[2] : null;
return SubstringFunction.create(input, start, length);
} else if (function == HazelcastSqlOperatorTable.LTRIM) {
return TrimFunction.create(operands[0], null, true, false);
} else if (function == HazelcastSqlOperatorTable.RTRIM) {
return TrimFunction.create(operands[0], null, false, true);
} else if (function == HazelcastSqlOperatorTable.BTRIM) {
return TrimFunction.create(operands[0], null, true, true);
} else if (function == HazelcastSqlOperatorTable.REPLACE) {
return ReplaceFunction.create(operands[0], operands[1], operands[2]);
} else if (function == HazelcastSqlOperatorTable.POSITION) {
Expression<?> start = operands.length > 2 ? operands[2] : null;
return PositionFunction.create(operands[0], operands[1], start);
} else if (function == HazelcastSqlOperatorTable.TO_TIMESTAMP_TZ) {
return ToTimestampTzFunction.create(operands[0]);
} else if (function == HazelcastSqlOperatorTable.TO_EPOCH_MILLIS) {
return ToEpochMillisFunction.create(operands[0]);
} else if (function == HazelcastSqlOperatorTable.CONCAT_WS) {
return ConcatWSFunction.create(operands);
} else if (function == HazelcastSqlOperatorTable.JSON_QUERY) {
final SqlJsonQueryWrapperBehavior wrapperBehavior = ((SymbolExpression) operands[2]).getSymbol();
final SqlJsonQueryEmptyOrErrorBehavior onEmpty = ((SymbolExpression) operands[3]).getSymbol();
final SqlJsonQueryEmptyOrErrorBehavior onError = ((SymbolExpression) operands[4]).getSymbol();
return JsonQueryFunction.create(operands[0], operands[1], wrapperBehavior, onEmpty, onError);
} else if (function == HazelcastJsonParseFunction.INSTANCE) {
return JsonParseFunction.create(operands[0]);
} else if (function == HazelcastSqlOperatorTable.JSON_VALUE) {
final SqlJsonValueEmptyOrErrorBehavior onEmpty = ((SymbolExpression) operands[4]).getSymbol();
final SqlJsonValueEmptyOrErrorBehavior onError = ((SymbolExpression) operands[5]).getSymbol();
return JsonValueFunction.create(operands[0], operands[1], operands[2], operands[3], resultType, onEmpty, onError);
} else if (function == HazelcastSqlOperatorTable.JSON_OBJECT) {
final SqlJsonConstructorNullClause nullClause = ((SymbolExpression) operands[0]).getSymbol();
final Expression<?>[] fields = Arrays.copyOfRange(operands, 1, operands.length);
return JsonObjectFunction.create(fields, nullClause);
} else if (function == HazelcastSqlOperatorTable.JSON_ARRAY) {
final SqlJsonConstructorNullClause nullClause = ((SymbolExpression) operands[0]).getSymbol();
final Expression<?>[] fields = Arrays.copyOfRange(operands, 1, operands.length);
return JsonArrayFunction.create(fields, nullClause);
}
break;
default:
break;
}
throw QueryException.error("Unsupported operator: " + operator);
}
Aggregations