datafusion-distributed 2.0.0

Framework for enhancing Apache DataFusion with distributed capabilities
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use datafusion::execution::TaskContext;
use std::sync::Arc;

/// Builds a new [TaskContext] by cloning the provided one and adding the config extension to it.
pub(crate) fn task_ctx_with_extension<T: Send + Sync + 'static>(
    ctx: &TaskContext,
    ext: T,
) -> TaskContext {
    TaskContext::new(
        ctx.task_id(),
        ctx.session_id(),
        ctx.session_config().clone().with_extension(Arc::new(ext)),
        ctx.scalar_functions().clone(),
        ctx.higher_order_functions().clone(),
        ctx.aggregate_functions().clone(),
        ctx.window_functions().clone(),
        ctx.runtime_env(),
    )
}