risingwave_expr_macro/
types.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
15//! This module provides utility functions for SQL data type conversion and manipulation.
16
17//  name        data type   array type          owned type      ref type            primitive
18const TYPE_MATRIX: &str = "
19    boolean     Boolean     BoolArray           bool            bool                _
20    int2        Int16       I16Array            i16             i16                 y
21    int4        Int32       I32Array            i32             i32                 y
22    int8        Int64       I64Array            i64             i64                 y
23    int256      Int256      Int256Array         Int256          Int256Ref<'_>       _
24    float4      Float32     F32Array            F32             F32                 y
25    float8      Float64     F64Array            F64             F64                 y
26    decimal     Decimal     DecimalArray        Decimal         Decimal             y
27    serial      Serial      SerialArray         Serial          Serial              y
28    date        Date        DateArray           Date            Date                y
29    time        Time        TimeArray           Time            Time                y
30    timestamp   Timestamp   TimestampArray      Timestamp       Timestamp           y
31    timestamptz Timestamptz TimestamptzArray    Timestamptz     Timestamptz         y
32    interval    Interval    IntervalArray       Interval        Interval            y
33    varchar     Varchar     Utf8Array           Box<str>        &str                _
34    bytea       Bytea       BytesArray          Box<[u8]>       &[u8]               _
35    jsonb       Jsonb       JsonbArray          JsonbVal        JsonbRef<'_>        _
36    anyarray    List        ListArray           ListValue       ListRef<'_>         _
37    struct      Struct      StructArray         StructValue     StructRef<'_>       _
38    anymap      Map         MapArray            MapValue        MapRef<'_>          _
39    any         ???         ArrayImpl           ScalarImpl      ScalarRefImpl<'_>   _
40";
41
42/// Maps a data type to its corresponding data type name.
43pub fn data_type(ty: &str) -> &str {
44    lookup_matrix(ty, 1)
45}
46
47/// Maps a data type to its corresponding array type name.
48pub fn array_type(ty: &str) -> &str {
49    lookup_matrix(ty, 2)
50}
51
52/// Maps a data type to its corresponding `Scalar` type name.
53pub fn owned_type(ty: &str) -> &str {
54    lookup_matrix(ty, 3)
55}
56
57/// Maps a data type to its corresponding `ScalarRef` type name.
58pub fn ref_type(ty: &str) -> &str {
59    lookup_matrix(ty, 4)
60}
61
62/// Checks if a data type is primitive.
63pub fn is_primitive(ty: &str) -> bool {
64    lookup_matrix(ty, 5) == "y"
65}
66
67fn lookup_matrix(mut ty: &str, idx: usize) -> &str {
68    if ty.ends_with("[]") {
69        ty = "anyarray";
70    } else if ty.starts_with("struct") {
71        ty = "struct";
72    } else if ty == "void" {
73        // XXX: we don't support void type yet.
74        //      replace it with int for now.
75        ty = "int4";
76    }
77    let s = TYPE_MATRIX.trim().lines().find_map(|line| {
78        let mut parts = line.split_whitespace();
79        if parts.next() == Some(ty) {
80            Some(parts.nth(idx - 1).unwrap())
81        } else {
82            None
83        }
84    });
85    s.unwrap_or_else(|| panic!("failed to lookup type matrix: unknown type: {}", ty))
86}
87
88/// Expands a type wildcard string into a list of concrete types.
89pub fn expand_type_wildcard(ty: &str) -> Vec<&str> {
90    match ty {
91        "*" => TYPE_MATRIX
92            .trim()
93            .lines()
94            .map(|l| l.split_whitespace().next().unwrap())
95            .filter(|l| *l != "any")
96            .collect(),
97        "*int" => vec!["int2", "int4", "int8"],
98        "*float" => vec!["float4", "float8"],
99        _ => vec![ty],
100    }
101}
102
103/// Computes the minimal compatible type between a pair of data types.
104///
105/// This function is used to determine the `auto` type.
106pub fn min_compatible_type(types: &[impl AsRef<str>]) -> &str {
107    if types.len() == 1 {
108        return types[0].as_ref();
109    }
110    assert_eq!(types.len(), 2);
111    match (types[0].as_ref(), types[1].as_ref()) {
112        (a, b) if a == b => a,
113
114        ("int2", "int2") => "int2",
115        ("int2", "int4") => "int4",
116        ("int2", "int8") => "int8",
117
118        ("int4", "int2") => "int4",
119        ("int4", "int4") => "int4",
120        ("int4", "int8") => "int8",
121
122        ("int8", "int2") => "int8",
123        ("int8", "int4") => "int8",
124        ("int8", "int8") => "int8",
125
126        ("int2", "int256") => "int256",
127        ("int4", "int256") => "int256",
128        ("int8", "int256") => "int256",
129        ("int256", "int2") => "int256",
130        ("int256", "int4") => "int256",
131        ("int256", "int8") => "int256",
132        ("int256", "float8") => "float8",
133        ("float8", "int256") => "float8",
134
135        ("float4", "float4") => "float4",
136        ("float4", "float8") => "float8",
137
138        ("float8", "float4") => "float8",
139        ("float8", "float8") => "float8",
140
141        ("decimal", "decimal") => "decimal",
142
143        ("date", "timestamp") => "timestamp",
144        ("timestamp", "date") => "timestamp",
145        ("time", "interval") => "interval",
146        ("interval", "time") => "interval",
147
148        (a, b) => panic!("unknown minimal compatible type for {a:?} and {b:?}"),
149    }
150}
risingwave_expr_macro/types.rs

risingwave_expr_macro/
types.rs
