KE-37052 translate boolean column to V2Predicate (apache#477)

RolatZhang · Jun 7, 2022 · 7796f19 · 7796f19
1 parent 0553594
commit 7796f19
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/catalyst/util/V2ExpressionBuilder.scala b/sql/core/src/main/scala/org/apache/spark/sql/catalyst/util/V2ExpressionBuilder.scala
@@ -56,8 +56,12 @@ class V2ExpressionBuilder(
       } else {
         Some(FieldReference(name))
       }
-    case pushableColumn(name) if !nestedPredicatePushdownEnabled =>
-      Some(FieldReference(name))
+    case col @ pushableColumn(name) if !nestedPredicatePushdownEnabled =>
+      if (isPredicate && col.dataType.isInstanceOf[BooleanType]) {
+        Some(new V2Predicate("=", Array(FieldReference(name), LiteralValue(true, BooleanType))))
+      } else {
+        Some(FieldReference(name))
+      }
     case in @ InSet(child, hset) =>
       generateExpression(child).map { v =>
         val children =