Enum AggState

pub enum AggState {
    Value(AggregateState),
    MaterializedInput(Box<MaterializedInputState>),
}

State for single aggregation call. It manages the state cache and interact with the underlying state store if necessary.

Variants

Value(AggregateState)

State as a single scalar value. e.g. count, sum, append-only min/max.

MaterializedInput(Box<MaterializedInputState>)

State as materialized input chunk, e.g. non-append-only min/max, string_agg.

Implementations

impl AggState

pub fn create( version: PbAggNodeVersion, agg_call: &AggCall, agg_func: &BoxedAggregateFunction, storage: &AggStateStorage<impl StateStore>, encoded_state: Option<&Datum>, pk_indices: &PkIndices, extreme_cache_size: usize, input_schema: &Schema, ) -> StreamExecutorResult<Self>

Create an AggState from a given AggCall.

pub async fn apply_chunk( &mut self, chunk: &StreamChunk, call: &AggCall, func: &BoxedAggregateFunction, visibility: Bitmap, ) -> StreamExecutorResult<()>

Apply input chunk to the state.

pub async fn get_output( &mut self, storage: &AggStateStorage<impl StateStore>, func: &BoxedAggregateFunction, group_key: Option<&GroupKey>, ) -> StreamExecutorResult<(Datum, AggStateCacheStats)>

Get the output of aggregations.

pub fn reset( &mut self, func: &BoxedAggregateFunction, ) -> StreamExecutorResult<()>

Reset the value state to initial state.

Trait Implementations

impl EstimateSize for AggState

fn estimated_heap_size(&self) -> usize

The estimated heap size of the current struct in bytes.

fn estimated_size(&self) -> usize

where Self: Sized,

The estimated total size of the current struct in bytes, including the estimated_heap_size and the size of Self.