reifydb-evaluate 0.10.0

Runtime expression evaluator and variable environment for ReifyDB
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2026 ReifyDB

use std::sync::Arc;

use arrow_array::RecordBatch;
use reifydb_core::{
	error::diagnostic::query::column_not_found, expression::AccessObjectExpression,
	interface::identifier::ColumnObject,
};
use reifydb_value::{error, fragment::Fragment, value::system_columns::is_system_field};

use crate::Result;

pub(crate) fn access_position(batch: &RecordBatch, expr: &AccessObjectExpression) -> Result<usize> {
	let source = match &expr.column.object {
		ColumnObject::Qualified {
			name,
			..
		} => name,
		ColumnObject::Alias(alias) => alias,
	};
	let column = expr.column.name.text().to_string();

	let qualified_name = format!("{}.{}", source.text(), &column);

	let matching_col =
		batch.schema_ref().fields().iter().enumerate().filter(|(_, field)| !is_system_field(field)).find(
			|(_, field)| {
				if field.name() == &qualified_name {
					return true;
				}

				if matches!(&expr.column.object, ColumnObject::Qualified { .. })
					&& field.name() == &column
				{
					return !field.name().contains('.');
				}

				false
			},
		);

	if let Some((index, _)) = matching_col {
		Ok(index)
	} else {
		Err(error!(column_not_found(Fragment::Statement {
			column: expr.column.name.column(),
			line: expr.column.name.line(),
			text: Arc::from(format!("{}.{}", source.text(), &column)),
		})))
	}
}