Handle multiple results per document in a sync batch

1 files changed, 17 insertions, 0 deletions

diff --git a/vespa-http-client/src/main/java/com/yahoo/vespa/http/client/SyncFeedClient.java b/vespa-http-client/src/main/java/com/yahoo/vespa/http/client/SyncFeedClient.java
index c2d237d63af..c1e7f9f04c5 100644
--- a/vespa-http-client/src/main/java/com/yahoo/vespa/http/client/SyncFeedClient.java
+++ b/vespa-http-client/src/main/java/com/yahoo/vespa/http/client/SyncFeedClient.java
@@ -123,6 +123,23 @@ public class SyncFeedClient implements AutoCloseable {
          */
         private LinkedHashMap<String, Result> results = null;
 
+        void resetExpectedResults() {
+            synchronized (monitor) {
+                if (results != null)
+                    throw new ConcurrentModificationException("A SyncFeedClient instance is used by multiple threads");
+
+                resultsReceived = 0;
+                exception = null;
+                results = new LinkedHashMap<>();
+            }
+        }
+
+        void addExpectationOfResultFor(String operationId) {
+            synchronized (monitor) {
+                results.put(operationId, null);
+            }
+        }
+
         void expectResultsOf(List<SyncOperation> operations) {
             synchronized (monitor) {
                 if (results != null)