io.deephaven.engine.table.impl.util.TableBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of deephaven-engine-table Show documentation
Engine Table: Implementation and closely-coupled utilities
There is a newer version: 0.37.1
Show newest version
/**
 * Copyright (c) 2016-2022 Deephaven Data Labs and Patent Pending
 */
package io.deephaven.engine.table.impl.util;

import io.deephaven.engine.rowset.RowSetFactory;
import io.deephaven.engine.table.ColumnDefinition;
import io.deephaven.engine.table.Table;
import io.deephaven.engine.table.TableDefinition;
import io.deephaven.engine.table.WritableColumnSource;
import io.deephaven.engine.table.impl.QueryTable;
import io.deephaven.engine.table.impl.sources.ArrayBackedColumnSource;
import io.deephaven.util.type.TypeUtils;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

/**
 * Class to aid in building Tables from a TableDefinition.
 */
@Deprecated
public class TableBuilder {
    private final TableDefinition def;
    private final List rows;

    /**
     * Creates a TableBuilder object based on a table Definition.
     *
     * @param def the definition of the table that you want to build
     */
    public TableBuilder(TableDefinition def) {
        this(def, 1000);
    }

    public TableBuilder(TableDefinition def, int initialSize) {
        this.def = def;
        rows = new ArrayList<>(initialSize);
    }

    /**
     * returns the number of rows the table has
     *
     * @return the size of the row List
     */
    public int rowCount() {
        return rows.size();
    }

    /**
     * Adds a row to the table. Items will be inserted into the row the order they are put into this method
     *
     * @param items The items that will appear in the row. Must be the same amount of items as columns
     */
    public void addRow(Object... items) {
        checkRow(items);
        rows.add(items);
    }

    /**
     * Checks if a the right number of items were added and that they were the right type.
     *
     * @param items the item array to be checked
     */
    private void checkRow(Object[] items) {
        List> colTypes = def.getColumnTypes();
        if (items.length != colTypes.size()) {
            throw new IllegalArgumentException(
                    "Incorrect column count: expected " + colTypes.size() + " got " + items.length);
        }

        for (int i = 0; i < colTypes.size(); i++) {
            if (items[i] != null && !TypeUtils.getUnboxedTypeIfBoxed(colTypes.get(i))
                    .isAssignableFrom(TypeUtils.getUnboxedTypeIfBoxed(items[i].getClass()))) {
                throw new IllegalArgumentException("Incorrect type for column " + def.getColumnNames().get(i)
                        + ": expected " + colTypes.get(i).getName()
                        + " got " + items[i].getClass().getName());
            }
        }

    }

    /**
     * Builds the table from the TableDefinition and the rows added
     *
     * @return the table
     */
    public Table build() {
        Map> map = new LinkedHashMap<>();
        for (ColumnDefinition columnDefinition : def.getColumns()) {
            WritableColumnSource cs = ArrayBackedColumnSource.getMemoryColumnSource(
                    rows.size(), columnDefinition.getDataType());
            // noinspection unchecked
            map.put(columnDefinition.getName(), (WritableColumnSource