Custom Batch Save Repository

AbstractBatchCustomRepositoryImpl.java

package com.bmchild.data.shared.batch;

import javax.sql.DataSource;

import org.apache.commons.lang.Validate;
import org.springframework.jdbc.core.BatchPreparedStatementSetter;
import org.springframework.jdbc.core.JdbcTemplate;

/**
 * @author bchild
 *
 */
public abstract class AbstractBatchCustomRepositoryImpl {
	
	private JdbcTemplate jdbcTemplate = null;

	protected int executeBatch(String sql, BatchPreparedStatementSetter batchPreparedStatementSetter) {
		Validate.notNull(jdbcTemplate, "Datasource not set for batch statement.");
		int[] rows = jdbcTemplate.batchUpdate(sql, batchPreparedStatementSetter);
		return sumRows(rows);
	}
	
	private int sumRows(int[] rows) {
		int rowCount = 0;
		for(int row : rows) {
			rowCount += row;
		}
		return rowCount;
	}

	public void setDataSource(DataSource dataSource) {
		this.jdbcTemplate = new JdbcTemplate(dataSource);
	}

}

BatchSaveCustomRepository.java

package com.bmchild.data.shared.batch;

import java.util.Collection;

/**
 * 
 * @author brettchild
 *
 * @param <T>
 */
public interface BatchSaveCustomRepository<T> {
	
	/**
	 * @param records
	 * @return number of records saved
	 */
	int batchSave(Collection<T> records);
}

MyEntityRepository.java

package com.bmchild.data.cti.repository;

import java.util.Date;
import java.util.List;

import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.repository.query.Param;

import com.bmchild.data.shared.batch.BatchSaveCustomRepository;

/*
* Our Repository get the default functionality of the JPARepository plus we all also add our batchSave.
* If we wanted to also add a batch update we could make a batch update interface and implement that in our concrete impl class
*/
public interface MyEntityRepository extends JpaRepository<MyEntity, Long>, BatchSaveCustomRepository<MyEntity> {
	
}

MyEntityRepositoryImpl.java

package com.bmchild.data.cti.repository;

import java.sql.Date;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.Collection;
import java.util.List;

import javax.annotation.PostConstruct;
import javax.sql.DataSource;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.jdbc.core.BatchPreparedStatementSetter;

import com.bmchild.data.shared.batch.AbstractBatchCustomRepositoryImpl;
import com.bmchild.data.shared.batch.BatchSaveCustomRepository;
import com.bmchild.data.shared.common.util.CollectionUtil;

/*
* Impl of our repo so we can add our batch save functionality
*/
public class MyEntityRepositoryImpl extends AbstractBatchCustomRepositoryImpl implements BatchSaveCustomRepository<MyEntity> {

	private static final String BATCH_INSERT_SQL = "insert into myentity (field1, field2) "
			+ " VALUES (?, ?)";

	/* We limit the number of records we process at a time to avoid too many parameters per query */		
	private static final Integer BATCH_SIZE = 500; 

	@Autowired
	@Qualifier("dataSource")
	private DataSource dataSource;

	@PostConstruct
	public void postConstruct() {
		setDataSource(dataSource);
	}
	
	@Override
	public int batchSave(Collection<MyEntity> records) {
		int count = 0;
		for(final List<MyEntity> batch : CollectionUtil.split(records, BATCH_SIZE)) {
			count += executeBatch(BATCH_INSERT_SQL, new BatchPreparedStatementSetter() {
			
				@Override
				public void setValues(PreparedStatement ps, int i) throws SQLException {
					MyEntity l = batch.get(i);
					ps.setString(1, l.getField1());
					ps.setString(2, l.getField2());
				}
				
				@Override
				public int getBatchSize() {
					return batch.size();
				}
			});
		}
		return count;
	}

}