1 files changed, 83 insertions, 0 deletions
diff --git a/vespaclient-container-plugin/src/main/java/com/yahoo/vespa/http/server/DocumentOperationMessageV3.java b/vespaclient-container-plugin/src/main/java/com/yahoo/vespa/http/server/DocumentOperationMessageV3.java
new file mode 100644
index 00000000000..25bf5815907
--- /dev/null
+++ b/vespaclient-container-plugin/src/main/java/com/yahoo/vespa/http/server/DocumentOperationMessageV3.java
@@ -0,0 +1,83 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.vespa.http.server;
+
+import com.yahoo.document.DocumentPut;
+import com.yahoo.document.DocumentRemove;
+import com.yahoo.document.DocumentUpdate;
+import com.yahoo.documentapi.messagebus.protocol.PutDocumentMessage;
+import com.yahoo.documentapi.messagebus.protocol.RemoveDocumentMessage;
+import com.yahoo.documentapi.messagebus.protocol.UpdateDocumentMessage;
+import com.yahoo.jdisc.Metric;
+import com.yahoo.messagebus.Message;
+import com.yahoo.vespaxmlparser.FeedOperation;
+
+/**
+ * Keeps an operation with its message.
+ *
+ * This implementation is based on V2, but the code is restructured.
+ *
+ * @author dybis
+ */
+class DocumentOperationMessageV3 {
+
+    private final String operationId;
+    private final Message message;
+
+    private DocumentOperationMessageV3(String operationId, Message message) {
+        this.operationId = operationId;
+        this.message = message;
+    }
+
+    Message getMessage() {
+        return message;
+    }
+
+    String getOperationId() {
+        return operationId;
+    }
+
+    private static DocumentOperationMessageV3 newUpdateMessage(FeedOperation op, String operationId) {
+        DocumentUpdate update = op.getDocumentUpdate();
+        update.setCondition(op.getCondition());
+        Message msg = new UpdateDocumentMessage(update);
+
+        String id = (operationId == null) ? update.getId().toString() : operationId;
+        return new DocumentOperationMessageV3(id, msg);
+    }
+
+    static DocumentOperationMessageV3 newRemoveMessage(FeedOperation op, String operationId) {
+        DocumentRemove remove = new DocumentRemove(op.getRemove());
+        remove.setCondition(op.getCondition());
+        Message msg = new RemoveDocumentMessage(remove);
+
+        String id = (operationId == null) ? remove.getId().toString() : operationId;
+        return new DocumentOperationMessageV3(id, msg);
+    }
+
+    private static DocumentOperationMessageV3 newPutMessage(FeedOperation op, String operationId) {
+        DocumentPut put = new DocumentPut(op.getDocument());
+        put.setCondition(op.getCondition());
+        Message msg = new PutDocumentMessage(put);
+
+        String id = (operationId == null) ? put.getId().toString() : operationId;
+        return new DocumentOperationMessageV3(id, msg);
+    }
+
+    static DocumentOperationMessageV3 create(FeedOperation operation, String operationId, Metric metric) {
+        switch (operation.getType()) {
+            case DOCUMENT:
+                metric.add(MetricNames.NUM_PUTS, 1, null /*metricContext*/);
+                return newPutMessage(operation, operationId);
+            case REMOVE:
+                metric.add(MetricNames.NUM_REMOVES, 1, null /*metricContext*/);
+                return newRemoveMessage(operation, operationId);
+            case UPDATE:
+                metric.add(MetricNames.NUM_UPDATES, 1, null /*metricContext*/);
+                return newUpdateMessage(operation, operationId);
+            default:
+                // typical end of feed
+                return null;
+        }
+    }
+
+}