com.hazelcast.org.apache.calcite.rel.type.DynamicRecordTypeImpl 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.rel.type;
import com.hazelcast.org.apache.calcite.sql.type.SqlTypeExplicitPrecedenceList;
import com.hazelcast.org.apache.calcite.sql.type.SqlTypeFamily;
import com.hazelcast.org.apache.calcite.sql.type.SqlTypeName;
import com.hazelcast.org.apache.calcite.util.Pair;
import com.hazelcast.com.google.common.collect.ImmutableList;
import com.hazelcast.org.checkerframework.checker.nullness.qual.Nullable;
import java.util.List;
/**
* Implementation of {@link RelDataType} for a dynamic table.
*
* It's used during SQL validation, where the field list is mutable for
* the getField() call. After SQL validation, a normal {@link RelDataTypeImpl}
* with an immutable field list takes the place of the DynamicRecordTypeImpl
* instance.
*/
public class DynamicRecordTypeImpl extends DynamicRecordType {
private final RelDataTypeHolder holder;
/** Creates a DynamicRecordTypeImpl. */
@SuppressWarnings("method.invocation.invalid")
public DynamicRecordTypeImpl(RelDataTypeFactory typeFactory) {
this.holder = new RelDataTypeHolder(typeFactory);
computeDigest();
}
@Override public List getFieldList() {
return holder.getFieldList();
}
@Override public int getFieldCount() {
return holder.getFieldCount();
}
@Override public @Nullable RelDataTypeField getField(String fieldName,
boolean caseSensitive, boolean elideRecord) {
final Pair pair =
holder.getFieldOrInsert(fieldName, caseSensitive);
// If a new field is added, we should re-compute the digest.
if (pair.right) {
computeDigest();
}
return pair.left;
}
@Override public List getFieldNames() {
return holder.getFieldNames();
}
@Override public SqlTypeName getSqlTypeName() {
return SqlTypeName.ROW;
}
@Override public RelDataTypePrecedenceList getPrecedenceList() {
return new SqlTypeExplicitPrecedenceList(ImmutableList.of());
}
@Override protected void generateTypeString(StringBuilder sb, boolean withDetail) {
sb.append("(DynamicRecordRow").append(getFieldNames()).append(")");
}
@Override public boolean isStruct() {
return true;
}
@Override public RelDataTypeFamily getFamily() {
SqlTypeFamily family = getSqlTypeName().getFamily();
return family != null ? family : this;
}
}