From b2ed86fb55a6fad998b9a42d30563720eb3e210a Mon Sep 17 00:00:00 2001 From: Jiayu Liu Date: Tue, 22 Jun 2021 08:52:31 +0800 Subject: [PATCH] update sort partition points --- datafusion/src/physical_plan/windows.rs | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/datafusion/src/physical_plan/windows.rs b/datafusion/src/physical_plan/windows.rs index a214ef17a9f83..bb255511371e7 100644 --- 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] { 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] }