apache / datafusion

Apache DataFusion SQL Query Engine
https://datafusion.apache.org/
Apache License 2.0
5.91k stars 1.12k forks source link

Support StringView for binary operators like `~`, `!~`, etc #12180

Open alamb opened 2 weeks ago

alamb commented 2 weeks ago

Is your feature request related to a problem or challenge?

Part of https://github.com/apache/datafusion/issues/11752

As we work to complete StringView support in DataFusion @2010YOUY01 noticed on https://github.com/apache/datafusion/issues/11752#issuecomment-2308176932 that we don't currently support Regexp like binary operators https://datafusion.apache.org/user-guide/sql/operators.html#op-re-match for string view

Reproducer

CREATE TABLE t0(v0 DOUBLE, v1 DOUBLE, v2 BOOLEAN, v3 BOOLEAN, v4 BOOLEAN, v5 STRING);
INSERT INTO t0(v1, v5, v2) VALUES (0.7183242196192607, 'Tn', true);
CREATE TABLE t0_stringview AS SELECT v0, v1, v2, v3, v4, arrow_cast(v5, 'Utf8View') as v5 FROM t0;
> select v5 ~ 'foo' from t0_stringview;
Internal error: Data type Utf8View not supported for binary_string_array_flag_op_scalar operation 'regexp_is_match' on string array.
This was likely caused by a bug in DataFusion's code and we would welcome that you file an bug report in our issue tracker
> select regexp_match(v5, 'foo') from t0_stringview;
+--------------------------------------------+
| regexp_match(t0_stringview.v5,Utf8("foo")) |
+--------------------------------------------+
|                                            |
+--------------------------------------------+
1 row(s) fetched.
Elapsed 0.034 seconds.

Describe the solution you'd like

StringView should be supported for these operators (aka the query should run without error)

Describe alternatives you've considered

Here are the relevant operator names:

            | Operator::RegexMatch
            | Operator::RegexIMatch
            | Operator::RegexNotMatch
            | Operator::RegexNotIMatch
            | Operator::LikeMatch
            | Operator::ILikeMatch
            | Operator::NotLikeMatch
            | Operator::NotILikeMatch

Here is the dispatch code:

https://github.com/apache/datafusion/blob/0f96af5b500efff72314f840a59a736787cc3def/datafusion/physical-expr/src/expressions/binary.rs#L621-L632

It appears that the corresponding arrow-rs kernel does not yet have support for StringView https://docs.rs/arrow-string/52.2.0/src/arrow_string/regexp.rs.html#307-311

So what I would suggest is:

  1. Implement a PR in datafusion with coercion from Utf8View --> Utf8 (aka cast arguments back to string)
  2. File an upstream ticket in arrow-rs for supporting string view with the regexp_like kernels and leave a link to that ticket in the datafusion code

Additional context

No response

tlm365 commented 2 weeks ago

Looks like #12168 also might be a starting point to solve this problem. I also think about solution 2 when handling that PR, and IMO solution 2 is the better option. I will take this one.

File an upstream ticket in arrow-rs for supporting string view with the regexp_like kernels and leave a link to that ticket in the datafusion code

tlm365 commented 2 weeks ago

take

alamb commented 2 weeks ago

I also think about solution 2 when handling that PR, and IMO solution 2 is the better option.

I agree solution 2 is better -- it just will take longer as it needs two coordinated PRs. One thing we have done in the past is do the initial implementation in DataFusion, and then file a ticket / port the code upstream to arrow-rs. Once the code is release in arrow-rs and available in Datafusion we remove the copy in DataFusion