feat[expr-common]: support regex and LIKE coercion on REE<Dict> types

The coerce_fn applied to REE values needs to be the higher-level coerce
function so that any REE value can be coerced (not just primitive types).

Signed-off-by: Alfonso Subiotto Marques <alfonso.subiotto@polarsignals.com>

Author: asubiotto

datafusion/expr-common/src/type_coercion/binary.rs

@@ -1767,7 +1767,7 @@ pub fn like_coercion(lhs_type: &DataType, rhs_type: &DataType) -> Option<DataTyp
     string_coercion(lhs_type, rhs_type)
         .or_else(|| binary_to_string_coercion(lhs_type, rhs_type))
         .or_else(|| dictionary_coercion(lhs_type, rhs_type, false, string_coercion))
-        .or_else(|| ree_coercion(lhs_type, rhs_type, false, string_coercion))
+        .or_else(|| ree_coercion(lhs_type, rhs_type, false, like_coercion))
         .or_else(|| regex_null_coercion(lhs_type, rhs_type))
         .or_else(|| null_coercion(lhs_type, rhs_type))
 }
@@ -1788,7 +1788,7 @@ fn regex_null_coercion(lhs_type: &DataType, rhs_type: &DataType) -> Option<DataT
 pub fn regex_coercion(lhs_type: &DataType, rhs_type: &DataType) -> Option<DataType> {
     string_coercion(lhs_type, rhs_type)
         .or_else(|| dictionary_coercion(lhs_type, rhs_type, false, string_coercion))
-        .or_else(|| ree_coercion(lhs_type, rhs_type, false, string_coercion))
+        .or_else(|| ree_coercion(lhs_type, rhs_type, false, regex_coercion))
         .or_else(|| regex_null_coercion(lhs_type, rhs_type))
 }
 

datafusion/sqllogictest/test_files/run_end_encoded.slt

@@ -67,3 +67,21 @@ FROM sensor_readings;
 23 23
 24 24
 25 25
+
+# Regex / LIKE match against an REE column whose values are themselves Dictionary-encoded
+query BB rowsort
+SELECT
+    arrow_cast(
+        arrow_cast(sensor_id, 'Dictionary(UInt32, Utf8)'),
+        'RunEndEncoded("run_ends": non-null Int32, "values": Dictionary(UInt32, Utf8))'
+    ) ~ 'sensor_A',
+    arrow_cast(
+        arrow_cast(sensor_id, 'Dictionary(UInt32, Utf8)'),
+        'RunEndEncoded("run_ends": non-null Int32, "values": Dictionary(UInt32, Utf8))'
+    ) LIKE 'sensor_A'
+FROM sensor_readings;
+----
+false false
+true true
+true true
+true true