use crate::core::Value;
use crate::functions::{
AggregateFunction, FunctionDataType, FunctionInfo, FunctionSignature, FunctionType,
};
#[derive(Default)]
pub struct LastFunction {
last_value: Option<Value>,
}
impl AggregateFunction for LastFunction {
fn name(&self) -> &str {
"LAST"
}
fn info(&self) -> FunctionInfo {
FunctionInfo::new(
"LAST",
FunctionType::Aggregate,
"Returns the last non-NULL value in the specified column",
FunctionSignature::new(FunctionDataType::Any, vec![FunctionDataType::Any], 1, 1),
)
}
fn accumulate(&mut self, value: &Value, _distinct: bool) {
if value.is_null() {
return;
}
self.last_value = Some(value.clone());
}
fn result(&self) -> Value {
self.last_value.clone().unwrap_or_else(Value::null_unknown)
}
fn reset(&mut self) {
self.last_value = None;
}
fn clone_box(&self) -> Box<dyn AggregateFunction> {
Box::new(LastFunction::default())
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_last_basic() {
let mut last = LastFunction::default();
last.accumulate(&Value::Integer(1), false);
last.accumulate(&Value::Integer(2), false);
last.accumulate(&Value::Integer(3), false);
assert_eq!(last.result(), Value::Integer(3));
}
#[test]
fn test_last_ignores_null() {
let mut last = LastFunction::default();
last.accumulate(&Value::Integer(1), false);
last.accumulate(&Value::Integer(2), false);
last.accumulate(&Value::null_unknown(), false);
last.accumulate(&Value::null_unknown(), false);
assert_eq!(last.result(), Value::Integer(2));
}
#[test]
fn test_last_empty() {
let last = LastFunction::default();
assert!(last.result().is_null());
}
#[test]
fn test_last_all_null() {
let mut last = LastFunction::default();
last.accumulate(&Value::null_unknown(), false);
last.accumulate(&Value::null_unknown(), false);
assert!(last.result().is_null());
}
#[test]
fn test_last_reset() {
let mut last = LastFunction::default();
last.accumulate(&Value::Integer(1), false);
last.accumulate(&Value::Integer(2), false);
last.reset();
last.accumulate(&Value::Integer(5), false);
assert_eq!(last.result(), Value::Integer(5));
}
#[test]
fn test_last_string() {
let mut last = LastFunction::default();
last.accumulate(&Value::text("hello"), false);
last.accumulate(&Value::text("world"), false);
assert_eq!(last.result(), Value::text("world"));
}
}