Add additional sorts to other locations

Signed-off-by: Simeon Widdis <[email protected]>
opensearch-project · Oct 10, 2024 · 3ba5a7e · 3ba5a7e
1 parent a8af162
commit 3ba5a7e
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 0 deletions.
diff --git a/legacy/src/main/java/org/opensearch/sql/legacy/executor/ElasticHitsExecutor.java b/legacy/src/main/java/org/opensearch/sql/legacy/executor/ElasticHitsExecutor.java
@@ -19,6 +19,7 @@
 import org.opensearch.common.unit.TimeValue;
 import org.opensearch.search.SearchHits;
 import org.opensearch.search.builder.PointInTimeBuilder;
+import org.opensearch.search.sort.SortOrder;
 import org.opensearch.sql.legacy.domain.Select;
 import org.opensearch.sql.legacy.esdomain.LocalClusterState;
 import org.opensearch.sql.legacy.exception.SqlParseException;
@@ -70,6 +71,7 @@ public SearchResponse getResponseWithHits(
       boolean ordered = select.isOrderdSelect();
       if (!ordered) {
         request.addSort(DOC_FIELD_NAME, ASC);
+        request.addSort("_id", SortOrder.ASC);
       }
       // Set PIT
       request.setPointInTime(new PointInTimeBuilder(pit.getPitId()));

diff --git a/legacy/src/main/java/org/opensearch/sql/legacy/executor/join/ElasticUtils.java b/legacy/src/main/java/org/opensearch/sql/legacy/executor/join/ElasticUtils.java
@@ -39,6 +39,7 @@ public static SearchResponse scrollOneTimeWithHits(
     boolean ordered = originalSelect.isOrderdSelect();
     if (!ordered) {
       scrollRequest.addSort(FieldSortBuilder.DOC_FIELD_NAME, SortOrder.ASC);
+      scrollRequest.addSort("_id", SortOrder.ASC);
     }
     SearchResponse responseWithHits = scrollRequest.get();
     // on ordered select - not using SCAN , elastic returns hits on first scroll

diff --git a/legacy/src/main/java/org/opensearch/sql/legacy/query/DefaultQueryAction.java b/legacy/src/main/java/org/opensearch/sql/legacy/query/DefaultQueryAction.java
@@ -110,6 +110,7 @@ public void checkAndSetScroll() {
         boolean ordered = select.isOrderdSelect();
         if (!ordered) {
           request.addSort(FieldSortBuilder.DOC_FIELD_NAME, SortOrder.ASC);
+          request.addSort("_id", SortOrder.ASC);
         }
         // Request also requires PointInTime, but we should create pit while execution.
       } else {

diff --git a/...n/java/org/opensearch/sql/legacy/query/planner/physical/node/pointInTime/PointInTime.java b/...n/java/org/opensearch/sql/legacy/query/planner/physical/node/pointInTime/PointInTime.java
@@ -42,6 +42,7 @@ protected void loadFirstBatch() {
         request
             .getRequestBuilder()
             .addSort(FieldSortBuilder.DOC_FIELD_NAME, SortOrder.ASC)
+            .addSort("_id", SortOrder.ASC)
             .setSize(pageSize)
             .setTimeout(TimeValue.timeValueSeconds(timeout))
             .setPointInTime(new PointInTimeBuilder(pitId))

diff --git a/...cy/src/main/java/org/opensearch/sql/legacy/query/planner/physical/node/scroll/Scroll.java b/...cy/src/main/java/org/opensearch/sql/legacy/query/planner/physical/node/scroll/Scroll.java
@@ -40,6 +40,7 @@ protected void loadFirstBatch() {
         request
             .getRequestBuilder()
             .addSort(FieldSortBuilder.DOC_FIELD_NAME, SortOrder.ASC)
+            .addSort("_id", SortOrder.ASC)
             .setSize(pageSize)
             .setScroll(TimeValue.timeValueSeconds(timeout))
             .get();