Function arrow_sum

nodedb/src/control/arrow_convert.rs:173–185

Compute SUM of a numeric column in a RecordBatch.

(batch: &RecordBatch, column_name: &str)

/// Compute SUM of a numeric column in a RecordBatch.
pub fn arrow_sum(batch: &RecordBatch, column_name: &str) -> Option<f64> {
    let idx = batch.schema().index_of(column_name).ok()?;
    let array = batch.column(idx);

    if let Some(f64_arr) = array.as_any().downcast_ref::<Float64Array>() {
        Some(arrow::compute::kernels::aggregate::sum(f64_arr).unwrap_or(0.0))
    } else if let Some(i64_arr) = array.as_any().downcast_ref::<Int64Array>() {
        let sum = arrow::compute::kernels::aggregate::sum(i64_arr).unwrap_or(0);
        Some(sum as f64)
    } else {
        None
    }
}

/// Compute MIN of a numeric column using Arrow SIMD kernels.
pub fn arrow_min(batch: &RecordBatch, column_name: &str) -> Option<f64> {