Skip to content

Commit b654fde

Browse files
authored
Finish removing deprecated datafusion::logical_plan module (#3816)
* remove deprecated module * fix imports * infer schema for parquet in benchmarks * Revert "infer schema for parquet in benchmarks" This reverts commit 6899e98.
1 parent 051f243 commit b654fde

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

47 files changed

+92
-215
lines changed

benchmarks/src/bin/parquet_filter_pushdown.rs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,11 @@ use arrow::array::{
2222
use arrow::datatypes::{DataType, Field, Int32Type, Schema, SchemaRef, TimeUnit};
2323
use arrow::record_batch::RecordBatch;
2424
use arrow::util::pretty;
25-
use datafusion::common::Result;
25+
use datafusion::common::{Result, ToDFSchema};
2626
use datafusion::datasource::listing::{ListingTableUrl, PartitionedFile};
2727
use datafusion::datasource::object_store::ObjectStoreUrl;
2828
use datafusion::execution::context::ExecutionProps;
2929
use datafusion::logical_expr::{lit, or, Expr};
30-
use datafusion::logical_plan::ToDFSchema;
3130
use datafusion::optimizer::utils::combine_filters_disjunctive;
3231
use datafusion::physical_expr::create_physical_expr;
3332
use datafusion::physical_plan::collect;

datafusion-examples/examples/custom_datasource.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,16 +20,17 @@ use datafusion::arrow::array::{UInt64Builder, UInt8Builder};
2020
use datafusion::arrow::datatypes::{DataType, Field, Schema, SchemaRef};
2121
use datafusion::arrow::record_batch::RecordBatch;
2222
use datafusion::dataframe::DataFrame;
23+
use datafusion::datasource::provider_as_source;
2324
use datafusion::datasource::{TableProvider, TableType};
2425
use datafusion::error::Result;
2526
use datafusion::execution::context::{SessionState, TaskContext};
26-
use datafusion::logical_plan::{provider_as_source, Expr, LogicalPlanBuilder};
2727
use datafusion::physical_plan::expressions::PhysicalSortExpr;
2828
use datafusion::physical_plan::memory::MemoryStream;
2929
use datafusion::physical_plan::{
3030
project_schema, ExecutionPlan, SendableRecordBatchStream, Statistics,
3131
};
3232
use datafusion::prelude::*;
33+
use datafusion_expr::{Expr, LogicalPlanBuilder};
3334
use std::any::Any;
3435
use std::collections::{BTreeMap, HashMap};
3536
use std::fmt::{Debug, Formatter};

datafusion-examples/examples/expr_api.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,13 +16,12 @@
1616
// under the License.
1717

1818
use datafusion::arrow::datatypes::{DataType, Field, Schema, TimeUnit};
19-
2019
use datafusion::error::Result;
21-
use datafusion::logical_plan::ToDFSchema;
2220
use datafusion::optimizer::expr_simplifier::{ExprSimplifier, SimplifyContext};
2321
use datafusion::physical_expr::execution_props::ExecutionProps;
2422
use datafusion::prelude::*;
25-
use datafusion::{logical_plan::Operator, scalar::ScalarValue};
23+
use datafusion_common::{ScalarValue, ToDFSchema};
24+
use datafusion_expr::Operator;
2625

2726
/// This example demonstrates the DataFusion [`Expr`] API.
2827
///

datafusion-examples/examples/simple_udaf.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,11 +21,11 @@ use datafusion::arrow::{
2121
array::ArrayRef, array::Float32Array, array::Float64Array, datatypes::DataType,
2222
record_batch::RecordBatch,
2323
};
24-
2524
use datafusion::from_slice::FromSlice;
2625
use datafusion::logical_expr::AggregateState;
27-
use datafusion::{error::Result, logical_plan::create_udaf, physical_plan::Accumulator};
26+
use datafusion::{error::Result, physical_plan::Accumulator};
2827
use datafusion::{logical_expr::Volatility, prelude::*, scalar::ScalarValue};
28+
use datafusion_expr::create_udaf;
2929
use std::sync::Arc;
3030

3131
// create local session context with an in-memory table

datafusion/core/src/dataframe.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,9 @@ use crate::execution::{
2727
context::{SessionState, TaskContext},
2828
FunctionRegistry,
2929
};
30-
use crate::logical_expr::{utils::find_window_exprs, TableType};
31-
use crate::logical_plan::{
32-
col, DFSchema, Expr, JoinType, LogicalPlan, LogicalPlanBuilder, Partitioning,
30+
use crate::logical_expr::{
31+
col, utils::find_window_exprs, Expr, JoinType, LogicalPlan, LogicalPlanBuilder,
32+
Partitioning, TableType,
3333
};
3434
use crate::physical_plan::file_format::{plan_to_csv, plan_to_json, plan_to_parquet};
3535
use crate::physical_plan::SendableRecordBatchStream;
@@ -38,7 +38,7 @@ use crate::physical_plan::{execute_stream, execute_stream_partitioned, Execution
3838
use crate::prelude::SessionContext;
3939
use crate::scalar::ScalarValue;
4040
use async_trait::async_trait;
41-
use datafusion_common::Column;
41+
use datafusion_common::{Column, DFSchema};
4242
use parking_lot::RwLock;
4343
use parquet::file::properties::WriterProperties;
4444
use std::any::Any;

datafusion/core/src/datasource/datasource.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ pub use datafusion_expr::{TableProviderFilterPushDown, TableType};
2626
use crate::arrow::datatypes::SchemaRef;
2727
use crate::error::Result;
2828
use crate::execution::context::SessionState;
29-
use crate::logical_plan::Expr;
29+
use crate::logical_expr::Expr;
3030
use crate::physical_plan::ExecutionPlan;
3131

3232
/// Source table

datafusion/core/src/datasource/empty.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ use async_trait::async_trait;
2626
use crate::datasource::{TableProvider, TableType};
2727
use crate::error::Result;
2828
use crate::execution::context::SessionState;
29-
use crate::logical_plan::Expr;
29+
use crate::logical_expr::Expr;
3030
use crate::physical_plan::project_schema;
3131
use crate::physical_plan::{empty::EmptyExec, ExecutionPlan};
3232

datafusion/core/src/datasource/file_format/avro.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ use object_store::{GetResult, ObjectMeta, ObjectStore};
2828
use super::FileFormat;
2929
use crate::avro_to_arrow::read_avro_schema_from_reader;
3030
use crate::error::Result;
31-
use crate::logical_plan::Expr;
31+
use crate::logical_expr::Expr;
3232
use crate::physical_plan::file_format::{AvroExec, FileScanConfig};
3333
use crate::physical_plan::ExecutionPlan;
3434
use crate::physical_plan::Statistics;

datafusion/core/src/datasource/file_format/csv.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ use super::FileFormat;
3535
use crate::datasource::file_format::file_type::FileCompressionType;
3636
use crate::datasource::file_format::DEFAULT_SCHEMA_INFER_MAX_RECORD;
3737
use crate::error::Result;
38-
use crate::logical_plan::Expr;
38+
use crate::logical_expr::Expr;
3939
use crate::physical_plan::file_format::{CsvExec, FileScanConfig};
4040
use crate::physical_plan::ExecutionPlan;
4141
use crate::physical_plan::Statistics;

datafusion/core/src/datasource/file_format/json.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ use super::FileScanConfig;
3636
use crate::datasource::file_format::file_type::FileCompressionType;
3737
use crate::datasource::file_format::DEFAULT_SCHEMA_INFER_MAX_RECORD;
3838
use crate::error::Result;
39-
use crate::logical_plan::Expr;
39+
use crate::logical_expr::Expr;
4040
use crate::physical_plan::file_format::NdJsonExec;
4141
use crate::physical_plan::ExecutionPlan;
4242
use crate::physical_plan::Statistics;

datafusion/core/src/datasource/file_format/mod.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ use std::sync::Arc;
3232

3333
use crate::arrow::datatypes::SchemaRef;
3434
use crate::error::Result;
35-
use crate::logical_plan::Expr;
35+
use crate::logical_expr::Expr;
3636
use crate::physical_plan::file_format::FileScanConfig;
3737
use crate::physical_plan::{ExecutionPlan, Statistics};
3838

datafusion/core/src/datasource/file_format/parquet.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ use crate::arrow::array::{
4141
use crate::arrow::datatypes::{DataType, Field};
4242
use crate::datasource::{create_max_min_accs, get_col_stats};
4343
use crate::error::Result;
44-
use crate::logical_plan::Expr;
44+
use crate::logical_expr::Expr;
4545
use crate::physical_plan::expressions::{MaxAccumulator, MinAccumulator};
4646
use crate::physical_plan::file_format::{ParquetExec, SchemaAdapter};
4747
use crate::physical_plan::{Accumulator, ExecutionPlan, Statistics};

datafusion/core/src/datasource/listing/helpers.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -357,7 +357,7 @@ fn parse_partitions_for_path<'a>(
357357

358358
#[cfg(test)]
359359
mod tests {
360-
use crate::logical_plan::{case, col, lit};
360+
use crate::logical_expr::{case, col, lit};
361361
use crate::test::object_store::make_test_store;
362362
use futures::StreamExt;
363363

datafusion/core/src/datasource/listing/table.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ use crate::logical_expr::TableProviderFilterPushDown;
4242
use crate::{
4343
error::{DataFusionError, Result},
4444
execution::context::SessionState,
45-
logical_plan::Expr,
45+
logical_expr::Expr,
4646
physical_plan::{
4747
empty::EmptyExec,
4848
file_format::{FileScanConfig, DEFAULT_PARTITION_COLUMN_DATATYPE},
@@ -499,7 +499,7 @@ mod tests {
499499
use crate::prelude::SessionContext;
500500
use crate::{
501501
datasource::file_format::{avro::AvroFormat, parquet::ParquetFormat},
502-
logical_plan::{col, lit},
502+
logical_expr::{col, lit},
503503
test::{columns, object_store::register_test_store},
504504
};
505505
use arrow::datatypes::DataType;

datafusion/core/src/datasource/memory.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ use async_trait::async_trait;
3030
use crate::datasource::{TableProvider, TableType};
3131
use crate::error::{DataFusionError, Result};
3232
use crate::execution::context::{SessionState, TaskContext};
33-
use crate::logical_plan::Expr;
33+
use crate::logical_expr::Expr;
3434
use crate::physical_plan::common;
3535
use crate::physical_plan::memory::MemoryExec;
3636
use crate::physical_plan::ExecutionPlan;

datafusion/core/src/datasource/view.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ use datafusion_expr::LogicalPlanBuilder;
2525

2626
use crate::{
2727
error::Result,
28-
logical_plan::{Expr, LogicalPlan},
28+
logical_expr::{Expr, LogicalPlan},
2929
physical_plan::ExecutionPlan,
3030
};
3131

datafusion/core/src/execution/context.rs

Lines changed: 14 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ use crate::{
2929
},
3030
MemTable, ViewTable,
3131
},
32-
logical_plan::{PlanType, ToStringifiedPlan},
32+
logical_expr::{PlanType, ToStringifiedPlan},
3333
optimizer::optimizer::Optimizer,
3434
physical_optimizer::{
3535
aggregate_statistics::AggregateStatistics,
@@ -59,13 +59,15 @@ use crate::catalog::{
5959
schema::{MemorySchemaProvider, SchemaProvider},
6060
};
6161
use crate::dataframe::DataFrame;
62-
use crate::datasource::listing::{ListingTableConfig, ListingTableUrl};
63-
use crate::datasource::TableProvider;
62+
use crate::datasource::{
63+
listing::{ListingTableConfig, ListingTableUrl},
64+
provider_as_source, TableProvider,
65+
};
6466
use crate::error::{DataFusionError, Result};
65-
use crate::logical_plan::{
66-
provider_as_source, CreateCatalog, CreateCatalogSchema, CreateExternalTable,
67-
CreateMemoryTable, CreateView, DropTable, FunctionRegistry, LogicalPlan,
68-
LogicalPlanBuilder, UNNAMED_TABLE,
67+
use crate::logical_expr::{
68+
CreateCatalog, CreateCatalogSchema, CreateExternalTable, CreateMemoryTable,
69+
CreateView, DropTable, DropView, Explain, LogicalPlan, LogicalPlanBuilder,
70+
TableSource, TableType, UNNAMED_TABLE,
6971
};
7072
use crate::optimizer::optimizer::{OptimizerConfig, OptimizerRule};
7173
use datafusion_sql::{ResolvedTableReference, TableReference};
@@ -80,8 +82,7 @@ use crate::config::{
8082
};
8183
use crate::datasource::datasource::TableProviderFactory;
8284
use crate::datasource::file_format::file_type::{FileCompressionType, FileType};
83-
use crate::execution::runtime_env::RuntimeEnv;
84-
use crate::logical_expr::Explain;
85+
use crate::execution::{runtime_env::RuntimeEnv, FunctionRegistry};
8586
use crate::physical_plan::file_format::{plan_to_csv, plan_to_json, plan_to_parquet};
8687
use crate::physical_plan::planner::DefaultPhysicalPlanner;
8788
use crate::physical_plan::udaf::AggregateUDF;
@@ -92,8 +93,6 @@ use crate::variable::{VarProvider, VarType};
9293
use async_trait::async_trait;
9394
use chrono::{DateTime, Utc};
9495
use datafusion_common::ScalarValue;
95-
use datafusion_expr::logical_plan::DropView;
96-
use datafusion_expr::{TableSource, TableType};
9796
use datafusion_sql::{
9897
parser::DFParser,
9998
planner::{ContextProvider, SqlToRel},
@@ -1868,20 +1867,17 @@ impl FunctionRegistry for TaskContext {
18681867
#[cfg(test)]
18691868
mod tests {
18701869
use super::*;
1870+
use crate::assert_batches_eq;
18711871
use crate::execution::context::QueryPlanner;
1872+
use crate::physical_plan::expressions::AvgAccumulator;
18721873
use crate::test;
18731874
use crate::test_util::parquet_test_data;
18741875
use crate::variable::VarType;
1875-
use crate::{
1876-
assert_batches_eq,
1877-
logical_plan::{create_udf, Expr},
1878-
};
1879-
use crate::{logical_plan::create_udaf, physical_plan::expressions::AvgAccumulator};
18801876
use arrow::array::ArrayRef;
18811877
use arrow::datatypes::*;
18821878
use arrow::record_batch::RecordBatch;
18831879
use async_trait::async_trait;
1884-
use datafusion_expr::Volatility;
1880+
use datafusion_expr::{create_udaf, create_udf, Expr, Volatility};
18851881
use datafusion_physical_expr::functions::make_scalar_function;
18861882
use std::fs::File;
18871883
use std::sync::Weak;
@@ -2398,7 +2394,7 @@ mod tests {
23982394
fn create_physical_expr(
23992395
&self,
24002396
_expr: &Expr,
2401-
_input_dfschema: &crate::logical_plan::DFSchema,
2397+
_input_dfschema: &crate::common::DFSchema,
24022398
_input_schema: &Schema,
24032399
_session_state: &SessionState,
24042400
) -> Result<Arc<dyn crate::physical_plan::PhysicalExpr>> {

datafusion/core/src/lib.rs

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -221,10 +221,6 @@ pub mod dataframe;
221221
pub mod datasource;
222222
pub mod error;
223223
pub mod execution;
224-
#[deprecated]
225-
// logical_plan module just contains re-exports and will be removed in a future release
226-
// https://github.com/apache/arrow-datafusion/issues/2683
227-
pub mod logical_plan;
228224
pub mod physical_optimizer;
229225
pub mod physical_plan;
230226
pub mod prelude;

datafusion/core/src/logical_plan/expr.rs

Lines changed: 0 additions & 21 deletions
This file was deleted.

datafusion/core/src/logical_plan/mod.rs

Lines changed: 0 additions & 57 deletions
This file was deleted.

datafusion/core/src/logical_plan/window_frames.rs

Lines changed: 0 additions & 20 deletions
This file was deleted.

datafusion/core/src/physical_optimizer/aggregate_statistics.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -265,7 +265,7 @@ mod tests {
265265
use datafusion_physical_expr::PhysicalExpr;
266266

267267
use crate::error::Result;
268-
use crate::logical_plan::Operator;
268+
use crate::logical_expr::Operator;
269269
use crate::physical_plan::aggregates::{AggregateExec, PhysicalGroupBy};
270270
use crate::physical_plan::coalesce_partitions::CoalescePartitionsExec;
271271
use crate::physical_plan::common;

0 commit comments

Comments
 (0)