Maintain backward compatibility for RescoreContext public API

girish jeyakumar · girish jeyakumar · commit 67c3ecb49017 · 2025-07-08T12:47:10.000-07:00
- Restore getQueries() method for backward compatibility
- Add getParsedQueries() method for new functionality

Signed-off-by: Girish Jeyakumar &lt;girishjeyakumar@dropbox.com&gt;
diff --git a/server/src/main/java/org/opensearch/search/rescore/QueryRescorer.java b/server/src/main/java/org/opensearch/search/rescore/QueryRescorer.java
@@ -34,6 +34,7 @@
 
 import org.apache.lucene.search.Explanation;
 import org.apache.lucene.search.IndexSearcher;
+import org.apache.lucene.search.Query;
 import org.apache.lucene.search.ScoreDoc;
 import org.apache.lucene.search.TopDocs;
 import org.opensearch.index.query.ParsedQuery;
@@ -209,6 +210,11 @@ public ParsedQuery parsedQuery() {
             return parsedQuery;
         }
 
+        @Override
+        public List<Query> getQueries() {
+            return parsedQuery != null ? Collections.singletonList(parsedQuery.query()) : Collections.emptyList();
+        }
+
         @Override
         public List<ParsedQuery> getParsedQueries() {
             return parsedQuery != null ? Collections.singletonList(parsedQuery) : Collections.emptyList();
diff --git a/server/src/main/java/org/opensearch/search/rescore/RescoreContext.java b/server/src/main/java/org/opensearch/search/rescore/RescoreContext.java
@@ -32,6 +32,7 @@
 
 package org.opensearch.search.rescore;
 
+import org.apache.lucene.search.Query;
 import org.opensearch.common.annotation.PublicApi;
 import org.opensearch.index.query.ParsedQuery;
 
@@ -87,6 +88,13 @@ public Set<Integer> getRescoredDocs() {
         return rescoredDocs;
     }
 
+    /**
+     * Returns queries associated with the rescorer
+     */
+    public List<Query> getQueries() {
+        return Collections.emptyList();
+    }
+
     /**
      * Returns parsed queries associated with the rescorer
      */
