VariousColumnTable

VariousColumnTable.java

package com.naosim.table;

import java.util.HashMap;
import java.util.Map;

public class VariousColumnTable<ROW> {
    private final Map<Class, Integer> columnIndex;
    private final Map<ROW, Object[]> rows;

    public VariousColumnTable(Class ...columns) {
        this.rows = new HashMap<>();
        this.columnIndex = new HashMap<>();
        for(int i = 0; i < columns.length; i++) {
            columnIndex.put(columns[i], i);
        }
    }

    public VariousColumnTable put(ROW row, Object ...cells) {
        rows.put(row, cells);
        return this;
    }

    public <T> T get(ROW row, Class<T> column) {
        Object result = rows.get(row)[columnIndex.get(column)];
        return column.cast(result);
    }
}