reifydb_function/is/
some.rs1use reifydb_core::value::column::data::ColumnData;
5use reifydb_type::value::r#type::Type;
6
7use crate::{ScalarFunction, ScalarFunctionContext, error::ScalarFunctionError};
8
9pub struct IsSome;
10
11impl IsSome {
12 pub fn new() -> Self {
13 Self
14 }
15}
16
17impl ScalarFunction for IsSome {
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}