com.hazelcast.org.apache.calcite.sql.validate.WithItemNamespace 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.validate;
import com.hazelcast.org.apache.calcite.rel.type.RelDataType;
import com.hazelcast.org.apache.calcite.rel.type.RelDataTypeFactory;
import com.hazelcast.org.apache.calcite.rel.type.RelDataTypeField;
import com.hazelcast.org.apache.calcite.sql.SqlIdentifier;
import com.hazelcast.org.apache.calcite.sql.SqlNode;
import com.hazelcast.org.apache.calcite.sql.SqlNodeList;
import com.hazelcast.org.apache.calcite.sql.SqlWithItem;
import com.hazelcast.org.apache.calcite.util.Pair;
import com.hazelcast.org.checkerframework.checker.nullness.qual.Nullable;
/** Very similar to {@link AliasNamespace}. */
class WithItemNamespace extends AbstractNamespace {
private final SqlWithItem withItem;
WithItemNamespace(SqlValidatorImpl validator, SqlWithItem withItem,
SqlNode enclosingNode) {
super(validator, enclosingNode);
this.withItem = withItem;
}
@Override protected RelDataType validateImpl(RelDataType targetRowType) {
final SqlValidatorNamespace childNs =
validator.getNamespaceOrThrow(withItem.query);
final RelDataType rowType = childNs.getRowTypeSansSystemColumns();
SqlNodeList columnList = withItem.columnList;
if (columnList == null) {
return rowType;
}
final RelDataTypeFactory.Builder builder =
validator.getTypeFactory().builder();
Pair.forEach(SqlIdentifier.simpleNames(columnList),
rowType.getFieldList(),
(name, field) -> builder.add(name, field.getType()));
return builder.build();
}
@Override public @Nullable SqlNode getNode() {
return withItem;
}
@Override public String translate(String name) {
if (withItem.columnList == null) {
return name;
}
final RelDataType underlyingRowType =
validator.getValidatedNodeType(withItem.query);
int i = 0;
for (RelDataTypeField field : getRowType().getFieldList()) {
if (field.getName().equals(name)) {
return underlyingRowType.getFieldList().get(i).getName();
}
++i;
}
throw new AssertionError("unknown field '" + name
+ "' in rowtype " + underlyingRowType);
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy