org.tinygroup.tinyscript.database.ResultSetDataSet Maven / Gradle / Ivy

Go to download
package org.tinygroup.tinyscript.database;

import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;

import org.tinygroup.tinyscript.dataset.AbstractDataSet;
import org.tinygroup.tinyscript.dataset.Field;
import org.tinygroup.tinyscript.interpret.ResourceBundleUtil;

/**
 * Created by luoguo on 2014/7/4.
 */
public class ResultSetDataSet extends AbstractDataSet {
	private final ResultSet resultSet;
	private int columnCount;

	public ResultSetDataSet(ResultSet resultSet) {
		this.resultSet = resultSet;
		try {
			ResultSetMetaData metaData = resultSet.getMetaData();
			columnCount = metaData.getColumnCount();
			for (int i = 1; i <= columnCount; i++) {
				Field field = new Field();
				field.setName(metaData.getColumnLabel(i)); // 需要支持as场景
				field.setTitle(metaData.getColumnLabel(i));
				field.setType(metaData.getColumnClassName(i));
				fields.add(field);
			}
			setFields(fields);
		} catch (SQLException e) {
			throw new RuntimeException(e);
		}
	}

	public ResultSetDataSet(ResultSet resultSet, boolean tag) {
		this(resultSet);
		this.setIndexFromOne(tag);
	}

	public boolean isReadOnly() {
		try {
			return resultSet.getConcurrency() == ResultSet.CONCUR_READ_ONLY;
		} catch (SQLException e) {
			throw new RuntimeException(e);
		}
	}

	private void checkNoForwardType() throws Exception {
		try {
			if (resultSet.getType() == ResultSet.TYPE_FORWARD_ONLY) {
				throw new Exception(ResourceBundleUtil.getResourceMessage("database", "database.prev.error"));
			}
		} catch (SQLException e) {
			throw new Exception(e);
		}
	}

	public void first() throws Exception {
		checkNoForwardType();
		try {
			resultSet.first();
		} catch (SQLException e) {
			throw new Exception(e);
		}
	}

	public boolean previous() throws Exception {
		checkNoForwardType();
		try {
			return resultSet.previous();
		} catch (SQLException e) {
			throw new Exception(e);
		}
	}

	public void beforeFirst() throws Exception {
		checkNoForwardType();
		try {
			resultSet.beforeFirst();
		} catch (SQLException e) {
			throw new Exception(e);
		}

	}

	public void afterLast() throws Exception {
		checkNoForwardType();
		try {
			resultSet.afterLast();
		} catch (SQLException e) {
			throw new Exception(e);
		}
	}

	public boolean next() throws Exception {
		try {
			return resultSet.next();
		} catch (SQLException e) {
			throw new Exception(e);
		}

	}

	public boolean absolute(int row) throws Exception {
		checkNoForwardType();
		try {
			if (isIndexFromOne()) {
				return resultSet.absolute(row);
			} else {
				return resultSet.absolute(row + 1);
			}

		} catch (SQLException e) {
			throw new Exception(e);
		}
	}

	public int getRows() throws Exception {
		throw new Exception(ResourceBundleUtil.getResourceMessage("database", "database.method.nosupport","getRows"));
	}

	public int getColumns() throws Exception {
		return columnCount;
	}

	public  T getData(int row, int col) throws Exception {
		if (absolute(row)) {
			return getData(col);
		}
		throw new Exception(ResourceBundleUtil.getResourceMessage("database", "database.data.noexists", row));
	}

	public  void setData(int row, int col, T data) throws Exception {
		throwNotSupportMethod();
	}

	@SuppressWarnings("unchecked")
	public  T getData(int col) throws Exception {
		try {
			if (isIndexFromOne()) {
				return (T) resultSet.getObject(col);
			} else {
				return (T) resultSet.getObject(col + 1);
			}

		} catch (SQLException e) {
			throw new Exception(e);
		}
	}

	public  void setData(int col, T data) throws Exception {
		throwNotSupportMethod();
	}

	public  void setData(String filedName, T data) throws Exception {
		throwNotSupportMethod();
	}

	public void clean() {
		super.clean();
		try {
			resultSet.close();
		} catch (SQLException e) {
			// TODO doNothing
		}
	}
}