diff options
author | Jon Marius Venstad <venstad@gmail.com> | 2020-06-10 17:40:30 +0200 |
---|---|---|
committer | Jon Marius Venstad <venstad@gmail.com> | 2020-06-11 13:45:11 +0200 |
commit | 16ddfeb30367b2ee20ceaa7d89594585b5365e77 (patch) | |
tree | 13d84544c2495b1e11dfb5a9f30d0433a63072eb | |
parent | 4911919473cf6511fbd193c95cf0db81e5ebae07 (diff) |
Minor fixes
-rw-r--r-- | documentapi/src/main/java/com/yahoo/documentapi/local/LocalVisitorSession.java | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/documentapi/src/main/java/com/yahoo/documentapi/local/LocalVisitorSession.java b/documentapi/src/main/java/com/yahoo/documentapi/local/LocalVisitorSession.java index 4a11d8ae8bc..022260d275d 100644 --- a/documentapi/src/main/java/com/yahoo/documentapi/local/LocalVisitorSession.java +++ b/documentapi/src/main/java/com/yahoo/documentapi/local/LocalVisitorSession.java @@ -62,7 +62,7 @@ public class LocalVisitorSession implements VisitorSession { // Iterate through all documents and pass on to data handler outstanding.forEach((id, document) -> { data.onMessage(new PutDocumentMessage(new DocumentPut(document)), - new AckToken(id)); + new AckToken(id)); }); // Transition to a terminal state when done state.updateAndGet(current -> { @@ -93,7 +93,8 @@ public class LocalVisitorSession implements VisitorSession { @Override public boolean isDone() { - return outstanding.isEmpty(); + return outstanding.isEmpty() // All documents ack'ed + && control.isDone(); // Control handler has been notified } @Override @@ -134,8 +135,7 @@ public class LocalVisitorSession implements VisitorSession { @Override public void destroy() { - if ( ! isDone()) - abort(); + abort(); } } |