Fixed deserialization of SearchRequest when _source is an array

Signed-off-by: Thomas Farr <[email protected]>
opensearch-project · Aug 1, 2024 · 714de50 · 714de50
1 parent 4d8ec04
commit 714de50
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -42,6 +42,7 @@ This section is for maintaining a changelog for all breaking changes for the cli
 
 ### Fixed
 - Fixed error when deserializing a normalizer without 'type' ([#1111](https://github.com/opensearch-project/opensearch-java/pull/1111))
+- Fixed deserialization of SearchRequest when `_source` is an array ([#]())
 
 ### Security
 

diff --git a/java-client/src/main/java/org/opensearch/client/json/ObjectDeserializer.java b/java-client/src/main/java/org/opensearch/client/json/ObjectDeserializer.java
@@ -239,7 +239,12 @@ public void shortcutProperty(String name) {
             throw new NoSuchElementException("No deserializer was setup for '" + name + "'");
         }
 
-        acceptedEvents = EventSetObjectAndString;
+        this.acceptedEvents = EventSetObjectAndString.clone();
+
+        if (this.shortcutProperty instanceof FieldObjectDeserializer) {
+            JsonpDeserializer<?> shortcutDeserializer = ((FieldObjectDeserializer<?, ?>) this.shortcutProperty).deserializer;
+            this.acceptedEvents.addAll(shortcutDeserializer.nativeEvents());
+        }
     }
 
     // ----- Object types

diff --git a/java-client/src/main/java/org/opensearch/client/json/UnionDeserializer.java b/java-client/src/main/java/org/opensearch/client/json/UnionDeserializer.java
@@ -179,8 +179,13 @@ public Builder<Union, Kind, Member> addMember(Kind tag, JsonpDeserializer<? exte
                 );
                 objectMembers.add(member);
                 if (od.shortcutProperty() != null) {
-                    // also add it as a string
-                    addMember(Event.VALUE_STRING, tag, member);
+                    // also add it as the shortcut property events
+                    for (Event e : od.acceptedEvents()) {
+                        if (e == Event.START_OBJECT || e == Event.KEY_NAME) {
+                            continue;
+                        }
+                        addMember(e, tag, member);
+                    }
                 }
             } else {
                 UnionDeserializer.SingleMemberHandler<Union, Kind, Member> member = new SingleMemberHandler<>(tag, deserializer);