org.meridor.perspective.sql.DataUtils Maven / Gradle / Ivy

Go to download
package org.meridor.perspective.sql;

import java.util.List;

public final class DataUtils {

    public static int getColumnIndex(Data data, String columnName) {
        List columnNames = data.getColumnNames();
        int firstColumnIndex = columnNames.indexOf(columnName);
        boolean columnExists = (firstColumnIndex != -1);
        if (!columnExists) {
            throw new IllegalArgumentException(String.format("No such column: %s", columnName));
        }
        return firstColumnIndex;
    }
    
    public static Object get(Data data, Row row, int columnIndex) {
        checkColumnIndex(data, columnIndex);
        return row.getValues().get(columnIndex);
    }

    public static Object get(Data data, Row row, String columnName) {
        return get(data, row, getColumnIndex(data, columnName));
    }

    public static void put(Data data, Row row, int columnIndex, Object value) {
        checkColumnIndex(data, columnIndex);
        row.getValues().add(columnIndex, value);
    }

    public static void put(Data data, Row row, String columnName, Object value) {
        int columnIndex = getColumnIndex(data, columnName);
        checkColumnIndex(data, columnIndex);
        row.getValues().add(columnIndex, value);
    }
    
    private static void checkColumnIndex(Data data, int columnIndex) {
        final int NUM_COLS = data.getColumnNames().size();
        if (columnIndex < 0 && columnIndex >= NUM_COLS) {
            throw new IllegalArgumentException(String.format(
                    "Column index should be between 0 and %d but %d was specified",
                    NUM_COLS,
                    columnIndex
            ));
        }
    }
    
    private DataUtils() {
    }
}