walterddr commented on code in PR #9937: URL: https://github.com/apache/pinot/pull/9937#discussion_r1042801259
########## pinot-query-planner/src/main/java/org/apache/calcite/sql/util/PinotChainedSqlOperatorTable.java: ########## @@ -0,0 +1,91 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you 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 org.apache.calcite.sql.util; + +import org.apache.calcite.sql.SqlFunctionCategory; +import org.apache.calcite.sql.SqlIdentifier; +import org.apache.calcite.sql.SqlOperator; +import org.apache.calcite.sql.SqlOperatorTable; +import org.apache.calcite.sql.SqlSyntax; +import org.apache.calcite.sql.validate.SqlNameMatcher; + +import com.google.common.collect.ImmutableList; + +import org.checkerframework.checker.nullness.qual.Nullable; + +import java.util.ArrayList; +import java.util.List; + +/** + * ============================================================================ + * THIS CLASS IS COPIED FROM Calcite's {@link org.apache.calcite.sql.util.ChainedSqlOperatorTable} and modified the + * function lookup to terminate early once found from ordered SqlOperatorTable list. + * ============================================================================ + * + * PinotChainedSqlOperatorTable implements the {@link SqlOperatorTable} interface by + * chaining together any number of underlying operator table instances. + */ +public class PinotChainedSqlOperatorTable implements SqlOperatorTable { + //~ Instance fields -------------------------------------------------------- + + protected final List<SqlOperatorTable> _tableList; + + //~ Constructors ----------------------------------------------------------- + + public PinotChainedSqlOperatorTable(List<SqlOperatorTable> tableList) { + this(ImmutableList.copyOf(tableList)); + } + + /** Internal constructor; call {@link SqlOperatorTables#chain}. */ + protected PinotChainedSqlOperatorTable(ImmutableList<SqlOperatorTable> tableList) { + _tableList = ImmutableList.copyOf(tableList); + } + + //~ Methods ---------------------------------------------------------------- + + @Deprecated // to be removed before 2.0 + public void add(SqlOperatorTable table) { + if (!_tableList.contains(table)) { + _tableList.add(table); + } + } + + @Override public void lookupOperatorOverloads(SqlIdentifier opName, + @Nullable SqlFunctionCategory category, SqlSyntax syntax, + List<SqlOperator> operatorList, SqlNameMatcher nameMatcher) { + for (SqlOperatorTable table : _tableList) { + table.lookupOperatorOverloads(opName, category, syntax, operatorList, + nameMatcher); + // ====================================================================== + // CHANGED LINES BELOW + // ====================================================================== + if (!operatorList.isEmpty()) { Review Comment: this is b/c if a single operator returns a list of matches, it will NEVER result in multiple types of matches. in our case b/c we remove the case sensitivity restriction. `FLOOR` is resolved into a `SqlFloorOperator` and a `UserDefinedSqlFunction("floor")` one from the std table and one from pinot catalog. if this happens calcite has some weird non-asserted direct casting that causes trouble -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
