datafusion icon indicating copy to clipboard operation
datafusion copied to clipboard

Remove `Expr::GetIndexedField` and `GetFieldAccess` and always use function `get_field` for indexing

Open jayzhan211 opened this issue 1 year ago • 2 comments

Is your feature request related to a problem or challenge?

Discussion starts from https://github.com/apache/datafusion/issues/10181#issuecomment-2071274206

Describe the solution you'd like

As title mentioned, use function instead of Expr

Describe alternatives you've considered

No response

Additional context

No response

jayzhan211 avatar May 04 '24 05:05 jayzhan211

One potential downside as @westonpace mentioned somewhere I can't find now, is that systems that want to look for field accesses for their own analysis (e.g. to find all nested field references) would find it harder to do so

alamb avatar May 08 '24 17:05 alamb

I think it would actually be easier. The easiest was the old way where we have only Expr::GetIndexedField and get_field doesn't exist (no functions). However, that ship has already sailed (the functions are valuable because they allow extensibility).

Today, I have to account for both paths. We use the SQL parser as our input. I don't know (off the top of my head) if it uses Expr or get_field and it doesn't really matter because, if I don't account for both, it will inevitably change which one it uses (murphy's law).

So having one path, even if it is a slightly less obvious path, is better than having two paths.

westonpace avatar May 08 '24 18:05 westonpace

@alamb Do you think we should also rewrite the array operator to function in parser step? It is currently in optimizer step. I think the downside of moving array rewrite in parser step is that if any user expects different array function with the same syntax, then they can't do it since we don't have "user-defined" parser mechanism now. But the benefit is that we can eliminate intermediate binary expression.

Have you thought about "user-defined" parser idea before, the way that user can define their own expression to get from the syntax? Is it useful in production? 🤔

jayzhan211 avatar May 14 '24 13:05 jayzhan211

@alamb Do you think we should also rewrite the array operator to function in parser step? It is currently in optimizer step. I think the downside of moving array rewrite in parser step is that if any user expects different array function with the same syntax, then they can't do it since we don't have "user-defined" parser mechanism now. But the benefit is that we can eliminate intermediate binary expression.

I agree that changing the parser to insert a call to get field access directly is a good idea (and would be consistent and allow us to remove Expr::GetFieldAccess

Have you thought about "user-defined" parser idea before, the way that user can define their own expression to get from the syntax? Is it useful in production? 🤔

One thing I have thought about is changing the hard coded lookup of function names from a pattern like this

https://github.com/apache/datafusion/blob/fc34dacdb9842cde4d056d5a659796ede4ae5e74/datafusion/sql/src/expr/value.rs#L144-L150

To be something more structured

pub trait PlannerFunctions {
  /// return the UDF to use for creating arrays ("make_array") by default:
  fn make_array(&self) -> Result<ScalarUDF>;
...
  // similar functions for other built in functions
}

And then instead of

 if let Some(udf) = self.context_provider.get_function_meta("make_array") { 
     Ok(Expr::ScalarFunction(ScalarFunction::new_udf(udf, values))) 
 } else { 
     not_impl_err!( 
         "array_expression featrue is disable, So should implement make_array UDF by yourself" 
     ) 
 } 

The planner might look like

  let udf = self.planner_functions.make_array()?;
  Ok(Expr::ScalarFunction(ScalarFunction::new_udf(udf, values))) 

But I haven't had a usecase to do that myself yet

alamb avatar May 14 '24 19:05 alamb

@alamb Do you think we should also rewrite the array operator to function in parser step? It is currently in optimizer step. I think the downside of moving array rewrite in parser step is that if any user expects different array function with the same syntax, then they can't do it since we don't have "user-defined" parser mechanism now. But the benefit is that we can eliminate intermediate binary expression.

The array operator to function is syntax like array1 || array2 -> array_concat, which is in ArrayFunctionRewriter now, so I'm thinking about whether we should move this to the parser or not.

jayzhan211 avatar May 14 '24 23:05 jayzhan211

@alamb Do you think we should also rewrite the array operator to function in parser step? It is currently in optimizer step. I think the downside of moving array rewrite in parser step is that if any user expects different array function with the same syntax, then they can't do it since we don't have "user-defined" parser mechanism now. But the benefit is that we can eliminate intermediate binary expression.

The array operator to function is syntax like array1 || array2 -> array_concat, which is in ArrayFunctionRewriter now, so I'm thinking about whether we should move this to the parser or not.

Let's move the discussion to https://github.com/apache/datafusion/issues/10534

alamb avatar May 15 '24 18:05 alamb