reifydb_function/meta/
type.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::{constraint::bytes::MaxBytes, container::utf8::Utf8Container, r#type::Type as ValueType};
6
7use crate::{ScalarFunction, ScalarFunctionContext, error::ScalarFunctionError};
8
9pub struct Type;
10
11impl Type {
12	pub fn new() -> Self {
13		Self
14	}
15}
16
17impl ScalarFunction for Type {
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 col_type = column.data().get_type();
32		let type_name = col_type.to_string();
33
34		let result_data: Vec<String> = vec![type_name; row_count];
35
36		Ok(ColumnData::Utf8 {
37			container: Utf8Container::new(result_data),
38			max_bytes: MaxBytes::MAX,
39		})
40	}
41
42	fn return_type(&self, _input_types: &[ValueType]) -> ValueType {
43		ValueType::Utf8
44	}
45}
reifydb_function/meta/type.rs

reifydb_function/meta/
type.rs
