dk.eobjects.metamodel.data.DataSet Maven / Gradle / Ivy

Go to download
/**
 *  This file is part of MetaModel.
 *
 *  MetaModel is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  MetaModel is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with MetaModel.  If not, see .
 */
package dk.eobjects.metamodel.data;

import java.io.Closeable;
import java.sql.ResultSet;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;

import javax.swing.table.TableModel;

import org.apache.commons.lang.builder.EqualsBuilder;
import org.apache.commons.lang.builder.HashCodeBuilder;

import dk.eobjects.metamodel.MetaModelException;
import dk.eobjects.metamodel.query.Query;
import dk.eobjects.metamodel.query.SelectItem;

public class DataSet implements Closeable {

	private IDataSetStrategy _strategy;

	public DataSet(Query query, Statement statement, ResultSet resultSet) {
		this(new JdbcDataSetStrategy(query, statement, resultSet));
	}

	public DataSet(SelectItem[] selectItems, Statement statement,
			ResultSet resultSet) {
		this(new JdbcDataSetStrategy(selectItems, statement, resultSet));
	}

	public DataSet(IDataSetStrategy strategy) {
		if (strategy == null) {
			throw new IllegalArgumentException("Strategy cannot be null");
		}
		_strategy = strategy;
	}

	public DataSet(SelectItem[] selectItems, List data) {
		this(new InMemoryDataSetStrategy(selectItems, data, null));
	}

	public DataSet(List rows) {
		this(new InMemoryDataSetStrategy(rows));
	}

	public DataSet(String[] columnHeaders, List data) {
		SelectItem[] selectItems = new SelectItem[columnHeaders.length];
		for (int i = 0; i < columnHeaders.length; i++) {
			selectItems[i] = new SelectItem(columnHeaders[i], null);
		}
		_strategy = new InMemoryDataSetStrategy(selectItems, data, null);
	}

	public SelectItem[] getSelectItems() {
		return _strategy.getSelectItems();
	}

	public int indexOf(SelectItem item) {
		if (item != null) {
			SelectItem[] items = _strategy.getSelectItems();
			for (int i = 0; i < items.length; i++) {
				if (item.equalsIgnoreAlias(items[i])) {
					return i;
				}
			}
		}
		return -1;
	}

	public boolean next() {
		return _strategy.next();
	}

	public Row getRow() throws MetaModelException {
		return _strategy.getRow();
	}

	public void close() {
		_strategy.close();
	}

	/**
	 * Converts the DataSet into a TableModel
	 */
	public TableModel toTableModel() {
		TableModel tableModel = new DataSetTableModel(this);
		return tableModel;
	}

	/**
	 * Converts the DataSet into a list of object arrays
	 */
	public List toObjectArrays() {
		List objects = new ArrayList();
		while (next()) {
			Row row = getRow();
			objects.add(row.getValues());
		}
		close();
		return objects;
	}

	@Override
	public String toString() {
		return "DataSet[strategy=" + _strategy + "]";
	}

	@Override
	public boolean equals(Object obj) {
		if (obj instanceof DataSet) {
			DataSet that = (DataSet) obj;
			EqualsBuilder eb = new EqualsBuilder();
			eb.append(this._strategy, that._strategy);
			return eb.isEquals();
		}
		return false;
	}

	@Override
	public int hashCode() {
		return new HashCodeBuilder().append(_strategy).toHashCode();
	}
}