risingwave_frontend/expr/
parameter.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 std::fmt::{Debug, Formatter};
16use std::hash::{Hash, Hasher};
17
18use risingwave_common::types::DataType;
19
20use super::Expr;
21use crate::binder::ParameterTypes;
22
23#[derive(Clone)]
24pub struct Parameter {
25    pub index: u64,
26    param_types: ParameterTypes,
27}
28
29impl Debug for Parameter {
30    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
31        write!(
32            f,
33            "Parameter(index: {}, type: {:?})",
34            self.index,
35            self.param_types.read_type(self.index)
36        )
37    }
38}
39
40impl PartialEq for Parameter {
41    fn eq(&self, other: &Self) -> bool {
42        self.index == other.index
43    }
44}
45
46impl Eq for Parameter {}
47
48impl Hash for Parameter {
49    fn hash<H: Hasher>(&self, state: &mut H) {
50        self.index.hash(state);
51    }
52}
53
54impl Expr for Parameter {
55    fn return_type(&self) -> DataType {
56        self.param_types
57            .read_type(self.index)
58            .unwrap_or(DataType::Varchar)
59    }
60
61    fn to_expr_proto(&self) -> risingwave_pb::expr::ExprNode {
62        unreachable!("Parameter should not be serialized to ExprNode")
63    }
64}
65
66impl Parameter {
67    pub fn new(index: u64, mut param_types: ParameterTypes) -> Self {
68        param_types.record_new_param(index);
69        Self { index, param_types }
70    }
71
72    pub fn has_infer(&self) -> bool {
73        self.param_types.has_infer(self.index)
74    }
75
76    pub fn cast_infer_type(&mut self, data_type: DataType) {
77        self.param_types.record_infer_type(self.index, data_type);
78    }
79}
risingwave_frontend/expr/parameter.rs

risingwave_frontend/expr/
parameter.rs
