reifydb_function/json/
array.rs1use 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}