risingwave_expr_impl/window_function/
mod.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use itertools::Itertools;
16use risingwave_expr::window_function::{
17    BoxedWindowState, WINDOW_STATE_BUILDERS, WindowFuncCall, WindowFuncKind,
18};
19use risingwave_expr::{ExprError, Result};
20
21mod aggregate;
22mod buffer;
23mod range_utils;
24mod rank;
25
26#[linkme::distributed_slice(WINDOW_STATE_BUILDERS)]
27fn create_window_state_impl(call: &WindowFuncCall) -> Result<BoxedWindowState> {
28    assert!(call.frame.bounds.validate().is_ok());
29
30    use WindowFuncKind::*;
31    Ok(match &call.kind {
32        RowNumber => Box::new(rank::RankState::<rank::RowNumber>::new(call)),
33        Rank => Box::new(rank::RankState::<rank::Rank>::new(call)),
34        DenseRank => Box::new(rank::RankState::<rank::DenseRank>::new(call)),
35        Aggregate(_) => aggregate::new(call)?,
36        kind => {
37            return Err(ExprError::UnsupportedFunction(format!(
38                "{}({}) -> {}",
39                kind,
40                call.args.arg_types().iter().format(", "),
41                &call.return_type,
42            )));
43        }
44    })
45}
risingwave_expr_impl/window_function/mod.rs

risingwave_expr_impl/window_function/
mod.rs
