reifydb_engine/expression/arith/
sub.rs

1// SPDX-License-Identifier: AGPL-3.0-or-later
2// Copyright (c) 2025 ReifyDB
3
4use reifydb_core::value::column::{Column, data::ColumnData, push::Push};
5use reifydb_type::{
6	error::diagnostic::operator::sub_cannot_be_applied_to_incompatible_types,
7	fragment::LazyFragment,
8	return_error,
9	value::{
10		container::{number::NumberContainer, temporal::TemporalContainer},
11		is::IsNumber,
12		number::{promote::Promote, safe::sub::SafeSub},
13		r#type::{Type, get::GetType},
14	},
15};
16
17use crate::expression::context::EvalContext;
18
19pub(crate) fn sub_columns(
20	ctx: &EvalContext,
21	left: &Column,
22	right: &Column,
23	fragment: impl LazyFragment + Copy,
24) -> crate::Result<Column> {
25	crate::expression::option::binary_op_unwrap_option(left, right, fragment.fragment(), |left, right| {
26		let target = Type::promote(left.get_type(), right.get_type());
27
28		dispatch_arith!(
29			&left.data(), &right.data();
30			fixed: sub_numeric, arb: sub_numeric_clone (ctx, target, fragment);
31
32			// Duration - Duration
33			(ColumnData::Duration(l), ColumnData::Duration(r)) => {
34				let mut container = TemporalContainer::with_capacity(l.len());
35				for i in 0..l.len() {
36					match (l.get(i), r.get(i)) {
37						(Some(lv), Some(rv)) => container.push(*lv - *rv),
38						_ => container.push_default(),
39					}
40				}
41				Ok(Column {
42					name: fragment.fragment(),
43					data: ColumnData::Duration(container),
44				})
45			}
46
47			_ => return_error!(sub_cannot_be_applied_to_incompatible_types(
48				fragment.fragment(),
49				left.get_type(),
50				right.get_type(),
51			)),
52		)
53	})
54}
55
56fn sub_numeric<'a, L, R>(
57	ctx: &EvalContext,
58	l: &NumberContainer<L>,
59	r: &NumberContainer<R>,
60	target: Type,
61	fragment: impl LazyFragment + Copy,
62) -> crate::Result<Column>
63where
64	L: GetType + Promote<R> + IsNumber,
65	R: GetType + IsNumber,
66	<L as Promote<R>>::Output: IsNumber,
67	<L as Promote<R>>::Output: SafeSub,
68	ColumnData: Push<<L as Promote<R>>::Output>,
69{
70	debug_assert_eq!(l.len(), r.len());
71
72	let mut data = ColumnData::with_capacity(target, l.len());
73	let l_data = l.data();
74	let r_data = r.data();
75	for i in 0..l.len() {
76		if let Some(value) = ctx.sub(&l_data[i], &r_data[i], fragment)? {
77			data.push(value);
78		} else {
79			data.push_none()
80		}
81	}
82	Ok(Column {
83		name: fragment.fragment(),
84		data,
85	})
86}
87
88fn sub_numeric_clone<'a, L, R>(
89	ctx: &EvalContext,
90	l: &NumberContainer<L>,
91	r: &NumberContainer<R>,
92	target: Type,
93	fragment: impl LazyFragment + Copy,
94) -> crate::Result<Column>
95where
96	L: Clone + GetType + Promote<R> + IsNumber,
97	R: Clone + GetType + IsNumber,
98	<L as Promote<R>>::Output: IsNumber,
99	<L as Promote<R>>::Output: SafeSub,
100	ColumnData: Push<<L as Promote<R>>::Output>,
101{
102	debug_assert_eq!(l.len(), r.len());
103
104	let mut data = ColumnData::with_capacity(target, l.len());
105	let l_data = l.data();
106	let r_data = r.data();
107	for i in 0..l.len() {
108		let l_clone = l_data[i].clone();
109		let r_clone = r_data[i].clone();
110		if let Some(value) = ctx.sub(&l_clone, &r_clone, fragment)? {
111			data.push(value);
112		} else {
113			data.push_none()
114		}
115	}
116	Ok(Column {
117		name: fragment.fragment(),
118		data,
119	})
120}
reifydb_engine/expression/arith/sub.rs

reifydb_engine/expression/arith/
sub.rs
