com.hazelcast.org.apache.calcite.sql.util.ListSqlOperatorTable Maven / Gradle / Ivy
/*
* 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 com.hazelcast.org.apache.calcite.sql.util;
import com.hazelcast.org.apache.calcite.sql.SqlFunction;
import com.hazelcast.org.apache.calcite.sql.SqlFunctionCategory;
import com.hazelcast.org.apache.calcite.sql.SqlIdentifier;
import com.hazelcast.org.apache.calcite.sql.SqlOperator;
import com.hazelcast.org.apache.calcite.sql.SqlOperatorTable;
import com.hazelcast.org.apache.calcite.sql.SqlSyntax;
import com.hazelcast.org.apache.calcite.sql.validate.SqlNameMatcher;
import java.util.ArrayList;
import java.util.List;
/**
* Implementation of the {@link SqlOperatorTable} interface by using a list of
* {@link SqlOperator operators}.
*/
public class ListSqlOperatorTable implements SqlOperatorTable {
//~ Instance fields --------------------------------------------------------
private final List operatorList;
//~ Constructors -----------------------------------------------------------
public ListSqlOperatorTable() {
this(new ArrayList<>());
}
public ListSqlOperatorTable(List operatorList) {
this.operatorList = operatorList;
}
//~ Methods ----------------------------------------------------------------
public void add(SqlOperator op) {
operatorList.add(op);
}
public void lookupOperatorOverloads(SqlIdentifier opName,
SqlFunctionCategory category,
SqlSyntax syntax,
List operatorList,
SqlNameMatcher nameMatcher) {
for (SqlOperator operator : this.operatorList) {
if (operator.getSyntax() != syntax) {
continue;
}
if (!opName.isSimple()
|| !nameMatcher.matches(operator.getName(), opName.getSimple())) {
continue;
}
if (category != null
&& category != category(operator)
&& !category.isUserDefinedNotSpecificFunction()) {
continue;
}
operatorList.add(operator);
}
}
protected static SqlFunctionCategory category(SqlOperator operator) {
if (operator instanceof SqlFunction) {
return ((SqlFunction) operator).getFunctionType();
} else {
return SqlFunctionCategory.SYSTEM;
}
}
public List getOperatorList() {
return operatorList;
}
}