Refine

apache · May 21, 2024 · ee5952e · ee5952e
1 parent d4d61b1
commit ee5952e
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/spark/src/main/scala/org/apache/comet/serde/QueryPlanSerde.scala b/spark/src/main/scala/org/apache/comet/serde/QueryPlanSerde.scala
@@ -2417,10 +2417,12 @@ object QueryPlanSerde extends Logging with ShimQueryPlanSerde with CometExprShim
           return None
         }
 
-        if (join.buildSide == BuildRight && join.joinType == LeftAnti) {
-          // DataFusion HashJoin LeftAnti has bugs on null keys.
-          withInfo(join, "BuildRight with LeftAnti is not supported")
-          return None
+        join match {
+          case b: BroadcastHashJoinExec if b.isNullAwareAntiJoin =>
+            // DataFusion HashJoin LeftAnti has bugs on null keys.
+            withInfo(join, "DataFusion doesn't support null-aware anti join")
+            return None
+          case _ => // no-op
         }
 
         val condition = join.condition.map { cond =>