fix:using simplifyOptimizerRules wasn't RemoveEmptyUnionBranches

Author: Max-Cheng

core/trino-main/src/main/java/io/trino/sql/planner/PlanOptimizers.java

@@ -963,6 +963,7 @@ public PlanOptimizers(
                         .add(new PushFilterIntoValues(plannerContext))
                         .add(new ReplaceJoinOverConstantWithProject())
                         .add(new RemoveRedundantPredicateAboveTableScan(plannerContext))
+                        .add(new RemoveEmptyUnionBranches())
                         .build()));
         // Remove unsupported dynamic filters introduced by PredicatePushdown. Also, cleanup dynamic filters removed by
         // PushPredicateIntoTableScan and RemoveRedundantPredicateAboveTableScan due to those rules replacing table scans with empty ValuesNode

core/trino-main/src/test/java/io/trino/sql/planner/TestLogicalPlanner.java

@@ -65,6 +65,7 @@
 import io.trino.sql.planner.plan.IndexJoinNode;
 import io.trino.sql.planner.plan.JoinNode;
 import io.trino.sql.planner.plan.LimitNode;
+import io.trino.sql.planner.plan.OutputNode;
 import io.trino.sql.planner.plan.PlanNode;
 import io.trino.sql.planner.plan.SemiJoinNode;
 import io.trino.sql.planner.plan.SemiJoinNode.DistributionType;
@@ -1332,6 +1333,31 @@ public void testRemovesTrivialFilters()
                         values("nationkey", "name", "regionkey", "comment")));
     }
 
+    @Test
+    public void testRemoveTrivialFiltersWithPartitioningHandle()
+    {
+        assertThat(countOfMatchingNodes(
+                plan("SELECT * FROM (" +
+                        "  SELECT " +
+                        "    n.name AS \"col1\"," +
+                        "    n.nationkey AS \"col2\"," +
+                        "    CAST(null AS varchar) AS \"col3\"" +
+                        "  FROM nation n" +
+                        "  WHERE n.nationkey <= 3" +
+                        "  UNION ALL" +
+                        "  SELECT " +
+                        "    r.name AS \"col1\", " +
+                        "    CAST(r.regionkey AS bigint) AS \"col2\"," +
+                        "    r.comment AS \"col3\"" +
+                        "  FROM region r" +
+                        ") subquery" +
+                        " WHERE \"col3\" IN (" +
+                        "  SELECT r.comment " +
+                        "  FROM region r" +
+                        ")"),
+                ValuesNode.class::isInstance)).isEqualTo(0);
+    }
+
     @Test
     public void testRemovesNullFilter()
     {