common_function/scalars/geo/
helpers.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 datafusion::arrow::array::{ArrayRef, ArrowPrimitiveType};
16use datafusion::arrow::compute;
17
18macro_rules! ensure_and_coerce {
19    ($compare:expr, $coerce:expr) => {{
20        if !$compare {
21            return Err(datafusion_common::DataFusionError::Execution(
22                "argument out of valid range".to_string(),
23            ));
24        }
25        Ok(Some($coerce))
26    }};
27}
28
29pub(crate) use ensure_and_coerce;
30
31pub(crate) fn cast<T: ArrowPrimitiveType>(array: &ArrayRef) -> datafusion_common::Result<ArrayRef> {
32    let x = compute::cast_with_options(
33        array.as_ref(),
34        &T::DATA_TYPE,
35        &compute::CastOptions {
36            safe: false,
37            ..Default::default()
38        },
39    )?;
40    Ok(x)
41}