risingwave_common/util/
schema_check.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;
16
17use crate::array::{ArrayImpl, ArrayRef};
18use crate::for_all_variants;
19use crate::types::DataType;
20
21/// Check if the schema of `columns` matches the expected `data_types`. Used for debugging.
22pub fn schema_check<'a, T, C>(data_types: T, columns: C) -> Result<(), String>
23where
24    T: IntoIterator<Item = &'a DataType>,
25    C: IntoIterator<Item = &'a ArrayRef>,
26{
27    for (i, pair) in data_types
28        .into_iter()
29        .zip_longest(columns.into_iter().map(|c| &**c))
30        .enumerate()
31    {
32        macro_rules! matches {
33            ($( { $data_type:ident, $variant_name:ident, $suffix_name:ident, $scalar:ty, $scalar_ref:ty, $array:ty, $builder:ty }),*) => {
34                match (pair.as_ref().left(), pair.as_ref().right()) {
35                    $( (Some(DataType::$data_type { .. }), Some(ArrayImpl::$variant_name(_))) => continue, )*
36                    (data_type, array) => {
37                        let array_ident = array.map(|a| a.get_ident());
38                        return Err(format!(
39                            "column type mismatched at position {i}: expected {data_type:?}, found {array_ident:?}"
40                        ));
41                    }
42                }
43            }
44        }
45
46        for_all_variants! { matches }
47    }
48
49    Ok(())
50}
risingwave_common/util/schema_check.rs

risingwave_common/util/
schema_check.rs
