small fix

Signed-off-by: bowenlan-amzn <bowenlan23@gmail.com>
opensearch-project · Jul 30, 2024 · b9bf2d4 · b9bf2d4
1 parent 3cd735e
commit b9bf2d4
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/server/src/main/java/org/opensearch/index/query/TermsQueryBuilder.java b/server/src/main/java/org/opensearch/index/query/TermsQueryBuilder.java
@@ -530,10 +530,13 @@ private void fetch(TermsLookup termsLookup, Client client, ActionListener<List<O
         }
         client.get(getRequest, ActionListener.delegateFailure(actionListener, (delegatedListener, getResponse) -> {
             List<Object> terms = new ArrayList<>();
-            terms.addAll(getResponse.getField(termsLookup.path()).getValues());
-            if (getResponse.isSourceEmpty() == false) { // extract terms only if the doc source exists
-                List<Object> extractedValues = XContentMapValues.extractRawValues(termsLookup.path(), getResponse.getSourceAsMap());
-                terms.addAll(extractedValues);
+            if (termsLookup.store()) {
+                terms.addAll(getResponse.getField(termsLookup.path()).getValues());
+            } else {
+                if (getResponse.isSourceEmpty() == false) { // extract terms only if the doc source exists
+                    List<Object> extractedValues = XContentMapValues.extractRawValues(termsLookup.path(), getResponse.getSourceAsMap());
+                    terms.addAll(extractedValues);
+                }
             }
             delegatedListener.onResponse(terms);
         }));