feat: support RANGE frame for window function. (#10911)

Author: RinChanNOWWW

Cargo.lock

@@ -491,7 +491,7 @@ pub struct WindowFrame {
     pub end_bound: WindowFrameBound,
 }
 
-#[derive(Debug, Clone, PartialEq, Eq, Hash)]
+#[derive(Debug, Clone, PartialEq, Eq, Hash, EnumAsInner)]
 pub enum WindowFrameUnits {
     Rows,
     Range,

src/query/ast/src/ast/expr.rs

@@ -618,20 +618,20 @@ pub fn group_by_items(i: Input) -> IResult<GroupBy> {
 pub fn window_frame_bound(i: Input) -> IResult<WindowFrameBound> {
     alt((
         value(WindowFrameBound::CurrentRow, rule! { CURRENT ~ ROW }),
-        map(rule! { #subexpr(0) ~ PRECEDING }, |(expr, _)| {
-            WindowFrameBound::Preceding(Some(Box::new(expr)))
-        }),
         value(
             WindowFrameBound::Preceding(None),
             rule! { UNBOUNDED ~ PRECEDING },
         ),
-        map(rule! { #subexpr(0) ~ FOLLOWING }, |(expr, _)| {
-            WindowFrameBound::Following(Some(Box::new(expr)))
+        map(rule! { #subexpr(0) ~ PRECEDING }, |(expr, _)| {
+            WindowFrameBound::Preceding(Some(Box::new(expr)))
         }),
         value(
             WindowFrameBound::Following(None),
             rule! { UNBOUNDED ~ FOLLOWING },
         ),
+        map(rule! { #subexpr(0) ~ FOLLOWING }, |(expr, _)| {
+            WindowFrameBound::Following(Some(Box::new(expr)))
+        }),
     ))(i)
 }
 

src/query/ast/src/parser/query.rs

@@ -3350,22 +3350,7 @@ FunctionCall {
                     WindowFrame {
                         units: Rows,
                         start_bound: Preceding(
-                            Some(
-                                ColumnRef {
-                                    span: Some(
-                                        76..85,
-                                    ),
-                                    database: None,
-                                    table: None,
-                                    column: Identifier {
-                                        name: "UNBOUNDED",
-                                        quote: None,
-                                        span: Some(
-                                            76..85,
-                                        ),
-                                    },
-                                },
-                            ),
+                            None,
                         ),
                         end_bound: CurrentRow,
                     },
@@ -3593,22 +3578,7 @@ FunctionCall {
                     WindowFrame {
                         units: Rows,
                         start_bound: Preceding(
-                            Some(
-                                ColumnRef {
-                                    span: Some(
-                                        38..47,
-                                    ),
-                                    database: None,
-                                    table: None,
-                                    column: Identifier {
-                                        name: "UNBOUNDED",
-                                        quote: None,
-                                        span: Some(
-                                            38..47,
-                                        ),
-                                    },
-                                },
-                            ),
+                            None,
                         ),
                         end_bound: CurrentRow,
                     },

src/query/ast/tests/it/testdata/expr.txt

@@ -738,7 +738,11 @@ impl PipelineBuilder {
             .iter()
             .map(|o| {
                 let offset = input_schema.index_of(&o.order_by.to_string())?;
-                Ok(offset)
+                Ok(SortColumnDescription {
+                    offset,
+                    asc: o.asc,
+                    nulls_first: o.nulls_first,
+                })
             })
             .collect::<Result<Vec<_>>>()?;
 
@@ -754,13 +758,7 @@ impl PipelineBuilder {
                 })
             }
 
-            for (order_desc, offset) in window.order_by.iter().zip(order_by.iter()) {
-                sort_desc.push(SortColumnDescription {
-                    offset: *offset,
-                    asc: order_desc.asc,
-                    nulls_first: order_desc.nulls_first,
-                })
-            }
+            sort_desc.extend(order_by.clone());
 
             self.build_sort_pipeline(input_schema.clone(), sort_desc, window.plan_id, None)?;
         }