Skip to content

Commit 4ce49fe

Browse files
committed
reduce traversal
Signed-off-by: Adam Gutglick <adamgsal@gmail.com>
1 parent 968a072 commit 4ce49fe

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

datafusion/datasource-parquet/src/row_filter.rs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,6 @@ use arrow::error::{ArrowError, Result as ArrowResult};
7474
use arrow::record_batch::RecordBatch;
7575
use datafusion_functions::core::file_row_index::FileRowIndexFunc;
7676
use datafusion_functions::core::getfield::GetFieldFunc;
77-
use datafusion_physical_expr_adapter::expr_references_scalar_udf;
7877
use parquet::arrow::ProjectionMask;
7978
use parquet::arrow::arrow_reader::{ArrowPredicate, RowFilter};
8079
use parquet::file::metadata::ParquetMetaData;
@@ -490,7 +489,9 @@ impl TreeNodeVisitor<'_> for PushdownChecker<'_> {
490489
return Ok(recursion);
491490
}
492491

493-
if expr_references_scalar_udf::<FileRowIndexFunc>(node) {
492+
if ScalarFunctionExpr::try_downcast_func::<FileRowIndexFunc>(node.as_ref())
493+
.is_some()
494+
{
494495
self.has_unpushable_udfs = true;
495496
return Ok(TreeNodeRecursion::Jump);
496497
}

0 commit comments

Comments
 (0)