Trait UdfImpl

pub trait UdfImpl:
    Debug
    + Send
    + Sync {
    // Required methods
    fn call<'life0, 'life1, 'async_trait>(
        &'life0 self,
        input: &'life1 RecordBatch,
    ) -> Pin<Box<dyn Future<Output = Result<RecordBatch, Error>> + Send + 'async_trait>>
       where 'life0: 'async_trait,
             'life1: 'async_trait,
             Self: 'async_trait;
    fn call_table_function<'a, 'async_trait>(
        &'a self,
        input: &'a RecordBatch,
    ) -> Pin<Box<dyn Future<Output = Result<Pin<Box<dyn Stream<Item = Result<RecordBatch, Error>> + Send + 'a>>, Error>> + Send + 'async_trait>>
       where 'a: 'async_trait,
             Self: 'async_trait;

    // Provided methods
    fn call_agg_create_state<'life0, 'async_trait>(
        &'life0 self,
    ) -> Pin<Box<dyn Future<Output = Result<Arc<dyn Array>, Error>> + Send + 'async_trait>>
       where 'life0: 'async_trait,
             Self: 'async_trait { ... }
    fn call_agg_accumulate_or_retract<'life0, 'life1, 'life2, 'life3, 'async_trait>(
        &'life0 self,
        _state: &'life1 Arc<dyn Array>,
        _ops: &'life2 BooleanArray,
        _input: &'life3 RecordBatch,
    ) -> Pin<Box<dyn Future<Output = Result<Arc<dyn Array>, Error>> + Send + 'async_trait>>
       where 'life0: 'async_trait,
             'life1: 'async_trait,
             'life2: 'async_trait,
             'life3: 'async_trait,
             Self: 'async_trait { ... }
    fn call_agg_finish<'life0, 'life1, 'async_trait>(
        &'life0 self,
        _state: &'life1 Arc<dyn Array>,
    ) -> Pin<Box<dyn Future<Output = Result<Arc<dyn Array>, Error>> + Send + 'async_trait>>
       where 'life0: 'async_trait,
             'life1: 'async_trait,
             Self: 'async_trait { ... }
    fn is_legacy(&self) -> bool { ... }
    fn memory_usage(&self) -> usize { ... }
}

UDF implementation.

Required Methods

fn call<'life0, 'life1, 'async_trait>( &'life0 self, input: &'life1 RecordBatch, ) -> Pin<Box<dyn Future<Output = Result<RecordBatch, Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, Self: 'async_trait,

Call the scalar function.

fn call_table_function<'a, 'async_trait>( &'a self, input: &'a RecordBatch, ) -> Pin<Box<dyn Future<Output = Result<Pin<Box<dyn Stream<Item = Result<RecordBatch, Error>> + Send + 'a>>, Error>> + Send + 'async_trait>>

where 'a: 'async_trait, Self: 'async_trait,

Call the table function.

Provided Methods

fn call_agg_create_state<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<Arc<dyn Array>, Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, Self: 'async_trait,

For aggregate function, create the initial state.

fn call_agg_accumulate_or_retract<'life0, 'life1, 'life2, 'life3, 'async_trait>( &'life0 self, _state: &'life1 Arc<dyn Array>, _ops: &'life2 BooleanArray, _input: &'life3 RecordBatch, ) -> Pin<Box<dyn Future<Output = Result<Arc<dyn Array>, Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, 'life3: 'async_trait, Self: 'async_trait,

For aggregate function, accumulate or retract the state.

fn call_agg_finish<'life0, 'life1, 'async_trait>( &'life0 self, _state: &'life1 Arc<dyn Array>, ) -> Pin<Box<dyn Future<Output = Result<Arc<dyn Array>, Error>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, Self: 'async_trait,

For aggregate function, get aggregate result from the state.

fn is_legacy(&self) -> bool

Whether the UDF talks in legacy mode.

If true, decimal and jsonb types are mapped to Arrow LargeBinary and LargeUtf8 types. Otherwise, they are mapped to Arrow extension types. See https://github.com/risingwavelabs/arrow-udf/tree/main#extension-types.

fn memory_usage(&self) -> usize

Return the memory size consumed by UDF runtime in bytes.

If not available, return 0.

Implementors