From 666f793e89c686c19f4d252adcd91f61bc37a2bc Mon Sep 17 00:00:00 2001
From: robokaso <robokaso@2e8e966d-6415-0410-94b9-dc4e65435d03>
Date: Fri, 26 Sep 2008 12:31:41 +0000
Subject: [PATCH] RESOLVED - BATCH-785: Fix Parallel Sample to use a batch
 update for process indicator

---
 .../sample/common/StagingItemWriter.java      | 44 +++++++++----------
 1 file changed, 22 insertions(+), 22 deletions(-)

diff --git a/spring-batch-samples/src/main/java/org/springframework/batch/sample/common/StagingItemWriter.java b/spring-batch-samples/src/main/java/org/springframework/batch/sample/common/StagingItemWriter.java
index 5adae6a1d..4f423c649 100644
--- a/spring-batch-samples/src/main/java/org/springframework/batch/sample/common/StagingItemWriter.java
+++ b/spring-batch-samples/src/main/java/org/springframework/batch/sample/common/StagingItemWriter.java
@@ -4,15 +4,14 @@ import java.io.Serializable;
 import java.sql.PreparedStatement;
 import java.sql.SQLException;
 import java.util.List;
+import java.util.ListIterator;
 
 import org.apache.commons.lang.SerializationUtils;
 import org.springframework.batch.core.StepExecution;
 import org.springframework.batch.core.StepExecutionListener;
-import org.springframework.batch.item.ClearFailedException;
-import org.springframework.batch.item.FlushFailedException;
 import org.springframework.batch.item.ItemWriter;
 import org.springframework.batch.repeat.ExitStatus;
-import org.springframework.jdbc.core.PreparedStatementSetter;
+import org.springframework.jdbc.core.BatchPreparedStatementSetter;
 import org.springframework.jdbc.core.support.JdbcDaoSupport;
 import org.springframework.jdbc.support.incrementer.DataFieldMaxValueIncrementer;
 import org.springframework.jdbc.support.lob.DefaultLobHandler;
@@ -71,32 +70,33 @@ public class StagingItemWriter<T> extends JdbcDaoSupport implements StepExecutio
 	 * 
 	 * @see ItemWriter#write(java.util.List)
 	 */
-	public void write(List<? extends T> items) {
+	public void write(final List<? extends T> items) {
 
-		for (T data : items) {
+		final ListIterator<? extends T> itemIterator = items.listIterator();
+		getJdbcTemplate().batchUpdate("INSERT into BATCH_STAGING (ID, JOB_ID, VALUE, PROCESSED) values (?,?,?,?)",
+				new BatchPreparedStatementSetter() {
 
-			final long id = incrementer.nextLongValue();
-			final long jobId = stepExecution.getJobExecution().getJobId();
-			final byte[] blob = SerializationUtils.serialize((Serializable) data);
-			getJdbcTemplate().update("INSERT into BATCH_STAGING (ID, JOB_ID, VALUE, PROCESSED) values (?,?,?,?)",
-					new PreparedStatementSetter() {
-						public void setValues(PreparedStatement ps) throws SQLException {
-							ps.setLong(1, id);
-							ps.setLong(2, jobId);
-							lobHandler.getLobCreator().setBlobAsBytes(ps, 3, blob);
-							ps.setString(4, NEW);
-						}
+					public int getBatchSize() {
+						return items.size();
+					}
 
-					});
+					public void setValues(PreparedStatement ps, int i) throws SQLException {
 
-		}
+						long id = incrementer.nextLongValue();
+						long jobId = stepExecution.getJobExecution().getJobId();
 
-	}
+						Assert.state(itemIterator.nextIndex() == i,
+								"Item ordering must be preserved in batch sql update");
 
-	public void clear() throws ClearFailedException {
-	}
+						byte[] blob = SerializationUtils.serialize((Serializable) itemIterator.next());
+
+						ps.setLong(1, id);
+						ps.setLong(2, jobId);
+						lobHandler.getLobCreator().setBlobAsBytes(ps, 3, blob);
+						ps.setString(4, NEW);
+					}
+				});
 
-	public void flush() throws FlushFailedException {
 	}
 
 	/*