diff --git a/spring-batch-infrastructure/src/main/java/org/springframework/batch/item/xml/StaxEventItemReader.java b/spring-batch-infrastructure/src/main/java/org/springframework/batch/item/xml/StaxEventItemReader.java index 605f0539e..246dd901a 100644 --- a/spring-batch-infrastructure/src/main/java/org/springframework/batch/item/xml/StaxEventItemReader.java +++ b/spring-batch-infrastructure/src/main/java/org/springframework/batch/item/xml/StaxEventItemReader.java @@ -91,6 +91,7 @@ public class StaxEventItemReader extends ExecutionContextUserSupport implements public void close(ExecutionContext executionContext) { initialized = false; + currentRecordCount = 0; try { if (fragmentReader != null) { fragmentReader.close(); diff --git a/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/FooInputSource.java b/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/FooInputSource.java index a73badfb5..188e8a4f3 100644 --- a/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/FooInputSource.java +++ b/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/FooInputSource.java @@ -48,6 +48,7 @@ class FooItemReader implements ItemStream, ItemReader, DisposableBean, Initializ }; public void close(ExecutionContext executionContext) { + inputSource.close(executionContext); } /*