risinglight 0.2.0

An OLAP database system for educational purpose
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

// Copyright 2022 RisingLight Project Authors. Licensed under Apache-2.0.

// use crate::array::ArrayImplValidExt;
use super::*;
use crate::array::ArrayImplValidExt;

/// State for row count aggregation
pub struct CountAggregationState {
    result: DataValue,
}

impl CountAggregationState {
    pub fn new(init: DataValue) -> Self {
        Self { result: init }
    }
}

impl AggregationState for CountAggregationState {
    fn update(&mut self, array: &ArrayImpl) -> Result<(), ExecutorError> {
        // let temp = array.len() as i32;
        let temp = array.get_valid_bitmap().count_ones() as i32;
        self.result = match &self.result {
            DataValue::Null => DataValue::Int32(temp),
            DataValue::Int32(res) => DataValue::Int32(res + temp),
            _ => panic!("Mismatched type"),
        };
        Ok(())
    }

    fn update_single(&mut self, _: &DataValue) -> Result<(), ExecutorError> {
        self.result = match &self.result {
            DataValue::Null => DataValue::Int32(1),
            DataValue::Int32(res) => DataValue::Int32(res + 1),
            _ => panic!("Mismatched type"),
        };
        Ok(())
    }

    fn output(&self) -> DataValue {
        self.result.clone()
    }
}