RESOLVED - issue BATCH-76: ItemWriter for SQL batch updates

http://jira.springframework.org/browse/BATCH-76

spring-batch-samples/src/main/java/org/springframework/batch/sample/item/writer/BatchSqlCustomerCreditIncreaseWriter.java

@@ -0,0 +1,69 @@
+package org.springframework.batch.sample.item.writer;
+
+import java.math.BigDecimal;
+
+import org.springframework.batch.io.support.BatchSqlUpdateItemWriter;
+import org.springframework.batch.item.ItemWriter;
+import org.springframework.batch.item.exception.ClearFailedException;
+import org.springframework.batch.item.exception.FlushFailedException;
+import org.springframework.batch.sample.domain.CustomerCredit;
+import org.springframework.beans.factory.InitializingBean;
+import org.springframework.util.Assert;
+
+/**
+ * Increases customer's credit by fixed amount, delegating to a
+ * {@link BatchSqlUpdateItemWriter} to push the result out to persistent
+ * storage.
+ * 
+ * @author Dave Syer
+ */
+public class BatchSqlCustomerCreditIncreaseWriter implements ItemWriter, InitializingBean {
+
+	private ItemWriter delegate;
+
+	public static final BigDecimal FIXED_AMOUNT = new BigDecimal(1000);
+
+	/**
+	 * Public setter for the {@link ItemWriter}, which must be an instance of
+	 * {@link BatchSqlUpdateItemWriter}.
+	 * @param delegate the delegate to set
+	 */
+	public void setDelegate(ItemWriter delegate) {
+		this.delegate = delegate;
+	}
+
+	/*
+	 * (non-Javadoc)
+	 * @see org.springframework.beans.factory.InitializingBean#afterPropertiesSet()
+	 */
+	public void afterPropertiesSet() throws Exception {
+		Assert.state(delegate instanceof BatchSqlUpdateItemWriter, "Delegate must be set and must be an instance of BatchSqlUpdateItemWriter");
+	}
+
+	/*
+	 * (non-Javadoc)
+	 * @see org.springframework.batch.item.processor.DelegatingItemWriter#doProcess(java.lang.Object)
+	 */
+	public void write(Object data) throws Exception {
+		CustomerCredit customerCredit = (CustomerCredit) data;
+		customerCredit.increaseCreditBy(FIXED_AMOUNT);
+		delegate.write(customerCredit);
+	}
+
+	/**
+	 * @throws ClearFailedException
+	 * @see org.springframework.batch.io.support.BatchSqlUpdateItemWriter#clear()
+	 */
+	public void clear() throws ClearFailedException {
+		delegate.clear();
+	}
+
+	/**
+	 * @throws FlushFailedException
+	 * @see org.springframework.batch.io.support.BatchSqlUpdateItemWriter#flush()
+	 */
+	public void flush() throws FlushFailedException {
+		delegate.flush();
+	}
+
+}

spring-batch-samples/src/main/java/org/springframework/batch/sample/item/writer/CustomerCreditUpdatePreparedStatementSetter.java

@@ -0,0 +1,39 @@
+/*
+ * Copyright 2006-2007 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.batch.sample.item.writer;
+
+import java.sql.PreparedStatement;
+import java.sql.SQLException;
+
+import org.springframework.batch.io.support.ItemPreparedStatementSetter;
+import org.springframework.batch.sample.domain.CustomerCredit;
+
+/**
+ * @author Dave Syer
+ *
+ */
+public class CustomerCreditUpdatePreparedStatementSetter implements ItemPreparedStatementSetter {
+
+	/* (non-Javadoc)
+	 * @see org.springframework.batch.io.support.ItemPreparedStatementSetter#setValues(java.lang.Object, java.sql.PreparedStatement)
+	 */
+	public void setValues(Object item, PreparedStatement ps) throws SQLException {
+		CustomerCredit customerCredit = (CustomerCredit) item;
+		ps.setBigDecimal(1, customerCredit.getCredit());
+		ps.setLong(2, customerCredit.getId());
+	}
+
+}

spring-batch-samples/src/main/java/org/springframework/batch/sample/mapping/CustomerCreditRowMapper.java

@@ -8,12 +8,14 @@ import org.springframework.jdbc.core.RowMapper;
 
 public class CustomerCreditRowMapper implements RowMapper {
 	
+	public static final String ID_COLUMN = "id";
 	public static final String NAME_COLUMN = "name";
 	public static final String CREDIT_COLUMN = "credit";
 
 	public Object mapRow(ResultSet rs, int rowNum) throws SQLException {
         CustomerCredit customerCredit = new CustomerCredit();
 
+        customerCredit.setId(rs.getInt(ID_COLUMN));
         customerCredit.setName(rs.getString(NAME_COLUMN));
         customerCredit.setCredit(rs.getBigDecimal(CREDIT_COLUMN));