org.onetwo.dbm.jdbc.DbmRowMapperResultSetExtractor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of onetwo-dbm Show documentation

onetwo-dbm

The newest version!

package org.onetwo.dbm.jdbc;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collection;

import org.springframework.jdbc.core.ResultSetExtractor;
import org.springframework.jdbc.core.RowMapper;

public class DbmRowMapperResultSetExtractor, T> extends AbstractResultSetExtractor implements ResultSetExtractor{

	protected final int rowsExpected;

	public DbmRowMapperResultSetExtractor(RowMapper rowMapper) {
		this(rowMapper, 0);
	}
	public DbmRowMapperResultSetExtractor(RowMapper rowMapper, int rowsExpected) {
		super(rowMapper);
		this.rowsExpected = rowsExpected;
	}
	
	@SuppressWarnings("unchecked")
	protected C createResult(){
		return (C) (rowsExpected>0?new ArrayList(rowsExpected): new ArrayList());
	}

	public C extractData(ResultSet rs) throws SQLException {
		C results = createResult();
		int rowNum = 0;
		T val = null;
		while (rs.next()) {
			val = rowMapper.mapRow(rs, rowNum++);
			addToResult(results, val);
		}
		return results;
	}

	protected void addToResult(C result, T val){
		if(val==null)
			return ;
		result.add(val);
	}

}