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

reifydb_function/date/
age.rs
