Struct DatafusionQueryEngine

pub struct DatafusionQueryEngine {
    state: Arc<QueryEngineState>,
    plugins: Plugins,
}

Fields

state: Arc<QueryEngineState>

plugins: Plugins

Implementations

impl DatafusionQueryEngine

pub fn new(state: Arc<QueryEngineState>, plugins: Plugins) -> Self

async fn exec_query_plan( &self, plan: LogicalPlan, query_ctx: QueryContextRef, ) -> Result<Output>

async fn exec_dml_statement( &self, dml: DmlStatement, query_ctx: QueryContextRef, ) -> Result<Output>

async fn delete( &self, table_name: &ResolvedTableReference, table: &TableRef, column_vectors: HashMap<String, VectorRef>, query_ctx: QueryContextRef, ) -> Result<usize>

async fn insert( &self, table_name: &ResolvedTableReference, column_vectors: HashMap<String, VectorRef>, query_ctx: QueryContextRef, ) -> Result<Output>

async fn find_table( &self, table_name: &ResolvedTableReference, query_context: &QueryContextRef, ) -> Result<TableRef>

async fn create_physical_plan( &self, ctx: &mut QueryEngineContext, logical_plan: &LogicalPlan, ) -> Result<Arc<dyn ExecutionPlan>>

pub fn optimize( &self, context: &QueryEngineContext, plan: &LogicalPlan, ) -> Result<LogicalPlan>

fn optimize_physical_plan( &self, ctx: &mut QueryEngineContext, plan: Arc<dyn ExecutionPlan>, ) -> Result<Arc<dyn ExecutionPlan>>

Trait Implementations

impl QueryEngine for DatafusionQueryEngine

fn register_aggregate_function(&self, func: AggregateFunctionMetaRef)

Note in SQL queries, aggregate names are looked up using lowercase unless the query uses quotes. For example,

SELECT MY_UDAF(x)... will look for an aggregate named "my_udaf" SELECT "my_UDAF"(x) will look for an aggregate named "my_UDAF"

So it’s better to make UDAF name lowercase when creating one.

fn register_udf(&self, udf: ScalarUdf)

Register a ScalarUdf.

fn register_function(&self, func: FunctionRef)

Register an UDF function. Will override if the function with same name is already registered.

fn as_any(&self) -> &dyn Any

Returns the query engine as Any so that it can be downcast to a specific implementation.

fn planner(&self) -> Arc<dyn LogicalPlanner>

Returns the logical planner

fn name(&self) -> &str

Returns the query engine name.

fn describe<'life0, 'async_trait>( &'life0 self, plan: LogicalPlan, query_ctx: QueryContextRef, ) -> Pin<Box<dyn Future<Output = Result<DescribeResult>> + Send + 'async_trait>>

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

Describe the given [LogicalPlan].

fn execute<'life0, 'async_trait>( &'life0 self, plan: LogicalPlan, query_ctx: QueryContextRef, ) -> Pin<Box<dyn Future<Output = Result<Output>> + Send + 'async_trait>>

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

Execute the given [LogicalPlan].

fn read_table(&self, table: TableRef) -> Result<DataFrame>

Create a DataFrame from a table.

fn engine_context(&self, query_ctx: QueryContextRef) -> QueryEngineContext

Create a QueryEngineContext.

fn engine_state(&self) -> &QueryEngineState

Retrieve the query engine state QueryEngineState

impl QueryExecutor for DatafusionQueryEngine

fn execute_stream( &self, ctx: &QueryEngineContext, plan: &Arc<dyn ExecutionPlan>, ) -> Result<SendableRecordBatchStream>