Class InputColumn

server/src/main/java/io/crate/expression/symbol/InputColumn.java:42–145

A symbol which represents a column of a result array

 * A symbol which represents a column of a result array
 */
public class InputColumn implements Symbol, Comparable<InputColumn> {

    private final DataType<?> dataType;
    private final int index;

    /**
     * Map each data type to an {@link InputColumn} with an index corresponding to the data type's position in the list.
     */
    public static List<Symbol> mapToInputColumns(List<DataType<?>> dataTypes) {
        List<Symbol> inputColumns = new ArrayList<>(dataTypes.size());
        for (int i = 0; i < dataTypes.size(); i++) {
            inputColumns.add(new InputColumn(i, dataTypes.get(i)));
        }
        return inputColumns;
    }

    /**
     * Map each symbol to an {@link InputColumn} with an index corresponding to the symbol's position in the collection.
     */
    public static List<Symbol> mapToInputColumns(Collection<? extends Symbol> symbols) {
        List<Symbol> inputColumns = new ArrayList<>(symbols.size());
        int idx = 0;
        for (Symbol symbol : symbols) {
            inputColumns.add(new InputColumn(idx, symbol.valueType()));
            idx++;
        }
        return inputColumns;
    }

    public InputColumn(int index, @Nullable DataType<?> dataType) {
        assert index >= 0 : "index must be >= 0";
        this.index = index;
        this.dataType = Objects.requireNonNullElse(dataType, DataTypes.UNDEFINED);
    }

    public InputColumn(StreamInput in) throws IOException {
        index = in.readVInt();
        dataType = DataTypes.fromStream(in);
    }

    public InputColumn(int index) {
        this(index, null);
    }

    public int index() {
        return index;
    }

    @Override
    public SymbolType symbolType() {
        return SymbolType.INPUT_COLUMN;
    }

    @Override
    public DataType<?> valueType() {
        return dataType;
    }