java/sql-dk/src/info/globalcode/sql/dk/formatting/ColumnsHeader.java
author František Kučera <franta-hg@frantovo.cz>
Mon, 23 Dec 2013 11:50:24 +0100
branchv_0
changeset 37 9e6f8e5d5f98
parent 23 d8faf91519a5
child 39 be8db46a38c3
permissions -rw-r--r--
support SQL commands returning more ResultSets + remove COMMAND_TYPE (type is now derived from result returned from SQL – it is not needed to specify the type on CLI)

/**
 * SQL-DK
 * Copyright © 2013 František Kučera (frantovo.cz)
 *
 * This program 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.
 *
 * This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
 */
package info.globalcode.sql.dk.formatting;

import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

/**
 *
 * @author Ing. František Kučera (frantovo.cz)
 */
public class ColumnsHeader {

	private ResultSetMetaData metaData;

	public ColumnsHeader(ResultSetMetaData metaData) {
		this.metaData = metaData;
	}

	public int getColumnCount() {
		try {
			return metaData.getColumnCount();
		} catch (SQLException e) {
			throw new IllegalStateException("Error during getting column count.", e);
		}
	}

	public List<ColumnDescriptor> getColumnDescriptors() {
		try {
			int count = metaData.getColumnCount();
			List<ColumnDescriptor> list = new ArrayList<>(count);

			for (int i = 1; i <= count; i++) {
				ColumnDescriptor cd = new ColumnDescriptor();
				cd.setFirstColumn(i == 1);
				cd.setLastColumn(i == count);
				cd.setLabel(metaData.getColumnLabel(i));
				cd.setName(metaData.getColumnName(i));
				cd.setType(metaData.getColumnType(i));
				cd.setTypeName(metaData.getColumnTypeName(i));
				/** TODO: more properties */
				list.add(cd);
			}

			return list;
		} catch (SQLException e) {
			throw new IllegalStateException("Error during building column descriptors.", e);
		}
	}
}