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}