Struct TaskContext

pub struct TaskContext { /* private fields */ }

Task Execution Context

A TaskContext contains the state required during a single query’s execution. Please see the documentation on SessionContext for more information.

Implementations

impl TaskContext

pub fn new( task_id: Option<String>, session_id: String, session_config: SessionConfig, scalar_functions: HashMap<String, Arc<ScalarUDF>>, aggregate_functions: HashMap<String, Arc<AggregateUDF>>, window_functions: HashMap<String, Arc<WindowUDF>>, runtime: Arc<RuntimeEnv>, ) -> Self

Create a new TaskContext instance.

Most users will use SessionContext::task_ctx to create TaskContexts

pub fn session_config(&self) -> &SessionConfig

Return the SessionConfig associated with this TaskContext

pub fn session_id(&self) -> String

Return the session_id of this TaskContext

pub fn task_id(&self) -> Option<String>

Return the task_id of this TaskContext

pub fn memory_pool(&self) -> &Arc<dyn MemoryPool>

Return the MemoryPool associated with this TaskContext

pub fn runtime_env(&self) -> Arc<RuntimeEnv>

Return the RuntimeEnv associated with this TaskContext

pub fn with_session_config(self, session_config: SessionConfig) -> Self

Update the SessionConfig

pub fn with_runtime(self, runtime: Arc<RuntimeEnv>) -> Self

Update the RuntimeEnv

Trait Implementations

impl Debug for TaskContext

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for TaskContext

fn default() -> Self

Returns the “default value” for a type.

impl FunctionRegistry for TaskContext

fn udfs(&self) -> HashSet<String>

Set of all available udfs.

fn udf(&self, name: &str) -> Result<Arc<ScalarUDF>>

Returns a reference to the user defined scalar function (udf) named name.

fn udaf(&self, name: &str) -> Result<Arc<AggregateUDF>>

Returns a reference to the user defined aggregate function (udaf) named name.

fn udwf(&self, name: &str) -> Result<Arc<WindowUDF>>

Returns a reference to the user defined window function (udwf) named name.

fn register_udaf( &mut self, udaf: Arc<AggregateUDF>, ) -> Result<Option<Arc<AggregateUDF>>>

Registers a new AggregateUDF, returning any previously registered implementation.

fn register_udwf( &mut self, udwf: Arc<WindowUDF>, ) -> Result<Option<Arc<WindowUDF>>>

Registers a new WindowUDF, returning any previously registered implementation.

fn register_udf( &mut self, udf: Arc<ScalarUDF>, ) -> Result<Option<Arc<ScalarUDF>>>

Registers a new ScalarUDF, returning any previously registered implementation.

fn expr_planners(&self) -> Vec<Arc<dyn ExprPlanner>>

Set of all registered ExprPlanners

fn deregister_udf( &mut self, _name: &str, ) -> Result<Option<Arc<ScalarUDF>>, DataFusionError>

Deregisters a ScalarUDF, returning the implementation that was deregistered.

fn deregister_udaf( &mut self, _name: &str, ) -> Result<Option<Arc<AggregateUDF>>, DataFusionError>

Deregisters a AggregateUDF, returning the implementation that was deregistered.

fn deregister_udwf( &mut self, _name: &str, ) -> Result<Option<Arc<WindowUDF>>, DataFusionError>

Deregisters a WindowUDF, returning the implementation that was deregistered.

fn register_function_rewrite( &mut self, _rewrite: Arc<dyn FunctionRewrite + Sync + Send>, ) -> Result<(), DataFusionError>

Registers a new FunctionRewrite with the registry.

fn register_expr_planner( &mut self, _expr_planner: Arc<dyn ExprPlanner>, ) -> Result<(), DataFusionError>

Registers a new ExprPlanner with the registry.