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::{
8	ScalarFunction, ScalarFunctionContext,
9	error::{ScalarFunctionError, ScalarFunctionResult},
10};
11
12pub struct Type;
13
14impl Type {
15	pub fn new() -> Self {
16		Self
17	}
18}
19
20impl ScalarFunction for Type {
21	fn scalar(&self, ctx: ScalarFunctionContext) -> ScalarFunctionResult<ColumnData> {
22		let columns = ctx.columns;
23		let row_count = ctx.row_count;
24
25		if columns.len() != 1 {
26			return Err(ScalarFunctionError::ArityMismatch {
27				function: ctx.fragment.clone(),
28				expected: 1,
29				actual: columns.len(),
30			});
31		}
32
33		let column = columns.get(0).unwrap();
34		let col_type = column.data().get_type();
35		let type_name = col_type.to_string();
36
37		let result_data: Vec<String> = vec![type_name; row_count];
38
39		Ok(ColumnData::Utf8 {
40			container: Utf8Container::new(result_data),
41			max_bytes: MaxBytes::MAX,
42		})
43	}
44
45	fn return_type(&self, _input_types: &[ValueType]) -> ValueType {
46		ValueType::Utf8
47	}
48}
reifydb_function/meta/type.rs

reifydb_function/meta/
type.rs
