reifydb_engine/bulk_insert/
coerce.rs

1// SPDX-License-Identifier: Apache-2.0
2// Copyright (c) 2026 ReifyDB
3
4use reifydb_core::{
5	interface::catalog::column::Column,
6	value::column::{buffer::ColumnBuffer, cast::cast_column_data, columns::Columns},
7};
8use reifydb_evaluate::{expression::context::EvalContext, stack::SymbolTable};
9use reifydb_routine_abi::registry::Routines;
10use reifydb_runtime::context::{RuntimeContext, clock::Clock};
11use reifydb_value::{fragment::Fragment, params::Params, value::identity::IdentityId};
12
13use crate::Result;
14
15pub(super) fn coerce_columns(
16	column_data: &[ColumnBuffer],
17	columns: &[Column],
18	num_rows: usize,
19) -> Result<Vec<ColumnBuffer>> {
20	let runtime_ctx = RuntimeContext::with_clock(Clock::Real);
21	let routines = Routines::empty();
22	let ctx = EvalContext {
23		params: &Params::None,
24		symbols: &SymbolTable::new(),
25		routines: &routines,
26		runtime_context: &runtime_ctx,
27		identity: IdentityId::root(),
28		is_aggregate_context: false,
29		columns: Columns::empty(),
30		row_count: num_rows,
31		target: None,
32		take: None,
33	};
34
35	let mut coerced_columns: Vec<ColumnBuffer> = Vec::with_capacity(columns.len());
36
37	for (col_idx, col) in columns.iter().enumerate() {
38		let target = col.constraint.get_type();
39
40		let cast_target = target.inner_type().clone();
41		let source_data = &column_data[col_idx];
42
43		let coerced = cast_column_data(&ctx, source_data, cast_target, || Fragment::internal(&col.name))?;
44		coerced_columns.push(coerced);
45	}
46
47	Ok(coerced_columns)
48}
reifydb_engine/bulk_insert/coerce.rs

reifydb_engine/bulk_insert/
coerce.rs
