org.dinky.shaded.paimon.utils.ProjectedRow 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 org.dinky.shaded.paimon.utils;
import org.dinky.shaded.paimon.data.BinaryString;
import org.dinky.shaded.paimon.data.Decimal;
import org.dinky.shaded.paimon.data.InternalArray;
import org.dinky.shaded.paimon.data.InternalMap;
import org.dinky.shaded.paimon.data.InternalRow;
import org.dinky.shaded.paimon.data.Timestamp;
import org.dinky.shaded.paimon.types.DataType;
import org.dinky.shaded.paimon.types.RowKind;
import java.util.Arrays;
/**
* An implementation of {@link InternalRow} which provides a projected view of the underlying {@link
* InternalRow}.
*
* Projection includes both reducing the accessible fields and reordering them.
*
*
Note: This class supports only top-level projections, not nested projections.
*/
public class ProjectedRow implements InternalRow {
protected final int[] indexMapping;
protected InternalRow row;
protected ProjectedRow(int[] indexMapping) {
this.indexMapping = indexMapping;
}
/**
* Replaces the underlying {@link InternalRow} backing this {@link ProjectedRow}.
*
*
This method replaces the row data in place and does not return a new object. This is done
* for performance reasons.
*/
public ProjectedRow replaceRow(InternalRow row) {
this.row = row;
return this;
}
// ---------------------------------------------------------------------------------------------
@Override
public int getFieldCount() {
return indexMapping.length;
}
@Override
public RowKind getRowKind() {
return row.getRowKind();
}
@Override
public void setRowKind(RowKind kind) {
row.setRowKind(kind);
}
@Override
public boolean isNullAt(int pos) {
if (indexMapping[pos] < 0) {
// TODO move this logical to hive
return true;
}
return row.isNullAt(indexMapping[pos]);
}
@Override
public boolean getBoolean(int pos) {
return row.getBoolean(indexMapping[pos]);
}
@Override
public byte getByte(int pos) {
return row.getByte(indexMapping[pos]);
}
@Override
public short getShort(int pos) {
return row.getShort(indexMapping[pos]);
}
@Override
public int getInt(int pos) {
return row.getInt(indexMapping[pos]);
}
@Override
public long getLong(int pos) {
return row.getLong(indexMapping[pos]);
}
@Override
public float getFloat(int pos) {
return row.getFloat(indexMapping[pos]);
}
@Override
public double getDouble(int pos) {
return row.getDouble(indexMapping[pos]);
}
@Override
public BinaryString getString(int pos) {
return row.getString(indexMapping[pos]);
}
@Override
public Decimal getDecimal(int pos, int precision, int scale) {
return row.getDecimal(indexMapping[pos], precision, scale);
}
@Override
public Timestamp getTimestamp(int pos, int precision) {
return row.getTimestamp(indexMapping[pos], precision);
}
@Override
public byte[] getBinary(int pos) {
return row.getBinary(indexMapping[pos]);
}
@Override
public InternalArray getArray(int pos) {
return row.getArray(indexMapping[pos]);
}
@Override
public InternalMap getMap(int pos) {
return row.getMap(indexMapping[pos]);
}
@Override
public InternalRow getRow(int pos, int numFields) {
return row.getRow(indexMapping[pos], numFields);
}
@Override
public boolean equals(Object o) {
throw new UnsupportedOperationException("Projected row data cannot be compared");
}
@Override
public int hashCode() {
throw new UnsupportedOperationException("Projected row data cannot be hashed");
}
@Override
public String toString() {
return row.getRowKind().shortString()
+ "{"
+ "indexMapping="
+ Arrays.toString(indexMapping)
+ ", mutableRow="
+ row
+ '}';
}
/**
* Like {@link #from(int[])}, but throws {@link IllegalArgumentException} if the provided {@code
* projection} array contains nested projections, which are not supported by {@link
* ProjectedRow}.
*
*
The array represents the mapping of the fields of the original {@link DataType}, including
* nested rows. For example, {@code [[0, 2, 1], ...]} specifies to include the 2nd field of the
* 3rd field of the 1st field in the top-level row.
*
* @see Projection
* @see ProjectedRow
*/
public static ProjectedRow from(int[][] projection) throws IllegalArgumentException {
return new ProjectedRow(
Arrays.stream(projection)
.mapToInt(
arr -> {
if (arr.length != 1) {
throw new IllegalArgumentException(
"ProjectedRowData doesn't support nested projections");
}
return arr[0];
})
.toArray());
}
/**
* Create an empty {@link ProjectedRow} starting from a {@code projection} array.
*
*
The array represents the mapping of the fields of the original {@link DataType}. For
* example, {@code [0, 2, 1]} specifies to include in the following order the 1st field, the 3rd
* field and the 2nd field of the row.
*
* @see Projection
* @see ProjectedRow
*/
public static ProjectedRow from(int[] projection) {
return new ProjectedRow(projection);
}
/**
* Create an empty {@link ProjectedRow} starting from a {@link Projection}.
*
*
Throws {@link IllegalStateException} if the provided {@code projection} array contains
* nested projections, which are not supported by {@link ProjectedRow}.
*
* @see Projection
* @see ProjectedRow
*/
public static ProjectedRow from(Projection projection) {
return new ProjectedRow(projection.toTopLevelIndexes());
}
}