reifydb_function/date/
age.rs

1// SPDX-License-Identifier: AGPL-3.0-or-later
2// Copyright (c) 2025 ReifyDB
3
4use reifydb_core::value::column::data::ColumnData;
5use reifydb_type::value::{container::temporal::TemporalContainer, date::Date, duration::Duration, r#type::Type};
6
7use crate::{
8	ScalarFunction, ScalarFunctionContext,
9	error::{ScalarFunctionError, ScalarFunctionResult},
10	propagate_options,
11};
12
13pub struct DateAge;
14
15impl DateAge {
16	pub fn new() -> Self {
17		Self
18	}
19}
20
21/// Compute calendar-aware age between two dates.
22/// Returns Duration with months + days components.
23pub fn date_age(d1: &Date, d2: &Date) -> Duration {
24	let y1 = d1.year();
25	let m1 = d1.month() as i32;
26	let day1 = d1.day() as i32;
27
28	let y2 = d2.year();
29	let m2 = d2.month() as i32;
30	let day2 = d2.day() as i32;
31
32	let mut years = y1 - y2;
33	let mut months = m1 - m2;
34	let mut days = day1 - day2;
35
36	if days < 0 {
37		months -= 1;
38		// Borrow days from previous month (relative to d2's perspective)
39		let borrow_month = if m1 - 1 < 1 {
40			12
41		} else {
42			m1 - 1
43		};
44		let borrow_year = if m1 - 1 < 1 {
45			y1 - 1
46		} else {
47			y1
48		};
49		days += Date::days_in_month(borrow_year, borrow_month as u32) as i32;
50	}
51
52	if months < 0 {
53		years -= 1;
54		months += 12;
55	}
56
57	let total_months = years * 12 + months;
58	Duration::new(total_months, days, 0)
59}
60
61impl ScalarFunction for DateAge {
62	fn scalar(&self, ctx: ScalarFunctionContext) -> ScalarFunctionResult<ColumnData> {
63		if let Some(result) = propagate_options(self, &ctx) {
64			return result;
65		}
66
67		let columns = ctx.columns;
68		let row_count = ctx.row_count;
69
70		if columns.len() != 2 {
71			return Err(ScalarFunctionError::ArityMismatch {
72				function: ctx.fragment.clone(),
73				expected: 2,
74				actual: columns.len(),
75			});
76		}
77
78		let col1 = columns.get(0).unwrap();
79		let col2 = columns.get(1).unwrap();
80
81		match (col1.data(), col2.data()) {
82			(ColumnData::Date(container1), ColumnData::Date(container2)) => {
83				let mut container = TemporalContainer::with_capacity(row_count);
84
85				for i in 0..row_count {
86					match (container1.get(i), container2.get(i)) {
87						(Some(d1), Some(d2)) => {
88							container.push(date_age(&d1, &d2));
89						}
90						_ => container.push_default(),
91					}
92				}
93
94				Ok(ColumnData::Duration(container))
95			}
96			(ColumnData::Date(_), other) => Err(ScalarFunctionError::InvalidArgumentType {
97				function: ctx.fragment.clone(),
98				argument_index: 1,
99				expected: vec![Type::Date],
100				actual: other.get_type(),
101			}),
102			(other, _) => Err(ScalarFunctionError::InvalidArgumentType {
103				function: ctx.fragment.clone(),
104				argument_index: 0,
105				expected: vec![Type::Date],
106				actual: other.get_type(),
107			}),
108		}
109	}
110
111	fn return_type(&self, _input_types: &[Type]) -> Type {
112		Type::Duration
113	}
114}
reifydb_function/date/age.rs

reifydb_function/date/
age.rs
