tests_fuzz/ir/
insert_expr.rs

1// Copyright 2023 Greptime Team
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, Display};
16
17use datatypes::value::Value;
18
19use crate::ir::{Column, Ident};
20
21pub struct InsertIntoExpr {
22    pub table_name: Ident,
23    pub omit_column_list: bool,
24    pub columns: Vec<Column>,
25    pub values_list: Vec<RowValues>,
26}
27
28impl InsertIntoExpr {
29    /// Returns the timestamp column
30    pub fn timestamp_column(&self) -> Option<Column> {
31        self.columns.iter().find(|c| c.is_time_index()).cloned()
32    }
33
34    /// Returns index of the timestamp column
35    pub fn timestamp_column_idx(&self) -> Option<usize> {
36        self.columns
37            .iter()
38            .enumerate()
39            .find_map(|(idx, c)| if c.is_time_index() { Some(idx) } else { None })
40    }
41
42    /// Returns a vector of columns that are primary keys or time indices.
43    pub fn primary_key_columns(&self) -> Vec<Column> {
44        self.columns
45            .iter()
46            .filter(|c| c.is_primary_key() || c.is_time_index())
47            .cloned()
48            .collect::<Vec<_>>()
49    }
50
51    /// Returns the indices of columns that are primary keys or time indices.
52    pub fn primary_key_column_idx(&self) -> Vec<usize> {
53        self.columns
54            .iter()
55            .enumerate()
56            .filter_map(|(i, c)| {
57                if c.is_primary_key() || c.is_time_index() {
58                    Some(i)
59                } else {
60                    None
61                }
62            })
63            .collect::<Vec<_>>()
64    }
65}
66
67pub type RowValues = Vec<RowValue>;
68
69#[derive(PartialEq, PartialOrd, Clone)]
70pub enum RowValue {
71    Value(Value),
72    Default,
73}
74
75impl RowValue {
76    pub fn cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
77        match (self, other) {
78            (RowValue::Value(Value::Null), RowValue::Value(v2)) => v2.partial_cmp(&Value::Null),
79            (RowValue::Value(v1), RowValue::Value(Value::Null)) => Value::Null.partial_cmp(v1),
80            (RowValue::Value(v1), RowValue::Value(v2)) => v1.partial_cmp(v2),
81            _ => panic!("Invalid comparison: {:?} and {:?}", self, other),
82        }
83    }
84}
85
86impl Display for RowValue {
87    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
88        match self {
89            RowValue::Value(v) => match v {
90                Value::Null => write!(f, "NULL"),
91                v @ (Value::String(_) | Value::Timestamp(_) | Value::Date(_)) => {
92                    write!(f, "'{}'", v)
93                }
94                v => write!(f, "{}", v),
95            },
96            RowValue::Default => write!(f, "DEFAULT"),
97        }
98    }
99}
100
101impl Debug for RowValue {
102    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
103        match self {
104            RowValue::Value(v) => match v {
105                Value::Null => write!(f, "NULL"),
106                v @ (Value::String(_) | Value::Timestamp(_) | Value::Date(_)) => {
107                    write!(f, "'{}'", v)
108                }
109                v => write!(f, "{}", v),
110            },
111            RowValue::Default => write!(f, "DEFAULT"),
112        }
113    }
114}
115
116#[cfg(test)]
117mod tests {
118    use common_time::Timestamp;
119    use datatypes::value::Value;
120
121    use crate::ir::insert_expr::RowValue;
122
123    #[test]
124    fn test_value_cmp() {
125        let time_stampe1 =
126            Value::Timestamp(Timestamp::from_str_utc("-39988-01-31 01:21:12.848697+0000").unwrap());
127        let time_stampe2 =
128            Value::Timestamp(Timestamp::from_str_utc("+12970-09-22 08:40:58.392839+0000").unwrap());
129        let v1 = RowValue::Value(time_stampe1);
130        let v2 = RowValue::Value(time_stampe2);
131        assert_eq!(v1.cmp(&v2), Some(std::cmp::Ordering::Less));
132    }
133}
tests_fuzz/ir/insert_expr.rs

tests_fuzz/ir/
insert_expr.rs
