-
Notifications
You must be signed in to change notification settings - Fork 3.1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Move ImplementAvgBigint logic from connectors into base jdbc module
- Loading branch information
1 parent
1f781db
commit 1b5db4a
Showing
8 changed files
with
106 additions
and
181 deletions.
There are no files selected for viewing
85 changes: 85 additions & 0 deletions
85
...trino-base-jdbc/src/main/java/io/trino/plugin/jdbc/expression/BaseImplementAvgBigint.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,85 @@ | ||
| /* | ||
| * Licensed under the Apache License, Version 2.0 (the "License"); | ||
| * you may not use this file except in compliance with the License. | ||
| * You may obtain a copy of the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software | ||
| * distributed under the License is distributed on an "AS IS" BASIS, | ||
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| * See the License for the specific language governing permissions and | ||
| * limitations under the License. | ||
| */ | ||
| package io.trino.plugin.jdbc.expression; | ||
|
|
||
| import io.trino.matching.Capture; | ||
| import io.trino.matching.Captures; | ||
| import io.trino.matching.Pattern; | ||
| import io.trino.plugin.base.expression.AggregateFunctionRule; | ||
| import io.trino.plugin.jdbc.JdbcColumnHandle; | ||
| import io.trino.plugin.jdbc.JdbcExpression; | ||
| import io.trino.plugin.jdbc.JdbcTypeHandle; | ||
| import io.trino.spi.connector.AggregateFunction; | ||
| import io.trino.spi.expression.Variable; | ||
|
|
||
| import java.sql.Types; | ||
| import java.util.Optional; | ||
|
|
||
| import static com.google.common.base.Verify.verify; | ||
| import static io.trino.matching.Capture.newCapture; | ||
| import static io.trino.plugin.base.expression.AggregateFunctionPatterns.basicAggregation; | ||
| import static io.trino.plugin.base.expression.AggregateFunctionPatterns.expressionType; | ||
| import static io.trino.plugin.base.expression.AggregateFunctionPatterns.functionName; | ||
| import static io.trino.plugin.base.expression.AggregateFunctionPatterns.singleInput; | ||
| import static io.trino.plugin.base.expression.AggregateFunctionPatterns.variable; | ||
| import static io.trino.spi.type.BigintType.BIGINT; | ||
| import static io.trino.spi.type.DoubleType.DOUBLE; | ||
| import static java.lang.String.format; | ||
|
|
||
| /** | ||
| * Implements {@code avg(x)} for bigint columns while preserving Trino semantics. | ||
| * Trino semantics say the output should be a double but pushing down the aggregation to some databases | ||
| * can result in rounding of the output to a bigint. | ||
| */ | ||
| public abstract class BaseImplementAvgBigint | ||
| implements AggregateFunctionRule<JdbcExpression> | ||
| { | ||
| private final Capture<Variable> input; | ||
|
|
||
| public BaseImplementAvgBigint() | ||
| { | ||
| this.input = newCapture(); | ||
| } | ||
|
|
||
| @Override | ||
| public Pattern<AggregateFunction> getPattern() | ||
| { | ||
| return basicAggregation() | ||
| .with(functionName().equalTo("avg")) | ||
| .with(singleInput().matching( | ||
| variable() | ||
| .with(expressionType().matching(type -> type == BIGINT)) | ||
| .capturedAs(this.input))); | ||
| } | ||
|
|
||
| @Override | ||
| public Optional<JdbcExpression> rewrite(AggregateFunction aggregateFunction, Captures captures, RewriteContext context) | ||
| { | ||
| Variable input = captures.get(this.input); | ||
| JdbcColumnHandle columnHandle = (JdbcColumnHandle) context.getAssignment(input.getName()); | ||
| verify(aggregateFunction.getOutputType() == DOUBLE); | ||
|
|
||
| String columnName = context.getIdentifierQuote().apply(columnHandle.getColumnName()); | ||
|
|
||
| return Optional.of(new JdbcExpression( | ||
| format(getRewriteFormatExpression(), columnName), | ||
| new JdbcTypeHandle(Types.DOUBLE, Optional.of("double"), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty()))); | ||
| } | ||
|
|
||
| /** | ||
| * Implement this method for each connector supporting avg(bigint) pushdown | ||
| * @return A format string expression with a single placeholder for the column name; The string expression pushes down avg to the remote database | ||
| */ | ||
| protected abstract String getRewriteFormatExpression(); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters