risingwave_frontend/utils/
rewrite_index.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 super::ColIndexMapping;
16use crate::expr::{ExprImpl, ExprRewriter, InputRef};
17
18pub struct IndexRewriter {
19    mapping: ColIndexMapping,
20}
21
22impl IndexRewriter {
23    pub fn new(mapping: ColIndexMapping) -> Self {
24        Self { mapping }
25    }
26}
27
28impl ExprRewriter for IndexRewriter {
29    fn rewrite_input_ref(&mut self, mut input_ref: InputRef) -> ExprImpl {
30        input_ref.index = self.mapping.map(input_ref.index);
31
32        ExprImpl::InputRef(Box::new(input_ref))
33    }
34}

risingwave_frontend/utils/rewrite_index.rs

risingwave_frontend/utils/
rewrite_index.rs