Skip to main content

reifydb_function/is/
none.rs

1// SPDX-License-Identifier: AGPL-3.0-or-later
2// Copyright (c) 2025 ReifyDB
3
4use reifydb_core::value::column::data::ColumnData;
5use reifydb_type::value::r#type::Type;
6
7use crate::{ScalarFunction, ScalarFunctionContext, error::ScalarFunctionError};
8
9pub struct IsNone;
10
11impl IsNone {
12	pub fn new() -> Self {
13		Self
14	}
15}
16
17impl ScalarFunction for IsNone {
18	fn scalar(&self, ctx: ScalarFunctionContext) -> crate::error::ScalarFunctionResult<ColumnData> {
19		let columns = ctx.columns;
20		let row_count = ctx.row_count;
21
22		if columns.len() != 1 {
23			return Err(ScalarFunctionError::ArityMismatch {
24				function: ctx.fragment.clone(),
25				expected: 1,
26				actual: columns.len(),
27			});
28		}
29
30		let column = columns.get(0).unwrap();
31		let data: Vec<bool> = (0..row_count).map(|i| !column.data().is_defined(i)).collect();
32
33		Ok(ColumnData::bool(data))
34	}
35
36	fn return_type(&self, _input_types: &[Type]) -> Type {
37		Type::Boolean
38	}
39}