]> granicus.if.org Git - postgresql/commitdiff
Support flattening of empty-FROM subqueries and one-row VALUES tables.
authorTom Lane <tgl@sss.pgh.pa.us>
Thu, 12 Mar 2015 03:18:03 +0000 (23:18 -0400)
committerTom Lane <tgl@sss.pgh.pa.us>
Thu, 12 Mar 2015 03:18:03 +0000 (23:18 -0400)
We can't handle this in the general case due to limitations of the
planner's data representations; but we can allow it in many useful cases,
by being careful to flatten only when we are pulling a single-row subquery
up into a FROM (or, equivalently, inner JOIN) node that will still have at
least one remaining relation child.  Per discussion of an example from
Kyotaro Horiguchi.

src/backend/nodes/outfuncs.c
src/backend/optimizer/plan/planner.c
src/backend/optimizer/prep/prepjointree.c
src/include/nodes/relation.h
src/include/optimizer/prep.h
src/test/regress/expected/join.out
src/test/regress/expected/rangefuncs.out
src/test/regress/sql/join.sql
src/test/regress/sql/rangefuncs.sql

index ca596efb5b9f0ce2bb93ffe0f0d723abe67c3647..fc418fcf9de7882437e1e7d0cb62a9a068ef1ac7 100644 (file)
@@ -1762,6 +1762,7 @@ _outPlannerInfo(StringInfo str, const PlannerInfo *node)
        WRITE_BOOL_FIELD(hasInheritedTarget);
        WRITE_BOOL_FIELD(hasJoinRTEs);
        WRITE_BOOL_FIELD(hasLateralRTEs);
+       WRITE_BOOL_FIELD(hasDeletedRTEs);
        WRITE_BOOL_FIELD(hasHavingQual);
        WRITE_BOOL_FIELD(hasPseudoConstantQuals);
        WRITE_BOOL_FIELD(hasRecursion);
