Search in sources :

Example 21 with JdbcExpression

use of io.trino.plugin.jdbc.JdbcExpression in project trino by trinodb.

the class ImplementSqlServerVariance method rewrite.

@Override
public Optional<JdbcExpression> rewrite(AggregateFunction aggregateFunction, Captures captures, RewriteContext<String> context) {
    Variable argument = captures.get(ARGUMENT);
    JdbcColumnHandle columnHandle = (JdbcColumnHandle) context.getAssignment(argument.getName());
    verify(columnHandle.getColumnType().equals(DOUBLE));
    verify(aggregateFunction.getOutputType().equals(DOUBLE));
    return Optional.of(new JdbcExpression(format("VAR(%s)", context.rewriteExpression(argument).orElseThrow()), columnHandle.getJdbcTypeHandle()));
}
Also used : Variable(io.trino.spi.expression.Variable) JdbcColumnHandle(io.trino.plugin.jdbc.JdbcColumnHandle) JdbcExpression(io.trino.plugin.jdbc.JdbcExpression)

Aggregations

JdbcExpression (io.trino.plugin.jdbc.JdbcExpression)21 Variable (io.trino.spi.expression.Variable)21 JdbcColumnHandle (io.trino.plugin.jdbc.JdbcColumnHandle)18 JdbcTypeHandle (io.trino.plugin.jdbc.JdbcTypeHandle)2 CharType (io.trino.spi.type.CharType)2 DecimalType (io.trino.spi.type.DecimalType)2 VarcharType (io.trino.spi.type.VarcharType)2 Optional (java.util.Optional)1