Skip to main content

reifydb_function/json/
array.rs

1// SPDX-License-Identifier: Apache-2.0
2// Copyright (c) 2025 ReifyDB
3
4use reifydb_core::value::column::data::ColumnData;
5use reifydb_type::value::{Value, r#type::Type};
6
7use crate::{ScalarFunction, ScalarFunctionContext, error::ScalarFunctionResult, propagate_options};
8
9pub struct JsonArray;
10
11impl JsonArray {
12	pub fn new() -> Self {
13		Self
14	}
15}
16
17impl ScalarFunction for JsonArray {
18	fn scalar(&self, ctx: ScalarFunctionContext) -> ScalarFunctionResult<ColumnData> {
19		if let Some(result) = propagate_options(self, &ctx) {
20			return result;
21		}
22
23		let columns = ctx.columns;
24		let row_count = ctx.row_count;
25
26		let mut results: Vec<Box<Value>> = Vec::with_capacity(row_count);
27
28		for row in 0..row_count {
29			let mut items = Vec::with_capacity(columns.len());
30			for col in columns.iter() {
31				items.push(col.data().get_value(row));
32			}
33			results.push(Box::new(Value::List(items)));
34		}
35
36		Ok(ColumnData::any(results))
37	}
38
39	fn return_type(&self, _input_types: &[Type]) -> Type {
40		Type::Any
41	}
42}