index b02a1079ae4af7d0d93bcbf27087c4437fd26813..88b91f1b205a2f9ec4cc26a3dbf7e8f5c3a39582 100644 (file)
@@ -352,8 +352,7 @@ subquery_planner(PlannerGlobal *glob, Query *parse,
         * Check to see if any subqueries in the jointree can be merged into this
         * query.
         */
-       parse->jointree = (FromExpr *)
-               pull_up_subqueries(root, (Node *) parse->jointree);
+       pull_up_subqueries(root);
 
        /*
         * If this is a simple UNION ALL query, flatten it into an appendrel. We
index 8a0199ba9368d36d10871b5ce0fb9903c5476ff5..50acfe40e9711019a98cb89cc99802de27fa10b8 100644 (file)
@@ -65,12 +65,14 @@ static Node *pull_up_sublinks_qual_recurse(PlannerInfo *root, Node *node,
 static Node *pull_up_subqueries_recurse(PlannerInfo *root, Node *jtnode,
                                                   JoinExpr *lowest_outer_join,
                                                   JoinExpr *lowest_nulling_outer_join,
-                                                  AppendRelInfo *containing_appendrel);
+                                                  AppendRelInfo *containing_appendrel,
+                                                  bool deletion_ok);
 static Node *pull_up_simple_subquery(PlannerInfo *root, Node *jtnode,
                                                RangeTblEntry *rte,
                                                JoinExpr *lowest_outer_join,
                                                JoinExpr *lowest_nulling_outer_join,
-                                               AppendRelInfo *containing_appendrel);
+                                               AppendRelInfo *containing_appendrel,
+                                               bool deletion_ok);
 static Node *pull_up_simple_union_all(PlannerInfo *root, Node *jtnode,
                                                 RangeTblEntry *rte);
 static void pull_up_union_leaf_queries(Node *setOp, PlannerInfo *root,
@@ -79,7 +81,12 @@ static void pull_up_union_leaf_queries(Node *setOp, PlannerInfo *root,
 static void make_setop_translation_list(Query *query, Index newvarno,
                                                        List **translated_vars);
 static bool is_simple_subquery(Query *subquery, RangeTblEntry *rte,
-                                  JoinExpr *lowest_outer_join);
+                                  JoinExpr *lowest_outer_join,
+                                  bool deletion_ok);
+static Node *pull_up_simple_values(PlannerInfo *root, Node *jtnode,
+                                         RangeTblEntry *rte);
+static bool is_simple_values(PlannerInfo *root, RangeTblEntry *rte,
+                                bool deletion_ok);
 static bool is_simple_union_all(Query *subquery);
 static bool is_simple_union_all_recurse(Node *setOp, Query *setOpQuery,
                                                        List *colTypes);
@@ -95,6 +102,7 @@ static Node *pullup_replace_vars_callback(Var *var,
                                                         replace_rte_variables_context *context);
 static Query *pullup_replace_vars_subquery(Query *query,
                                                         pullup_replace_vars_context *context);
+static Node *pull_up_subqueries_cleanup(Node *jtnode);
 static reduce_outer_joins_state *reduce_outer_joins_pass1(Node *jtnode);
 static void reduce_outer_joins_pass2(Node *jtnode,
                                                 reduce_outer_joins_state *state,
@@ -593,20 +601,33 @@ inline_set_returning_functions(PlannerInfo *root)
  *             grouping/aggregation then we can merge it into the parent's jointree.
  *             Also, subqueries that are simple UNION ALL structures can be
  *             converted into "append relations".
- *
- * This recursively processes the jointree and returns a modified jointree.
  */
-Node *
-pull_up_subqueries(PlannerInfo *root, Node *jtnode)
+void
+pull_up_subqueries(PlannerInfo *root)
 {
-       /* Start off with no containing join nor appendrel */
-       return pull_up_subqueries_recurse(root, jtnode, NULL, NULL, NULL);
+       /* Top level of jointree must always be a FromExpr */
+       Assert(IsA(root->parse->jointree, FromExpr));
+       /* Reset flag saying we need a deletion cleanup pass */
+       root->hasDeletedRTEs = false;
+       /* Recursion starts with no containing join nor appendrel */
+       root->parse->jointree = (FromExpr *)
+               pull_up_subqueries_recurse(root, (Node *) root->parse->jointree,
+                                                                  NULL, NULL, NULL, false);
+       /* Apply cleanup phase if necessary */
+       if (root->hasDeletedRTEs)
+               root->parse->jointree = (FromExpr *)
+                       pull_up_subqueries_cleanup((Node *) root->parse->jointree);
+       Assert(IsA(root->parse->jointree, FromExpr));
 }
 
 /*
  * pull_up_subqueries_recurse
  *             Recursive guts of pull_up_subqueries.
  *
+ * This recursively processes the jointree and returns a modified jointree.
+ * Or, if it's valid to drop the current node from the jointree completely,
+ * it returns NULL.
+ *
  * If this jointree node is within either side of an outer join, then
  * lowest_outer_join references the lowest such JoinExpr node; otherwise
  * it is NULL.  We use this to constrain the effects of LATERAL subqueries.
@@ -622,28 +643,39 @@ pull_up_subqueries(PlannerInfo *root, Node *jtnode)
  * This forces use of the PlaceHolderVar mechanism for all non-Var targetlist
  * items, and puts some additional restrictions on what can be pulled up.
  *
+ * deletion_ok is TRUE if the caller can cope with us returning NULL for a
+ * deletable leaf node (for example, a VALUES RTE that could be pulled up).
+ * If it's FALSE, we'll avoid pullup in such cases.
+ *
  * A tricky aspect of this code is that if we pull up a subquery we have
  * to replace Vars that reference the subquery's outputs throughout the
  * parent query, including quals attached to jointree nodes above the one
  * we are currently processing!  We handle this by being careful not to
- * change the jointree structure while recursing: no nodes other than
- * subquery RangeTblRef entries will be replaced.  Also, we can't turn
- * pullup_replace_vars loose on the whole jointree, because it'll return a
- * mutated copy of the tree; we have to invoke it just on the quals, instead.
- * This behavior is what makes it reasonable to pass lowest_outer_join and
- * lowest_nulling_outer_join as pointers rather than some more-indirect way
- * of identifying the lowest OJs.  Likewise, we don't replace append_rel_list
- * members but only their substructure, so the containing_appendrel reference
- * is safe to use.
+ * change the jointree structure while recursing: no nodes other than leaf
+ * RangeTblRef entries and entirely-empty FromExprs will be replaced or
+ * deleted.  Also, we can't turn pullup_replace_vars loose on the whole
+ * jointree, because it'll return a mutated copy of the tree; we have to
+ * invoke it just on the quals, instead.  This behavior is what makes it
+ * reasonable to pass lowest_outer_join and lowest_nulling_outer_join as
+ * pointers rather than some more-indirect way of identifying the lowest
+ * OJs.  Likewise, we don't replace append_rel_list members but only their
+ * substructure, so the containing_appendrel reference is safe to use.
+ *
+ * Because of the rule that no jointree nodes with substructure can be
+ * replaced, we cannot fully handle the case of deleting nodes from the tree:
+ * when we delete one child of a JoinExpr, we need to replace the JoinExpr
+ * with a FromExpr, and that can't happen here.  Instead, we set the
+ * root->hasDeletedRTEs flag, which tells pull_up_subqueries() that an
+ * additional pass over the tree is needed to clean up.
  */
 static Node *
 pull_up_subqueries_recurse(PlannerInfo *root, Node *jtnode,
                                                   JoinExpr *lowest_outer_join,
                                                   JoinExpr *lowest_nulling_outer_join,
-                                                  AppendRelInfo *containing_appendrel)
+                                                  AppendRelInfo *containing_appendrel,
+                                                  bool deletion_ok)
 {
-       if (jtnode == NULL)
-               return NULL;
+       Assert(jtnode != NULL);
        if (IsA(jtnode, RangeTblRef))
        {
                int                     varno = ((RangeTblRef *) jtnode)->rtindex;
@@ -657,13 +689,15 @@ pull_up_subqueries_recurse(PlannerInfo *root, Node *jtnode,
                 * unless is_safe_append_member says so.
                 */
                if (rte->rtekind == RTE_SUBQUERY &&
-                       is_simple_subquery(rte->subquery, rte, lowest_outer_join) &&
+                       is_simple_subquery(rte->subquery, rte,
+                                                          lowest_outer_join, deletion_ok) &&
                        (containing_appendrel == NULL ||
                         is_safe_append_member(rte->subquery)))
                        return pull_up_simple_subquery(root, jtnode, rte,
                                                                                   lowest_outer_join,
                                                                                   lowest_nulling_outer_join,
-                                                                                  containing_appendrel);
+                                                                                  containing_appendrel,
+                                                                                  deletion_ok);
 
                /*
                 * Alternatively, is it a simple UNION ALL subquery?  If so, flatten
@@ -678,19 +712,68 @@ pull_up_subqueries_recurse(PlannerInfo *root, Node *jtnode,
                        is_simple_union_all(rte->subquery))
                        return pull_up_simple_union_all(root, jtnode, rte);
 
+               /*
+                * Or perhaps it's a simple VALUES RTE?
+                *
+                * We don't allow VALUES pullup below an outer join nor into an
+                * appendrel (such cases are impossible anyway at the moment).
+                */
+               if (rte->rtekind == RTE_VALUES &&
+                       lowest_outer_join == NULL &&
+                       containing_appendrel == NULL &&
+                       is_simple_values(root, rte, deletion_ok))
+                       return pull_up_simple_values(root, jtnode, rte);
+
                /* Otherwise, do nothing at this node. */
        }
        else if (IsA(jtnode, FromExpr))
        {
                FromExpr   *f = (FromExpr *) jtnode;
+               bool            have_undeleted_child = false;
                ListCell   *l;
 
                Assert(containing_appendrel == NULL);
+
+               /*
+                * If the FromExpr has quals, it's not deletable even if its parent
+                * would allow deletion.
+                */
+               if (f->quals)
+                       deletion_ok = false;
+
                foreach(l, f->fromlist)
+               {
+                       /*
+                        * In a non-deletable FromExpr, we can allow deletion of child
+                        * nodes so long as at least one child remains; so it's okay
+                        * either if any previous child survives, or if there's more to
+                        * come.  If all children are deletable in themselves, we'll force
+                        * the last one to remain unflattened.
+                        *
+                        * As a separate matter, we can allow deletion of all children of
+                        * the top-level FromExpr in a query, since that's a special case
+                        * anyway.
+                        */
+                       bool            sub_deletion_ok = (deletion_ok ||
+                                                                                  have_undeleted_child ||
+                                                                                  lnext(l) != NULL ||
+                                                                                  f == root->parse->jointree);
+
                        lfirst(l) = pull_up_subqueries_recurse(root, lfirst(l),
                                                                                                   lowest_outer_join,
                                                                                                   lowest_nulling_outer_join,
-                                                                                                  NULL);
+                                                                                                  NULL,
+                                                                                                  sub_deletion_ok);
+                       if (lfirst(l) != NULL)
+                               have_undeleted_child = true;
+               }
+
+               if (deletion_ok && !have_undeleted_child)
+               {
+                       /* OK to delete this FromExpr entirely */
+                       root->hasDeletedRTEs = true;            /* probably is set already */
+                       return NULL;
+               }
        }
        else if (IsA(jtnode, JoinExpr))
        {
@@ -701,14 +784,22 @@ pull_up_subqueries_recurse(PlannerInfo *root, Node *jtnode,
                switch (j->jointype)
                {
                        case JOIN_INNER:
+
+                               /*
+                                * INNER JOIN can allow deletion of either child node, but not
+                                * both.  So right child gets permission to delete only if
+                                * left child didn't get removed.
+                                */
                                j->larg = pull_up_subqueries_recurse(root, j->larg,
                                                                                                         lowest_outer_join,
                                                                                                   lowest_nulling_outer_join,
-                                                                                                        NULL);
+                                                                                                        NULL,
+                                                                                                        true);
                                j->rarg = pull_up_subqueries_recurse(root, j->rarg,
                                                                                                         lowest_outer_join,
                                                                                                   lowest_nulling_outer_join,
-                                                                                                        NULL);
+                                                                                                        NULL,
+                                                                                                        j->larg != NULL);
                                break;
                        case JOIN_LEFT:
                        case JOIN_SEMI:
@@ -716,31 +807,37 @@ pull_up_subqueries_recurse(PlannerInfo *root, Node *jtnode,
                                j->larg = pull_up_subqueries_recurse(root, j->larg,
                                                                                                         j,
                                                                                                   lowest_nulling_outer_join,
-                                                                                                        NULL);
+                                                                                                        NULL,
+                                                                                                        false);
                                j->rarg = pull_up_subqueries_recurse(root, j->rarg,
                                                                                                         j,
                                                                                                         j,
-                                                                                                        NULL);
+                                                                                                        NULL,
+                                                                                                        false);
                                break;
                        case JOIN_FULL:
                                j->larg = pull_up_subqueries_recurse(root, j->larg,
                                                                                                         j,
                                                                                                         j,
-                                                                                                        NULL);
+                                                                                                        NULL,
+                                                                                                        false);
                                j->rarg = pull_up_subqueries_recurse(root, j->rarg,
                                                                                                         j,
                                                                                                         j,
-                                                                                                        NULL);
+                                                                                                        NULL,
+                                                                                                        false);
                                break;
                        case JOIN_RIGHT:
                                j->larg = pull_up_subqueries_recurse(root, j->larg,
                                                                                                         j,
                                                                                                         j,
-                                                                                                        NULL);
+                                                                                                        NULL,
+                                                                                                        false);
                                j->rarg = pull_up_subqueries_recurse(root, j->rarg,
                                                                                                         j,
                                                                                                   lowest_nulling_outer_join,
-                                                                                                        NULL);
+                                                                                                        NULL,
+                                                                                                        false);
                                break;
                        default:
                                elog(ERROR, "unrecognized join type: %d",
@@ -760,8 +857,8 @@ pull_up_subqueries_recurse(PlannerInfo *root, Node *jtnode,
  *
  * jtnode is a RangeTblRef that has been tentatively identified as a simple
  * subquery by pull_up_subqueries.  We return the replacement jointree node,
- * or jtnode itself if we determine that the subquery can't be pulled up after
- * all.
+ * or NULL if the subquery can be deleted entirely, or jtnode itself if we
+ * determine that the subquery can't be pulled up after all.
  *
  * rte is the RangeTblEntry referenced by jtnode.  Remaining parameters are
  * as for pull_up_subqueries_recurse.
@@ -770,7 +867,8 @@ static Node *
 pull_up_simple_subquery(PlannerInfo *root, Node *jtnode, RangeTblEntry *rte,
                                                JoinExpr *lowest_outer_join,
                                                JoinExpr *lowest_nulling_outer_join,
-                                               AppendRelInfo *containing_appendrel)
+                                               AppendRelInfo *containing_appendrel,
+                                               bool deletion_ok)
 {
        Query      *parse = root->parse;
        int                     varno = ((RangeTblRef *) jtnode)->rtindex;
@@ -832,14 +930,12 @@ pull_up_simple_subquery(PlannerInfo *root, Node *jtnode, RangeTblEntry *rte,
         * pull_up_subqueries' processing is complete for its jointree and
         * rangetable.
         *
-        * Note: we should pass NULL for containing-join info even if we are
-        * within an outer join in the upper query; the lower query starts with a
-        * clean slate for outer-join semantics.  Likewise, we say we aren't
-        * handling an appendrel member.
+        * Note: it's okay that the subquery's recursion starts with NULL for
+        * containing-join info, even if we are within an outer join in the upper
+        * query; the lower query starts with a clean slate for outer-join
+        * semantics.  Likewise, we needn't pass down appendrel state.
         */
-       subquery->jointree = (FromExpr *)
-               pull_up_subqueries_recurse(subroot, (Node *) subquery->jointree,
-                                                                  NULL, NULL, NULL);
+       pull_up_subqueries(subroot);
 
        /*
         * Now we must recheck whether the subquery is still simple enough to pull
@@ -849,7 +945,8 @@ pull_up_simple_subquery(PlannerInfo *root, Node *jtnode, RangeTblEntry *rte,
         * easier just to keep this "if" looking the same as the one in
         * pull_up_subqueries_recurse.
         */
-       if (is_simple_subquery(subquery, rte, lowest_outer_join) &&
+       if (is_simple_subquery(subquery, rte,
+                                                  lowest_outer_join, deletion_ok) &&
                (containing_appendrel == NULL || is_safe_append_member(subquery)))
        {
                /* good to go */
@@ -1075,8 +1172,18 @@ pull_up_simple_subquery(PlannerInfo *root, Node *jtnode, RangeTblEntry *rte,
 
        /*
         * Return the adjusted subquery jointree to replace the RangeTblRef entry
-        * in parent's jointree.
+        * in parent's jointree; or, if we're flattening a subquery with empty
+        * FROM list, return NULL to signal deletion of the subquery from the
+        * parent jointree (and set hasDeletedRTEs to ensure cleanup later).
         */
+       if (subquery->jointree->fromlist == NIL)
+       {
+               Assert(deletion_ok);
+               Assert(subquery->jointree->quals == NULL);
+               root->hasDeletedRTEs = true;
+               return NULL;
+       }
+
        return (Node *) subquery->jointree;
 }
 
@@ -1203,12 +1310,15 @@ pull_up_union_leaf_queries(Node *setOp, PlannerInfo *root, int parentRTindex,
                 * must build the AppendRelInfo first, because this will modify it.)
                 * Note that we can pass NULL for containing-join info even if we're
                 * actually under an outer join, because the child's expressions
-                * aren't going to propagate up to the join.
+                * aren't going to propagate up to the join.  Also, we ignore the
+                * possibility that pull_up_subqueries_recurse() returns a different
+                * jointree node than what we pass it; if it does, the important thing
+                * is that it replaced the child relid in the AppendRelInfo node.
                 */
                rtr = makeNode(RangeTblRef);
                rtr->rtindex = childRTindex;
                (void) pull_up_subqueries_recurse(root, (Node *) rtr,
-                                                                                 NULL, NULL, appinfo);
+                                                                                 NULL, NULL, appinfo, false);
        }
        else if (IsA(setOp, SetOperationStmt))
        {
@@ -1263,10 +1373,12 @@ make_setop_translation_list(Query *query, Index newvarno,
  * (Note subquery is not necessarily equal to rte->subquery; it could be a
  * processed copy of that.)
  * lowest_outer_join is the lowest outer join above the subquery, or NULL.
+ * deletion_ok is TRUE if it'd be okay to delete the subquery entirely.
  */
 static bool
 is_simple_subquery(Query *subquery, RangeTblEntry *rte,
-                                  JoinExpr *lowest_outer_join)
+                                  JoinExpr *lowest_outer_join,
+                                  bool deletion_ok)
 {
        /*
         * Let's just make sure it's a valid subselect ...
@@ -1314,6 +1426,29 @@ is_simple_subquery(Query *subquery, RangeTblEntry *rte,
        if (rte->security_barrier)
                return false;
 
+       /*
+        * Don't pull up a subquery with an empty jointree, unless it has no quals
+        * and deletion_ok is TRUE.  query_planner() will correctly generate a
+        * Result plan for a jointree that's totally empty, but we can't cope with
+        * an empty FromExpr appearing lower down in a jointree: we identify join
+        * rels via baserelid sets, so we couldn't distinguish a join containing
+        * such a FromExpr from one without it.  This would for example break the
+        * PlaceHolderVar mechanism, since we'd have no way to identify where to
+        * evaluate a PHV coming out of the subquery.  We can only handle such
+        * cases if the place where the subquery is linked is a FromExpr or inner
+        * JOIN that would still be nonempty after removal of the subquery, so
+        * that it's still identifiable via its contained baserelids.  Safe
+        * contexts are signaled by deletion_ok.  But even in a safe context, we
+        * must keep the subquery if it has any quals, because it's unclear where
+        * to put them in the upper query.  (Note that deletion of a subquery is
+        * also dependent on the check below that its targetlist contains no
+        * set-returning functions.  Deletion from a FROM list or inner JOIN is
+        * okay only if the subquery must return exactly one row.)
+        */
+       if (subquery->jointree->fromlist == NIL &&
+               (subquery->jointree->quals || !deletion_ok))
+               return false;
+
        /*
         * If the subquery is LATERAL, check for pullup restrictions from that.
         */
@@ -1373,7 +1508,7 @@ is_simple_subquery(Query *subquery, RangeTblEntry *rte,
         * Don't pull up a subquery that has any set-returning functions in its
         * targetlist.  Otherwise we might well wind up inserting set-returning
         * functions into places where they mustn't go, such as quals of higher
-        * queries.
+        * queries.  This also ensures deletion of an empty jointree is valid.
         */
        if (expression_returns_set((Node *) subquery->targetList))
                return false;
@@ -1389,19 +1524,159 @@ is_simple_subquery(Query *subquery, RangeTblEntry *rte,
        if (contain_volatile_functions((Node *) subquery->targetList))
                return false;
 
+       return true;
+}
+
+/*
+ * pull_up_simple_values
+ *             Pull up a single simple VALUES RTE.
+ *
+ * jtnode is a RangeTblRef that has been identified as a simple VALUES RTE
+ * by pull_up_subqueries.  We always return NULL indicating that the RTE
+ * can be deleted entirely (all failure cases should have been detected by
+ * is_simple_values()).
+ *
+ * rte is the RangeTblEntry referenced by jtnode.  Because of the limited
+ * possible usage of VALUES RTEs, we do not need the remaining parameters
+ * of pull_up_subqueries_recurse.
+ */
+static Node *
+pull_up_simple_values(PlannerInfo *root, Node *jtnode, RangeTblEntry *rte)
+{
+       Query      *parse = root->parse;
+       int                     varno = ((RangeTblRef *) jtnode)->rtindex;
+       List       *values_list;
+       List       *tlist;
+       AttrNumber      attrno;
+       pullup_replace_vars_context rvcontext;
+       ListCell   *lc;
+
+       Assert(rte->rtekind == RTE_VALUES);
+       Assert(list_length(rte->values_lists) == 1);
+
        /*
-        * Don't pull up a subquery with an empty jointree.  query_planner() will
-        * correctly generate a Result plan for a jointree that's totally empty,
-        * but we can't cope with an empty FromExpr appearing lower down in a
-        * jointree: we identify join rels via baserelid sets, so we couldn't
-        * distinguish a join containing such a FromExpr from one without it. This
-        * would for example break the PlaceHolderVar mechanism, since we'd have
-        * no way to identify where to evaluate a PHV coming out of the subquery.
-        * Not worth working hard on this, just to collapse SubqueryScan/Result
-        * into Result; especially since the SubqueryScan can often be optimized
-        * away by setrefs.c anyway.
+        * Need a modifiable copy of the VALUES list to hack on, just in case it's
+        * multiply referenced.
         */
-       if (subquery->jointree->fromlist == NIL)
+       values_list = (List *) copyObject(linitial(rte->values_lists));
+
+       /*
+        * The VALUES RTE can't contain any Vars of level zero, let alone any that
+        * are join aliases, so no need to flatten join alias Vars.
+        */
+       Assert(!contain_vars_of_level((Node *) values_list, 0));
+
+       /*
+        * Set up required context data for pullup_replace_vars.  In particular,
+        * we have to make the VALUES list look like a subquery targetlist.
+        */
+       tlist = NIL;
+       attrno = 1;
+       foreach(lc, values_list)
+       {
+               tlist = lappend(tlist,
+                                               makeTargetEntry((Expr *) lfirst(lc),
+                                                                               attrno,
+                                                                               NULL,
+                                                                               false));
+               attrno++;
+       }
+       rvcontext.root = root;
+       rvcontext.targetlist = tlist;
+       rvcontext.target_rte = rte;
+       rvcontext.relids = NULL;
+       rvcontext.outer_hasSubLinks = &parse->hasSubLinks;
+       rvcontext.varno = varno;
+       rvcontext.need_phvs = false;
+       rvcontext.wrap_non_vars = false;
+       /* initialize cache array with indexes 0 .. length(tlist) */
+       rvcontext.rv_cache = palloc0((list_length(tlist) + 1) *
+                                                                sizeof(Node *));
+
+       /*
+        * Replace all of the top query's references to the RTE's outputs with
+        * copies of the adjusted VALUES expressions, being careful not to replace
+        * any of the jointree structure. (This'd be a lot cleaner if we could use
+        * query_tree_mutator.)  Much of this should be no-ops in the dummy Query
+        * that surrounds a VALUES RTE, but it's not enough code to be worth
+        * removing.
+        */
+       parse->targetList = (List *)
+               pullup_replace_vars((Node *) parse->targetList, &rvcontext);
+       parse->returningList = (List *)
+               pullup_replace_vars((Node *) parse->returningList, &rvcontext);
+       replace_vars_in_jointree((Node *) parse->jointree, &rvcontext, NULL);
+       Assert(parse->setOperations == NULL);
+       parse->havingQual = pullup_replace_vars(parse->havingQual, &rvcontext);
+
+       /*
+        * There should be no appendrels to fix, nor any join alias Vars, nor any
+        * outer joins and hence no PlaceHolderVars.
+        */
+       Assert(root->append_rel_list == NIL);
+       Assert(list_length(parse->rtable) == 1);
+       Assert(root->join_info_list == NIL);
+       Assert(root->lateral_info_list == NIL);
+       Assert(root->placeholder_list == NIL);
+
+       /*
+        * Return NULL to signal deletion of the VALUES RTE from the parent
+        * jointree (and set hasDeletedRTEs to ensure cleanup later).
+        */
+       root->hasDeletedRTEs = true;
+       return NULL;
+}
+
+/*
+ * is_simple_values
+ *       Check a VALUES RTE in the range table to see if it's simple enough
+ *       to pull up into the parent query.
+ *
+ * rte is the RTE_VALUES RangeTblEntry to check.
+ * deletion_ok is TRUE if it'd be okay to delete the VALUES RTE entirely.
+ */
+static bool
+is_simple_values(PlannerInfo *root, RangeTblEntry *rte, bool deletion_ok)
+{
+       Assert(rte->rtekind == RTE_VALUES);
+
+       /*
+        * We can only pull up a VALUES RTE if deletion_ok is TRUE.  It's
+        * basically the same case as a sub-select with empty FROM list; see
+        * comments in is_simple_subquery().
+        */
+       if (!deletion_ok)
+               return false;
+
+       /*
+        * Also, there must be exactly one VALUES list, else it's not semantically
+        * correct to delete the VALUES RTE.
+        */
+       if (list_length(rte->values_lists) != 1)
+               return false;
+
+       /*
+        * Because VALUES can't appear under an outer join (or at least, we won't
+        * try to pull it up if it does), we need not worry about LATERAL.
+        */
+
+       /*
+        * Don't pull up a VALUES that contains any set-returning or volatile
+        * functions.  Again, the considerations here are basically identical to
+        * restrictions on a subquery's targetlist.
+        */
+       if (expression_returns_set((Node *) rte->values_lists) ||
+               contain_volatile_functions((Node *) rte->values_lists))
+               return false;
+
+       /*
+        * Do not pull up a VALUES that's not the only RTE in its parent query.
+        * This is actually the only case that the parser will generate at the
+        * moment, and assuming this is true greatly simplifies
+        * pull_up_simple_values().
+        */
+       if (list_length(root->parse->rtable) != 1 ||
+               rte != (RangeTblEntry *) linitial(root->parse->rtable))
                return false;
 
        return true;
@@ -1909,6 +2184,65 @@ pullup_replace_vars_subquery(Query *query,
                                                                                   NULL);
 }
 
+/*
+ * pull_up_subqueries_cleanup
+ *             Recursively fix up jointree after deletion of some subqueries.
+ *
+ * The jointree now contains some NULL subtrees, which we need to get rid of.
+ * In a FromExpr, just rebuild the child-node list with null entries deleted.
+ * In an inner JOIN, replace the JoinExpr node with a one-child FromExpr.
+ */
+static Node *
+pull_up_subqueries_cleanup(Node *jtnode)
+{
+       Assert(jtnode != NULL);
+       if (IsA(jtnode, RangeTblRef))
+       {
+               /* Nothing to do at leaf nodes. */
+       }
+       else if (IsA(jtnode, FromExpr))
+       {
+               FromExpr   *f = (FromExpr *) jtnode;
+               List       *newfrom = NIL;
+               ListCell   *l;
+
+               foreach(l, f->fromlist)
+               {
+                       Node       *child = (Node *) lfirst(l);
+
+                       if (child == NULL)
+                               continue;
+                       child = pull_up_subqueries_cleanup(child);
+                       newfrom = lappend(newfrom, child);
+               }
+               f->fromlist = newfrom;
+       }
+       else if (IsA(jtnode, JoinExpr))
+       {
+               JoinExpr   *j = (JoinExpr *) jtnode;
+
+               if (j->larg)
+                       j->larg = pull_up_subqueries_cleanup(j->larg);
+               if (j->rarg)
+                       j->rarg = pull_up_subqueries_cleanup(j->rarg);
+               if (j->larg == NULL)
+               {
+                       Assert(j->jointype == JOIN_INNER);
+                       Assert(j->rarg != NULL);
+                       return (Node *) makeFromExpr(list_make1(j->rarg), j->quals);
+               }
+               else if (j->rarg == NULL)
+               {
+                       Assert(j->jointype == JOIN_INNER);
+                       return (Node *) makeFromExpr(list_make1(j->larg), j->quals);
+               }
+       }
+       else
+               elog(ERROR, "unrecognized node type: %d",
+                        (int) nodeTag(jtnode));
+       return jtnode;
+}
+
 
 /*
  * flatten_simple_union_all
index e6dd4e8515da5318a27f42852fd559f7ba77a141..334cf51976bce067c5b942dc125f916c08614482 100644 (file)
@@ -244,6 +244,7 @@ typedef struct PlannerInfo
                                                                                 * inheritance child rel */
        bool            hasJoinRTEs;    /* true if any RTEs are RTE_JOIN kind */
        bool            hasLateralRTEs; /* true if any RTEs are marked LATERAL */
+       bool            hasDeletedRTEs; /* true if any RTE was deleted from jointree */
        bool            hasHavingQual;  /* true if havingQual was non-null */
        bool            hasPseudoConstantQuals; /* true if any RestrictInfo has
                                                                                 * pseudoconstant = true */
index 385f4a866aefd6e0df3b4a0eba305445f2bcd4cb..05e46c5b7881568da1b17e36d7a946945ece1d0c 100644 (file)
@@ -23,7 +23,7 @@
  */
 extern void pull_up_sublinks(PlannerInfo *root);
 extern void inline_set_returning_functions(PlannerInfo *root);
-extern Node *pull_up_subqueries(PlannerInfo *root, Node *jtnode);
+extern void pull_up_subqueries(PlannerInfo *root);
 extern void flatten_simple_union_all(PlannerInfo *root);
 extern void reduce_outer_joins(PlannerInfo *root);
 extern Relids get_relids_in_jointree(Node *jtnode, bool include_joins);
index 8c17ef41b9204be3250f485e756af6bfd198267e..57fc910133be90b6ea94f2d2aceba75c38d25181 100644 (file)
@@ -3633,6 +3633,26 @@ select * from generate_series(100,200) g,
 explain (costs off)
   select count(*) from tenk1 a,
     tenk1 b join lateral (values(a.unique1)) ss(x) on b.unique2 = ss.x;
+                         QUERY PLAN                         
+------------------------------------------------------------
+ Aggregate
+   ->  Merge Join
+         Merge Cond: (a.unique1 = b.unique2)
+         ->  Index Only Scan using tenk1_unique1 on tenk1 a
+         ->  Index Only Scan using tenk1_unique2 on tenk1 b
+(5 rows)
+
+select count(*) from tenk1 a,
+  tenk1 b join lateral (values(a.unique1)) ss(x) on b.unique2 = ss.x;
+ count 
+-------
+ 10000
+(1 row)
+
+-- lateral with VALUES, no flattening possible
+explain (costs off)
+  select count(*) from tenk1 a,
+    tenk1 b join lateral (values(a.unique1),(-1)) ss(x) on b.unique2 = ss.x;
                             QUERY PLAN                            
 ------------------------------------------------------------------
  Aggregate
@@ -3646,7 +3666,7 @@ explain (costs off)
 (8 rows)
 
 select count(*) from tenk1 a,
-  tenk1 b join lateral (values(a.unique1)) ss(x) on b.unique2 = ss.x;
+  tenk1 b join lateral (values(a.unique1),(-1)) ss(x) on b.unique2 = ss.x;
  count 
 -------
  10000
@@ -4214,7 +4234,7 @@ select * from
     cross join
     lateral (select q1, coalesce(ss1.x,q2) as y from int8_tbl d) ss2
   ) on c.q2 = ss2.q1,
-  lateral (select ss2.y) ss3;
+  lateral (select ss2.y offset 0) ss3;
                                                                                   QUERY PLAN                                                                                  
 ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
  Nested Loop
@@ -4296,9 +4316,9 @@ select c.*,a.*,ss1.q1,ss2.q1,ss3.* from
 -- check processing of postponed quals (bug #9041)
 explain (verbose, costs off)
 select * from
-  (select 1 as x) x cross join (select 2 as y) y
+  (select 1 as x offset 0) x cross join (select 2 as y offset 0) y
   left join lateral (
-    select * from (select 3 as z) z where z.z = x.x
+    select * from (select 3 as z offset 0) z where z.z = x.x
   ) zz on zz.z = y.y;
                   QUERY PLAN                  
 ----------------------------------------------
index 7991e993f1477d48b2bcebeb27a0a8821752b816..6dabe503cc015380de0125fad14f6c16bf04e5c2 100644 (file)
@@ -2034,7 +2034,7 @@ select x from int8_tbl, extractq2(int8_tbl) f(x);
 (5 rows)
 
 create function extractq2_2(t int8_tbl) returns table(ret1 int8) as $$
-  select extractq2(t)
+  select extractq2(t) offset 0
 $$ language sql immutable;
 explain (verbose, costs off)
 select x from int8_tbl, extractq2_2(int8_tbl) f(x);
@@ -2058,3 +2058,25 @@ select x from int8_tbl, extractq2_2(int8_tbl) f(x);
  -4567890123456789
 (5 rows)
 
+-- without the "offset 0", this function gets optimized quite differently
+create function extractq2_2_opt(t int8_tbl) returns table(ret1 int8) as $$
+  select extractq2(t)
+$$ language sql immutable;
+explain (verbose, costs off)
+select x from int8_tbl, extractq2_2_opt(int8_tbl) f(x);
+         QUERY PLAN          
+-----------------------------
+ Seq Scan on public.int8_tbl
+   Output: int8_tbl.q2
+(2 rows)
+
+select x from int8_tbl, extractq2_2_opt(int8_tbl) f(x);
+         x         
+-------------------
+               456
+  4567890123456789
+               123
+  4567890123456789
+ -4567890123456789
+(5 rows)
+
index db7aefee28385da373024c866c01dd3fc3b08a4a..06a27ea151da3136cb89f5467dba12fe6c43ca12 100644 (file)
@@ -1137,6 +1137,13 @@ explain (costs off)
 select count(*) from tenk1 a,
   tenk1 b join lateral (values(a.unique1)) ss(x) on b.unique2 = ss.x;
 
+-- lateral with VALUES, no flattening possible
+explain (costs off)
+  select count(*) from tenk1 a,
+    tenk1 b join lateral (values(a.unique1),(-1)) ss(x) on b.unique2 = ss.x;
+select count(*) from tenk1 a,
+  tenk1 b join lateral (values(a.unique1),(-1)) ss(x) on b.unique2 = ss.x;
+
 -- lateral injecting a strange outer join condition
 explain (costs off)
   select * from int8_tbl a,
@@ -1247,7 +1254,7 @@ select * from
     cross join
     lateral (select q1, coalesce(ss1.x,q2) as y from int8_tbl d) ss2
   ) on c.q2 = ss2.q1,
-  lateral (select ss2.y) ss3;
+  lateral (select ss2.y offset 0) ss3;
 
 -- case that breaks the old ph_may_need optimization
 explain (verbose, costs off)
@@ -1265,9 +1272,9 @@ select c.*,a.*,ss1.q1,ss2.q1,ss3.* from
 -- check processing of postponed quals (bug #9041)
 explain (verbose, costs off)
 select * from
-  (select 1 as x) x cross join (select 2 as y) y
+  (select 1 as x offset 0) x cross join (select 2 as y offset 0) y
   left join lateral (
-    select * from (select 3 as z) z where z.z = x.x
+    select * from (select 3 as z offset 0) z where z.z = x.x
   ) zz on zz.z = y.y;
 
 -- test some error cases where LATERAL should have been used but wasn't
index 470571b0fb6362bad966ecd8e46afcc956f5cf8b..9484023f97b79531c11faba7e3dc730b58eb6f75 100644 (file)
@@ -621,10 +621,21 @@ select x from int8_tbl, extractq2(int8_tbl) f(x);
 select x from int8_tbl, extractq2(int8_tbl) f(x);
 
 create function extractq2_2(t int8_tbl) returns table(ret1 int8) as $$
-  select extractq2(t)
+  select extractq2(t) offset 0
 $$ language sql immutable;
 
 explain (verbose, costs off)
 select x from int8_tbl, extractq2_2(int8_tbl) f(x);
 
 select x from int8_tbl, extractq2_2(int8_tbl) f(x);
+
+-- without the "offset 0", this function gets optimized quite differently
+
+create function extractq2_2_opt(t int8_tbl) returns table(ret1 int8) as $$
+  select extractq2(t)
+$$ language sql immutable;
+
+explain (verbose, costs off)
+select x from int8_tbl, extractq2_2_opt(int8_tbl) f(x);
+
+select x from int8_tbl, extractq2_2_opt(int8_tbl) f(x);