update sort partition points

apache · Jun 23, 2021 · b2ed86f · b2ed86f
1 parent d55a105
commit b2ed86f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/datafusion/src/physical_plan/windows.rs b/datafusion/src/physical_plan/windows.rs
@@ -209,8 +209,9 @@ fn find_ranges_in_range<'a>(
 ) -> &'a [Range<usize>] {
     let start_idx = sort_partition_points
         .partition_point(|sort_range| sort_range.start < partition_range.start);
-    let end_idx = sort_partition_points
-        .partition_point(|sort_range| sort_range.end <= partition_range.end);
+    let end_idx = start_idx
+        + sort_partition_points[start_idx..]
+            .partition_point(|sort_range| sort_range.end <= partition_range.end);
     &sort_partition_points[start_idx..end_idx]
 }