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

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

import org.springframework.jdbc.support.rowset.SqlRowSet;
import org.springframework.jdbc.support.rowset.SqlRowSetMetaData;
import org.tinygroup.tinyscript.dataset.AbstractDataSet;
import org.tinygroup.tinyscript.dataset.Field;
import org.tinygroup.tinyscript.interpret.ResourceBundleUtil;

/**
 * 包装Spring记录集原生类型
 * @author yancheng11334
 *
 */
public class SqlRowSetDataSet extends AbstractDataSet{
    private SqlRowSet sqlRowSet;
    private int columnCount;
    
	public SqlRowSetDataSet(SqlRowSet sqlRowSet){
		this.sqlRowSet = sqlRowSet;
		try {
			SqlRowSetMetaData metaData= sqlRowSet.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 (Exception e) {
			throw new RuntimeException(e);
		}
	}
	
	public SqlRowSetDataSet(SqlRowSet sqlRowSet,boolean tag){
		this(sqlRowSet);
		this.setIndexFromOne(tag);
	}
	
	/**
	 * 返回Spring记录集原生类型
	 * @return
	 */
	public SqlRowSet getSqlRowSet() {
		return sqlRowSet;
	}

	public boolean isReadOnly() {
		return true;
	}

	public void first() throws Exception {
		sqlRowSet.first();
	}

	public boolean previous() throws Exception {
		return sqlRowSet.previous();
	}

	public void beforeFirst() throws Exception {
		sqlRowSet.beforeFirst();
	}

	public void afterLast() throws Exception {
		sqlRowSet.afterLast();
	}

	public boolean next() throws Exception {
		return sqlRowSet.next();
	}

	public boolean absolute(int row) throws Exception {
		return sqlRowSet.absolute(row);
	}

	public int getRows() throws Exception {
		throwNotSupportMethod();
		return 0;
	}

	public int getColumns() throws Exception {
		return sqlRowSet.getMetaData().getColumnCount();
	}

	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 {
		if(isIndexFromOne()){
			return (T) sqlRowSet.getObject(col);
		}else{
			return (T) sqlRowSet.getObject(col+1);
		}
	}

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

}