pub fn-s

Signed-off-by: blaginin <github@blaginin.me>

Author: blaginin

vortex-array/public-api.lock

@@ -128,6 +128,8 @@ pub fn vortex_array::aggregate_fn::fns::first::First::try_accumulate(&self, part
 
 pub struct vortex_array::aggregate_fn::fns::first::FirstPartial
 
+pub fn vortex_array::aggregate_fn::fns::first::first(array: &vortex_array::ArrayRef, ctx: &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<vortex_array::scalar::Scalar>
+
 pub mod vortex_array::aggregate_fn::fns::is_constant
 
 pub mod vortex_array::aggregate_fn::fns::is_constant::primitive
@@ -338,6 +340,8 @@ pub fn vortex_array::aggregate_fn::fns::last::Last::try_accumulate(&self, partia
 
 pub struct vortex_array::aggregate_fn::fns::last::LastPartial
 
+pub fn vortex_array::aggregate_fn::fns::last::last(array: &vortex_array::ArrayRef, ctx: &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<vortex_array::scalar::Scalar>
+
 pub mod vortex_array::aggregate_fn::fns::min_max
 
 pub struct vortex_array::aggregate_fn::fns::min_max::MinMax

vortex-array/src/aggregate_fn/fns/first/mod.rs

@@ -6,12 +6,23 @@ use vortex_error::VortexResult;
 use crate::ArrayRef;
 use crate::Columnar;
 use crate::ExecutionCtx;
+use crate::aggregate_fn::Accumulator;
 use crate::aggregate_fn::AggregateFnId;
 use crate::aggregate_fn::AggregateFnVTable;
+use crate::aggregate_fn::DynAccumulator;
 use crate::aggregate_fn::EmptyOptions;
 use crate::dtype::DType;
 use crate::scalar::Scalar;
 
+/// Return the first non-null value of an array.
+///
+/// See [`First`] for details.
+pub fn first(array: &ArrayRef, ctx: &mut ExecutionCtx) -> VortexResult<Scalar> {
+    let mut acc = Accumulator::try_new(First, EmptyOptions, array.dtype().clone())?;
+    acc.accumulate(array, ctx)?;
+    acc.finish()
+}
+
 /// Return the first non-null value seen across all batches.
 #[derive(Clone, Debug)]
 pub struct First;
@@ -126,8 +137,6 @@ mod tests {
     use vortex_buffer::buffer;
     use vortex_error::VortexResult;
 
-    use crate::ArrayRef;
-    use crate::ExecutionCtx;
     use crate::IntoArray;
     use crate::LEGACY_SESSION;
     use crate::VortexSessionExecute;
@@ -136,6 +145,7 @@ mod tests {
     use crate::aggregate_fn::DynAccumulator;
     use crate::aggregate_fn::EmptyOptions;
     use crate::aggregate_fn::fns::first::First;
+    use crate::aggregate_fn::fns::first::first;
     use crate::arrays::ChunkedArray;
     use crate::arrays::ConstantArray;
     use crate::arrays::PrimitiveArray;
@@ -147,12 +157,6 @@ mod tests {
     use crate::scalar::Scalar;
     use crate::validity::Validity;
 
-    fn first(array: &ArrayRef, ctx: &mut ExecutionCtx) -> VortexResult<Scalar> {
-        let mut acc = Accumulator::try_new(First, EmptyOptions, array.dtype().clone())?;
-        acc.accumulate(array, ctx)?;
-        acc.finish()
-    }
-
     #[test]
     fn first_non_null() -> VortexResult<()> {
         let array = PrimitiveArray::new(buffer![10i32, 20, 30], Validity::NonNullable).into_array();

vortex-array/src/aggregate_fn/fns/last/mod.rs

@@ -6,12 +6,23 @@ use vortex_error::VortexResult;
 use crate::ArrayRef;
 use crate::Columnar;
 use crate::ExecutionCtx;
+use crate::aggregate_fn::Accumulator;
 use crate::aggregate_fn::AggregateFnId;
 use crate::aggregate_fn::AggregateFnVTable;
+use crate::aggregate_fn::DynAccumulator;
 use crate::aggregate_fn::EmptyOptions;
 use crate::dtype::DType;
 use crate::scalar::Scalar;
 
+/// Return the last non-null value of an array.
+///
+/// See [`Last`] for details.
+pub fn last(array: &ArrayRef, ctx: &mut ExecutionCtx) -> VortexResult<Scalar> {
+    let mut acc = Accumulator::try_new(Last, EmptyOptions, array.dtype().clone())?;
+    acc.accumulate(array, ctx)?;
+    acc.finish()
+}
+
 /// Return the last non-null value seen across all batches.
 #[derive(Clone, Debug)]
 pub struct Last;
@@ -124,8 +135,6 @@ mod tests {
     use vortex_buffer::buffer;
     use vortex_error::VortexResult;
 
-    use crate::ArrayRef;
-    use crate::ExecutionCtx;
     use crate::IntoArray;
     use crate::LEGACY_SESSION;
     use crate::VortexSessionExecute;
@@ -134,6 +143,7 @@ mod tests {
     use crate::aggregate_fn::DynAccumulator;
     use crate::aggregate_fn::EmptyOptions;
     use crate::aggregate_fn::fns::last::Last;
+    use crate::aggregate_fn::fns::last::last;
     use crate::arrays::ChunkedArray;
     use crate::arrays::ConstantArray;
     use crate::arrays::PrimitiveArray;
@@ -145,12 +155,6 @@ mod tests {
     use crate::scalar::Scalar;
     use crate::validity::Validity;
 
-    fn last(array: &ArrayRef, ctx: &mut ExecutionCtx) -> VortexResult<Scalar> {
-        let mut acc = Accumulator::try_new(Last, EmptyOptions, array.dtype().clone())?;
-        acc.accumulate(array, ctx)?;
-        acc.finish()
-    }
-
     #[test]
     fn last_non_null() -> VortexResult<()> {
         let array = PrimitiveArray::new(buffer![10i32, 20, 30], Validity::NonNullable).into_array();