Allow access to update counts in JdbcBatchItemWriter

Resolves #3829

Author: Robert McNees

Diff for: spring-batch-infrastructure/src/main/java/org/springframework/batch/item/database/JdbcBatchItemWriter.java

@@ -207,7 +207,17 @@ public void write(final Chunk<? extends T> chunk) throws Exception {
 					}
 				}
 			}
+
+			processUpdateCounts(updateCounts);
 		}
 	}
 
+	/**
+	 * Extension point to post process the update counts for each item.
+	 * @param updateCounts the array of update counts for each item
+	 */
+	protected void processUpdateCounts(int[] updateCounts) {
+		// No Op
+	}
+
 }

Diff for: spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/JdbcBatchItemWriterClassicTests.java

@@ -22,6 +22,7 @@
 
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
+import org.mockito.Mockito;
 
 import org.springframework.batch.item.Chunk;
 import org.springframework.dao.DataAccessException;
@@ -35,6 +36,7 @@
 import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.spy;
 import static org.mockito.Mockito.when;
 
 /**
@@ -139,4 +141,23 @@ void testWriteAndFlushWithFailure() throws Exception {
 		assertTrue(list.contains("foo"));
 	}
 
+	@Test
+	void testProcessUpdateCountsIsCalled() throws Exception {
+		JdbcBatchItemWriter<String> customWriter = spy(new JdbcBatchItemWriter<>());
+
+		customWriter.setSql("SQL");
+		customWriter.setJdbcTemplate(new NamedParameterJdbcTemplate(jdbcTemplate));
+		customWriter.setItemPreparedStatementSetter((item, ps) -> list.add(item));
+		customWriter.afterPropertiesSet();
+
+		ps.addBatch();
+		int[] updateCounts = { 123 };
+		when(ps.executeBatch()).thenReturn(updateCounts);
+		customWriter.write(Chunk.of("bar"));
+		assertEquals(2, list.size());
+		assertTrue(list.contains("SQL"));
+
+		Mockito.verify(customWriter, Mockito.times(1)).processUpdateCounts(updateCounts);
+	}
+
 }