cleanups

src/main/java/org/elasticsearch/action/update/TransportUpdateAction.java

@@ -239,7 +239,7 @@ public class TransportUpdateAction extends TransportInstanceSingleOperationActio
             if (indexRequest.parent() != null) {
                 parent = indexRequest.parent();
             }
-            updateSource(updatedSourceAsMap, indexRequest.underlyingSourceAsMap());
+            XContentHelper.update(updatedSourceAsMap, indexRequest.underlyingSourceAsMap());
         } else {
             Map<String, Object> ctx = new HashMap<String, Object>(2);
             ctx.put("_source", sourceAndContent.v2());
@@ -388,24 +388,4 @@ public class TransportUpdateAction extends TransportInstanceSingleOperationActio
 
         return new GetResult(request.index(), request.type(), request.id(), version, true, sourceRequested ? sourceAsBytes : null, fields);
     }
-    
-    /**
-     * Updates the source with the specified changes.  Maps are updated recursively.
-     */
-    private void updateSource(Map<String, Object> source, Map<String, Object> changes) {
-        for (Map.Entry<String, Object> changesEntry : changes.entrySet()) {
-            if (!source.containsKey(changesEntry.getKey())) {
-                // safe to copy, change does not exist in source
-                source.put(changesEntry.getKey(), changesEntry.getValue());
-            } else {
-                if (source.get(changesEntry.getKey()) instanceof Map && changesEntry.getValue() instanceof Map) {
-                    // recursive merge maps
-                    updateSource((Map<String, Object>) source.get(changesEntry.getKey()), (Map<String, Object>) changesEntry.getValue());
-                } else {
-                    // update the field
-                    source.put(changesEntry.getKey(), changesEntry.getValue());
-                }
-            }
-        }  
-    }
 }

src/main/java/org/elasticsearch/action/update/UpdateRequest.java

@@ -529,20 +529,12 @@ public class UpdateRequest extends InstanceShardOperationRequest {
         consistencyLevel = WriteConsistencyLevel.fromId(in.readByte());
         type = in.readUTF();
         id = in.readUTF();
-        if (in.readBoolean()) {
+        routing = in.readOptionalUTF();
-            routing = in.readUTF();
+        script = in.readOptionalUTF();
-        }
+        scriptLang = in.readOptionalUTF();
-        if (in.readBoolean()) {
-            script = in.readUTF();
-        }
-        if (in.readBoolean()) {
-            scriptLang = in.readUTF();
-        }
         scriptParams = in.readMap();
         retryOnConflict = in.readVInt();
-        if (in.readBoolean()) {
+        percolate = in.readOptionalUTF();
-            percolate = in.readUTF();
-        }
         refresh = in.readBoolean();
         if (in.readBoolean()) {
             doc = new IndexRequest();
@@ -568,32 +560,12 @@ public class UpdateRequest extends InstanceShardOperationRequest {
         out.writeByte(consistencyLevel.id());
         out.writeUTF(type);
         out.writeUTF(id);
-        if (routing == null) {
+        out.writeOptionalUTF(routing);
-            out.writeBoolean(false);
+        out.writeOptionalUTF(script);
-        } else {
+        out.writeOptionalUTF(scriptLang);
-            out.writeBoolean(true);
-            out.writeUTF(routing);
-        }
-        if (script == null) {
-            out.writeBoolean(false);
-        } else {
-            out.writeBoolean(true);
-            out.writeUTF(script);
-        }
-        if (scriptLang == null) {
-            out.writeBoolean(false);
-        } else {
-            out.writeBoolean(true);
-            out.writeUTF(scriptLang);
-        }
         out.writeMap(scriptParams);
         out.writeVInt(retryOnConflict);
-        if (percolate == null) {
+        out.writeOptionalUTF(percolate);
-            out.writeBoolean(false);
-        } else {
-            out.writeBoolean(true);
-            out.writeUTF(percolate);
-        }
         out.writeBoolean(refresh);
         if (doc == null) {
             out.writeBoolean(false);

src/main/java/org/elasticsearch/common/xcontent/XContentHelper.java

@@ -37,6 +37,7 @@ import java.util.Map;
 /**
  *
  */
+@SuppressWarnings("unchecked")
 public class XContentHelper {
 
     public static XContentParser createParser(byte[] data, int offset, int length) throws IOException {
@@ -105,6 +106,27 @@ public class XContentHelper {
         }
     }
 
+    /**
+     * Updates the provided changes into the source. If the key exists in the changes, it overrides the one in source
+     * unless both are Maps, in which case it recuersively updated it.
+     */
+    public static void update(Map<String, Object> source, Map<String, Object> changes) {
+        for (Map.Entry<String, Object> changesEntry : changes.entrySet()) {
+            if (!source.containsKey(changesEntry.getKey())) {
+                // safe to copy, change does not exist in source
+                source.put(changesEntry.getKey(), changesEntry.getValue());
+            } else {
+                if (source.get(changesEntry.getKey()) instanceof Map && changesEntry.getValue() instanceof Map) {
+                    // recursive merge maps
+                    update((Map<String, Object>) source.get(changesEntry.getKey()), (Map<String, Object>) changesEntry.getValue());
+                } else {
+                    // update the field
+                    source.put(changesEntry.getKey(), changesEntry.getValue());
+                }
+            }
+        }
+    }
+
     /**
      * Merges the defaults provided as the second parameter into the content of the first. Only does recursive merge
      * for inner maps.