Function compute_aggregate

nodedb/src/control/server/post_aggregate.rs:196–228

Compute a single aggregate over a group of msgpack rows.

(op: &str, field: &str, rows: &[&[u8]])

/// Compute a single aggregate over a group of msgpack rows.
fn compute_aggregate(op: &str, field: &str, rows: &[&[u8]]) -> AggValue {
    match op {
        "count" => AggValue::Int(rows.len() as i64),
        "sum" => {
            let sum: f64 = rows.iter().filter_map(|r| extract_number(r, field)).sum();
            AggValue::Float(sum)
        }
        "avg" => {
            let values: Vec<f64> = rows
                .iter()
                .filter_map(|r| extract_number(r, field))
                .collect();
            if values.is_empty() {
                AggValue::Null
            } else {
                AggValue::Float(values.iter().sum::<f64>() / values.len() as f64)
            }
        }
        "min" => rows
            .iter()
            .filter_map(|r| extract_number(r, field))
            .min_by(|a, b| a.partial_cmp(b).unwrap_or(std::cmp::Ordering::Equal))
            .map(AggValue::Float)
            .unwrap_or(AggValue::Null),
        "max" => rows
            .iter()
            .filter_map(|r| extract_number(r, field))
            .max_by(|a, b| a.partial_cmp(b).unwrap_or(std::cmp::Ordering::Equal))
            .map(AggValue::Float)
            .unwrap_or(AggValue::Null),
        _ => AggValue::Null,
    }
}