fix

Signed-off-by: Weihao Li <18110526956@163.com>

Author: Wei-hao-Li

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/relational/planner/optimizations/SortElimination.java

@@ -121,7 +121,9 @@ public PlanNode visitSort(SortNode node, Context context) {
           && orderingScheme.getOrderBy().get(0).getName().equals(context.getTimeColumnName())) {
         return child;
       }
-      return node.replaceChildren(Collections.singletonList(child));
+      return context.canEliminateSort() && node.isOrderByAllIdsAndTime()
+          ? child
+          : node.replaceChildren(Collections.singletonList(child));
     }
 
     @Override
@@ -196,8 +198,6 @@ private static class Context {
 
     private String timeColumnName = null;
 
-    private boolean sortEliminated = false;
-
     Context() {}
 
     public void addDeviceEntrySize(int deviceEntrySize) {
@@ -223,13 +223,5 @@ public String getTimeColumnName() {
     public void setTimeColumnName(String timeColumnName) {
       this.timeColumnName = timeColumnName;
     }
-
-    public boolean isSortEliminated() {
-      return sortEliminated;
-    }
-
-    public void markSortEliminated() {
-      this.sortEliminated = true;
-    }
   }
 }