risingwave_common/array/
interval_array.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::{PrimitiveArray, PrimitiveArrayBuilder};
16use crate::types::Interval;
17
18pub type IntervalArray = PrimitiveArray<Interval>;
19pub type IntervalArrayBuilder = PrimitiveArrayBuilder<Interval>;
20
21#[cfg(test)]
22mod tests {
23    use super::IntervalArray;
24    use crate::array::interval_array::{Interval, IntervalArrayBuilder};
25    use crate::array::{Array, ArrayBuilder};
26    use crate::types::test_utils::IntervalTestExt;
27
28    #[test]
29    fn test_interval_array() {
30        let cardinality = 5;
31        let mut array_builder = IntervalArrayBuilder::new(cardinality);
32        for _ in 0..cardinality {
33            let v = Interval::from_ymd(1, 0, 0);
34            array_builder.append(Some(v));
35        }
36        let ret_arr = array_builder.finish();
37        for v in ret_arr.iter().flatten() {
38            assert_eq!(v.months(), 12);
39            assert_eq!(v.days(), 0);
40        }
41        let ret_arr = IntervalArray::from_iter([Some(Interval::from_ymd(1, 0, 0)), None]);
42        let v = ret_arr.value_at(0).unwrap();
43        assert_eq!(v.months(), 12);
44        assert_eq!(v.days(), 0);
45        let v = ret_arr.value_at(1);
46        assert_eq!(v, None);
47        let v = unsafe { ret_arr.value_at_unchecked(0).unwrap() };
48        assert_eq!(v.months(), 12);
49        assert_eq!(v.days(), 0);
50    }
51}