From 41a5368c200e5417c10d5f9bba9b21e7f18c00e6 Mon Sep 17 00:00:00 2001
From: Harald Musum <musum@yahooinc.com>
Date: Mon, 11 Mar 2024 14:38:08 +0100
Subject: Support serializing document remove operation to json format

---
 .../com/yahoo/vespa/feed/perf/SimpleFeeder.java    | 35 +++++++++++++++-------
 1 file changed, 24 insertions(+), 11 deletions(-)

(limited to 'vespaclient-java/src/main/java/com/yahoo')

diff --git a/vespaclient-java/src/main/java/com/yahoo/vespa/feed/perf/SimpleFeeder.java b/vespaclient-java/src/main/java/com/yahoo/vespa/feed/perf/SimpleFeeder.java
index b4ca98f316f..b41bb29376b 100644
--- a/vespaclient-java/src/main/java/com/yahoo/vespa/feed/perf/SimpleFeeder.java
+++ b/vespaclient-java/src/main/java/com/yahoo/vespa/feed/perf/SimpleFeeder.java
@@ -185,21 +185,34 @@ public class SimpleFeeder implements ReplyHandler {
             }
         }
         public void send(FeedOperation op) {
-            if (op.getType() == FeedOperation.Type.DOCUMENT) {
-                if (!isFirst) {
-                    try {
-                        outputStream.write(',');
-                        outputStream.write('\n');
-                    } catch (IOException e) {
-                        failure.set(e);
-                    }
-                } else {
-                    isFirst = false;
+            switch (op.getType()) {
+                case DOCUMENT -> {
+                    addCommaAndNewline();
+                    writer.write(op.getDocumentPut().getDocument());
                 }
-                writer.write(op.getDocumentPut().getDocument());
+                case REMOVE -> {
+                    addCommaAndNewline();
+                    writer.write(op.getDocumentRemove());
+                }
+                default -> { /* TODO: No more operations supported yet */ }
             }
             numReplies.incrementAndGet();
         }
+
+        private void addCommaAndNewline() {
+            if (! isFirst) {
+                try {
+                    outputStream.write(',');
+                    outputStream.write('\n');
+                } catch (IOException e) {
+                    failure.set(e);
+                }
+            }
+            else {
+                isFirst = false;
+            }
+        }
+
         public void close() throws Exception {
             outputStream.write('\n');
             outputStream.write(']');
-- 
cgit v1.2.3