Update VespaDocumentOperation.java

1 files changed, 5 insertions, 1 deletions

diff --git a/vespa-hadoop/src/main/java/com/yahoo/vespa/hadoop/pig/VespaDocumentOperation.java b/vespa-hadoop/src/main/java/com/yahoo/vespa/hadoop/pig/VespaDocumentOperation.java
index fa463a77923..1c7edf34fe0 100644
--- a/vespa-hadoop/src/main/java/com/yahoo/vespa/hadoop/pig/VespaDocumentOperation.java
+++ b/vespa-hadoop/src/main/java/com/yahoo/vespa/hadoop/pig/VespaDocumentOperation.java
@@ -69,6 +69,7 @@ public class VespaDocumentOperation extends EvalFunc<String> {
     private static final String SIMPLE_OBJECT_FIELDS = "simple-object-fields";
     private static final String CREATE_TENSOR_FIELDS = "create-tensor-fields";
     private static final String EXCLUDE_FIELDS = "exclude-fields";
+    private static final String TESTSET_CONDITION = "condition";
 
     private static final String PARTIAL_UPDATE_ASSIGN = "assign";
 
@@ -162,7 +163,10 @@ public class VespaDocumentOperation extends EvalFunc<String> {
         if (op == Operation.UPDATE && createIfNonExistent) {
             writeField("create", true, DataType.BOOLEAN, g, properties, schema, op, 0);
         }
-
+        String testSetCondition = properties.getProperty(TESTSET_CONDITION);
+        if (testSetCondition != null) {
+            writeField(TESTSET_CONDITION, testSetCondition, DataType.CHARARRAY, g, properties, schema, op, 0);
+        }
         if (op != Operation.REMOVE) {
             writeField("fields", fields, DataType.MAP, g, properties, schema, op, 0);
         }