[GitHub] [spark] tgravescs commented on a change in pull request #30504: [SPARK-33544][SQL] Optimizer should not insert filter when explode with CreateArray/CreateMap

[GitBox <gi...@apache.org>]

tgravescs commented on a change in pull request #30504:
URL: https://github.com/apache/spark/pull/30504#discussion_r530614422



##########
File path: sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/optimizer/Optimizer.scala
##########
@@ -873,24 +873,30 @@ object InferFiltersFromGenerate extends Rule[LogicalPlan] {
       if !e.deterministic || e.children.forall(_.foldable) => generate
 
     case generate @ Generate(g, _, false, _, _, _) if canInferFilters(g) =>
-      // Exclude child's constraints to guarantee idempotency
-      val inferredFilters = ExpressionSet(
-        Seq(
-          GreaterThan(Size(g.children.head), Literal(0)),
-          IsNotNull(g.children.head)
-        )
-      ) -- generate.child.constraints
-
-      if (inferredFilters.nonEmpty) {
-        generate.copy(child = Filter(inferredFilters.reduce(And), generate.child))
-      } else {
-        generate
+      g.children.head match {
+        case _: CreateNonEmptyNonNullCollection =>
+          // we don't need to add filters when creating an array because we know its size
+          // is > 0 and its not null
+          generate
+        case _ =>
+          // Exclude child's constraints to guarantee idempotency
+          val inferredFilters = ExpressionSet(
+            Seq(
+              GreaterThan(Size(g.children.head), Literal(0)),
+              IsNotNull(g.children.head)

Review comment:
       @cloud-fan, sorry, I'm not following what exactly you are suggesting, could you clarify?  Are you saying creating a new Rule outside of InferFiltersFromGenerate? Are you somehow saying just change it so that inferredFilters come out to be empty for the createArray/Map case?




----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org



---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscribe@spark.apache.org
For additional commands, e-mail: reviews-help@spark.apache.org