Support UPDATE_ROW operation

kdkeck · Apr 14, 2014 · b90c687 · b90c687 · cheald · Apr 17, 2014
1 parent f87d01f
commit b90c687
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/src/main/java/org/elasticsearch/river/mongodb/MongoDBRiver.java b/src/main/java/org/elasticsearch/river/mongodb/MongoDBRiver.java
@@ -97,6 +97,7 @@ public class MongoDBRiver extends AbstractRiverComponent implements River {
     public final static String OPLOG_UPDATE = "o2";
     public final static String OPLOG_OPERATION = "op";
     public final static String OPLOG_UPDATE_OPERATION = "u";
+    public final static String OPLOG_UPDATE_ROW_OPERATION = "ur";
     public final static String OPLOG_INSERT_OPERATION = "i";
     public final static String OPLOG_DELETE_OPERATION = "d";
     public final static String OPLOG_COMMAND_OPERATION = "c";

diff --git a/src/main/java/org/elasticsearch/river/mongodb/Operation.java b/src/main/java/org/elasticsearch/river/mongodb/Operation.java
@@ -21,6 +21,9 @@ public static Operation fromString(String value) {
                     return operation;
                 }
             }
+            if (MongoDBRiver.OPLOG_UPDATE_ROW_OPERATION.equalsIgnoreCase(value)) {
+                return Operation.UPDATE;
+            }
         }
         return Operation.UNKNOWN;
     }

diff --git a/src/main/java/org/elasticsearch/river/mongodb/Slurper.java b/src/main/java/org/elasticsearch/river/mongodb/Slurper.java
@@ -55,6 +55,7 @@ class SlurperException extends Exception {
     private final String gridfsOplogNamespace;
     private final String cmdOplogNamespace;
     private final ImmutableList<String> oplogOperations = ImmutableList.of(MongoDBRiver.OPLOG_DELETE_OPERATION,
+            MongoDBRiver.OPLOG_UPDATE_ROW_OPERATION, // from TokuMX
             MongoDBRiver.OPLOG_UPDATE_OPERATION, MongoDBRiver.OPLOG_INSERT_OPERATION, MongoDBRiver.OPLOG_COMMAND_OPERATION);
     private final Client client;
     private Mongo mongo;