Support all equality predicates in equality join (#4193)

* Support normal expressions in equality join * Add some tests * Improve comment * Update datafusion/sql/src/planner.rs Co-authored-by: Andrew Lamb <[email protected]> * Add another test case * Add comment * fix typo * remove dumplicated keys in projection and add test cases * fix comment * fix typo Co-authored-by: Andrew Lamb <[email protected]>
apache · Nov 17, 2022 · 822022d · 822022d
1 parent 062acad
commit 822022d
Show file tree

Hide file tree

Showing 2 changed files with 353 additions and 42 deletions.
diff --git a/datafusion/expr/src/utils.rs b/datafusion/expr/src/utils.rs
@@ -771,6 +771,18 @@ pub fn can_hash(data_type: &DataType) -> bool {
     }
 }
 
+/// Check whether all columns are from the schema.
+pub fn check_all_column_from_schema(
+    columns: &HashSet<Column>,
+    schema: DFSchemaRef,
+) -> Result<bool> {
+    let result = columns
+        .iter()
+        .all(|column| schema.index_of_column(column).is_ok());
+
+    Ok(result)
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;