Fix comment

apache · Dec 6, 2022 · edb06d8 · edb06d8
1 parent 01cea90
commit edb06d8
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/sql/src/main/java/org/apache/druid/sql/calcite/rel/PartialDruidQuery.java b/sql/src/main/java/org/apache/druid/sql/calcite/rel/PartialDruidQuery.java
@@ -388,14 +388,13 @@ public boolean canAccept(final Stage stage)
     } else if (stage.compareTo(currentStage) <= 0) {
       // Cannot go backwards.
       return false;
-    } else // Cannot do post-sort stages without a sort.
-      // Looks good.
-      if (stage.compareTo(Stage.AGGREGATE) > 0 && stage.compareTo(Stage.SORT) < 0 && aggregate == null) {
-        // Cannot do post-aggregation stages without an aggregation.
-        return false;
-      } else {
-        return stage.compareTo(Stage.SORT) <= 0 || sort != null;
-      }
+    } else if (stage.compareTo(Stage.AGGREGATE) > 0 && stage.compareTo(Stage.SORT) < 0 && aggregate == null) {
+      // Cannot do post-aggregation stages without an aggregation.
+      return false;
+    } else {
+      // If we are after the SORT phase, make sure we have a sort...
+      return stage.compareTo(Stage.SORT) <= 0 || sort != null;
+    }
   }
 
   /**