From ee30259612db49fc9535662ae262ad58b6f96d1b Mon Sep 17 00:00:00 2001
From: Michael Minella <mminella@gopivotal.com>
Date: Mon, 16 Dec 2013 12:59:30 -0600
Subject: [PATCH] Added double check that all partition data has been processed
 before ending

---
 .../batch/core/jsr/partition/JsrPartitionHandler.java           | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spring-batch-core/src/main/java/org/springframework/batch/core/jsr/partition/JsrPartitionHandler.java b/spring-batch-core/src/main/java/org/springframework/batch/core/jsr/partition/JsrPartitionHandler.java
index 780fc17c9..eb0ce0d25 100644
--- a/spring-batch-core/src/main/java/org/springframework/batch/core/jsr/partition/JsrPartitionHandler.java
+++ b/spring-batch-core/src/main/java/org/springframework/batch/core/jsr/partition/JsrPartitionHandler.java
@@ -178,7 +178,7 @@ public class JsrPartitionHandler implements PartitionHandler, InitializingBean {
 
 			processFinishedPartitions(tasks, result);
 
-			if(tasks.size() == 0) {
+			if(tasks.size() == 0 && partitionDataQueue.isEmpty()) {
 				break;
 			}
 		}