diff options
author | Harald Musum <musum@yahooinc.com> | 2024-03-11 14:38:08 +0100 |
---|---|---|
committer | Harald Musum <musum@yahooinc.com> | 2024-03-11 14:38:08 +0100 |
commit | 41a5368c200e5417c10d5f9bba9b21e7f18c00e6 (patch) | |
tree | 977c82a65cfc1dbef3026a9394eab1c3441a5b11 /vespaclient-java/src/main/java | |
parent | 8110c6d8a31620119ab0d255a1ecdfaa7daa06b7 (diff) |
Support serializing document remove operation to json format
Diffstat (limited to 'vespaclient-java/src/main/java')
-rw-r--r-- | vespaclient-java/src/main/java/com/yahoo/vespa/feed/perf/SimpleFeeder.java | 35 |
1 files changed, 24 insertions, 11 deletions
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(']'); |