reifydb-engine 0.4.11

Query execution and processing engine 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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

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

use reifydb_core::{
	error::diagnostic::engine,
	value::column::{Column, data::ColumnData},
};
use reifydb_rql::expression::ParameterExpression;
use reifydb_type::{
	error,
	fragment::Fragment,
	value::{Value, r#type::Type},
};

use crate::{Result, expression::context::EvalContext};

pub(crate) fn parameter_lookup(ctx: &EvalContext, expr: &ParameterExpression) -> Result<Column> {
	let value = match expr {
		ParameterExpression::Positional {
			fragment,
		} => {
			let index = fragment.text()[1..]
				.parse::<usize>()
				.map_err(|_| error!(engine::invalid_parameter_reference(fragment.clone())))?;

			ctx.params
				.get_positional(index - 1)
				.ok_or_else(|| error!(engine::parameter_not_found(fragment.clone())))?
		}
		ParameterExpression::Named {
			fragment,
		} => {
			let name = &fragment.text()[1..];

			ctx.params
				.get_named(name)
				.ok_or_else(|| error!(engine::parameter_not_found(fragment.clone())))?
		}
	};

	let column_data = match value {
		Value::Boolean(b) => ColumnData::bool(vec![*b; ctx.row_count]),
		Value::Float4(f) => ColumnData::float4(vec![f.value(); ctx.row_count]),
		Value::Float8(f) => ColumnData::float8(vec![f.value(); ctx.row_count]),
		Value::Int1(i) => ColumnData::int1(vec![*i; ctx.row_count]),
		Value::Int2(i) => ColumnData::int2(vec![*i; ctx.row_count]),
		Value::Int4(i) => ColumnData::int4(vec![*i; ctx.row_count]),
		Value::Int8(i) => ColumnData::int8(vec![*i; ctx.row_count]),
		Value::Int16(i) => ColumnData::int16(vec![*i; ctx.row_count]),
		Value::Uint1(u) => ColumnData::uint1(vec![*u; ctx.row_count]),
		Value::Uint2(u) => ColumnData::uint2(vec![*u; ctx.row_count]),
		Value::Uint4(u) => ColumnData::uint4(vec![*u; ctx.row_count]),
		Value::Uint8(u) => ColumnData::uint8(vec![*u; ctx.row_count]),
		Value::Uint16(u) => ColumnData::uint16(vec![*u; ctx.row_count]),
		Value::Utf8(s) => ColumnData::utf8(vec![s.clone(); ctx.row_count]),
		Value::Date(d) => ColumnData::date(vec![*d; ctx.row_count]),
		Value::DateTime(dt) => ColumnData::datetime(vec![*dt; ctx.row_count]),
		Value::Time(t) => ColumnData::time(vec![*t; ctx.row_count]),
		Value::Duration(i) => ColumnData::duration(vec![*i; ctx.row_count]),
		Value::Uuid4(u) => ColumnData::uuid4(vec![*u; ctx.row_count]),
		Value::Uuid7(u) => ColumnData::uuid7(vec![*u; ctx.row_count]),
		Value::Blob(b) => ColumnData::blob(vec![b.clone(); ctx.row_count]),
		Value::IdentityId(id) => ColumnData::identity_id(vec![*id; ctx.row_count]),
		Value::DictionaryId(v) => ColumnData::dictionary_id(vec![*v; ctx.row_count]),
		Value::Int(bi) => ColumnData::int(vec![bi.clone(); ctx.row_count]),
		Value::Uint(bu) => ColumnData::uint(vec![bu.clone(); ctx.row_count]),
		Value::Decimal(bd) => ColumnData::decimal(vec![bd.clone(); ctx.row_count]),
		Value::None {
			..
		} => ColumnData::none_typed(Type::Boolean, ctx.row_count),
		Value::Type(_) | Value::Any(_) | Value::List(_) | Value::Record(_) | Value::Tuple(_) => {
			unreachable!("Any/Type/List/Record/Tuple not supported as parameter")
		}
	};
	Ok(Column {
		name: Fragment::internal("parameter"),
		data: column_data,
	})
}