feat: support RightAnti for SortMergeJoin (#13680)

* feat: support `RightAnti` for `SortMergeJoin`
apache · Jan 5, 2025 · 69687da · 69687da
1 parent 42dce6e
commit 69687da
Show file tree

Hide file tree

Showing 4 changed files with 643 additions and 163 deletions.
diff --git a/datafusion/core/tests/fuzz_cases/join_fuzz.rs b/datafusion/core/tests/fuzz_cases/join_fuzz.rs
@@ -210,7 +210,7 @@ async fn test_semi_join_1k_filtered() {
 }
 
 #[tokio::test]
-async fn test_anti_join_1k() {
+async fn test_left_anti_join_1k() {
     JoinFuzzTestCase::new(
         make_staggered_batches(1000),
         make_staggered_batches(1000),
@@ -222,7 +222,7 @@ async fn test_anti_join_1k() {
 }
 
 #[tokio::test]
-async fn test_anti_join_1k_filtered() {
+async fn test_left_anti_join_1k_filtered() {
     JoinFuzzTestCase::new(
         make_staggered_batches(1000),
         make_staggered_batches(1000),
@@ -233,6 +233,30 @@ async fn test_anti_join_1k_filtered() {
     .await
 }
 
+#[tokio::test]
+async fn test_right_anti_join_1k() {
+    JoinFuzzTestCase::new(
+        make_staggered_batches(1000),
+        make_staggered_batches(1000),
+        JoinType::RightAnti,
+        None,
+    )
+    .run_test(&[HjSmj, NljHj], false)
+    .await
+}
+
+#[tokio::test]
+async fn test_right_anti_join_1k_filtered() {
+    JoinFuzzTestCase::new(
+        make_staggered_batches(1000),
+        make_staggered_batches(1000),
+        JoinType::RightAnti,
+        Some(Box::new(col_lt_col_filter)),
+    )
+    .run_test(&[HjSmj, NljHj], false)
+    .await
+}
+
 #[tokio::test]
 async fn test_left_mark_join_1k() {
     JoinFuzzTestCase::new(