文章目录
- 文档用途
- 详细信息
文档用途
剖析UNION类型子查询提升的条件和过程
详细信息
注：图片较大，可在浏览器新标签页打开。
SQL:
SELECT * FROM score sc, LATERAL(SELECT * FROM student WHERE sno = 1 UNION ALL SELECT * FROM student WHERE sno = sc .sno) st WHERE st.sno > 0;
查询树结构：
分析该查询树，主查询包含2个RangeTblEntry：sc和st；其中st这个表的类型是子查询，包含2个RangeTblEntry，从SQL也可以看出这2个RangeTblEntry对应两个select查询，按照Query结构去分层该查询树为3层。
https://github.com/nullemp/postgres_notes/blob/master/image/subquery.node.jpg?raw=true
稍微简化一下，结构如下图所示：
打印该SQL的执行计划：
提升子查询后查询树结构：
https://github.com/nullemp/postgres_notes/blob/master/image/subquery_opti.node.jpg?raw=true
根据执行计划和查询树优化前后对比，对于UNION类型的子查询提升主要是将UNION两侧子查询提升，反映在查询树中即是这2个子查询类型的RangeTblEntry添加到主查询对应的rtable队列中，3层查询优化为2层查询结构。
提升流程：查找范围表中可以提升到父查询中的子查询。如果子查询没有特殊的特性，比如分组/聚合，那么我们可以将其合并到父查询的联接树中。此外，简单的 UNION ALL 结构的子查询可以转换为“追加关系”。
void pull_up_subqueries(PlannerInfo *root)
{
Assert(IsA(root->parse->jointree, FromExpr));
root->parse->jointree = (FromExpr *)
pull_up_subqueries_recurse(root, (Node *) root->parse->jointree,
NULL, NULL);
Assert(IsA(root->parse->jointree, FromExpr));
}
jointree中包含了FROM…WHERE…所引用的表，该递归结构通过pull_up_subqueries_recurse对其进行递归处理，所以优化执行时先去深度遍历FromExpr中的列表中的每一项成员：
if (IsA(jtnode, FromExpr))
{
FromExpr *f = (FromExpr *) jtnode;
ListCell *l;
Assert(containing_appendrel == NULL);
foreach(l, f->fromlist)
{
lfirst(l) = pull_up_subqueries_recurse(root, lfirst(l),
lowest_outer_join,
NULL);
}
}
如果RangeTblEntry是subquery类型并且满足简单子查询条件，使用pull_up_simple_union_all处理，该函数接受3个参数，分别是：查询树上下文， RangeTblRef, RangeTblEntry。
int varno = ((RangeTblRef *) jtnode)->rtindex;
RangeTblEntry *rte = rt_fetch(varno, root->parse->rtable);
if (rte->rtekind == RTE_SUBQUERY &&
is_simple_union_all(rte->subquery))
return pull_up_simple_union_all(root, jtnode, rte);
pull_up_simple_union_all:
根据优化后的查询树结构，提升的主要目的是把三个层次变成两个层次，那么如果“子子查询”中引用了顶层的列属性，那么这些变量应该提升一个层次，也就是调用incrementVarSublevelsUp_ rtable(rtable, -1 , 1 )。比如本例SQL：SELECT * FROM student WHERE sno = sc .sno ， sc.sno 就引用了第一个层次中的列表量,它的 Var >varlevlesup 的原值是 2（相对值），子查询提升之后应该变成1。
2.下发LATERAL，本例中是（SELECT * FROM student WHERE sno = 1）和 ( SELECT * FROM student WERE sno = sc.sno ）这两个子查询都变成 LATERAL，而不是只是针对引用父查询属性子查询才会拥有LATERAL语义。
if (rte->lateral)
{
ListCell *rt;
foreach(rt, rtable)
{
RangeTblEntry *child_rte = (RangeTblEntry *) lfirst(rt);
Assert(child_rte->rtekind == RTE_SUBQUERY);
child_rte->lateral = true;
}
}
3.把第三层次的两个RangeTblEntry：（SELECT * FROM student WHERE sno = 1）和（SELECT * FROM student WHERE sno = sc.sno ）两个子查询附加到第一层的 Query->rtable 列表中，在这第3步过后，后续的子查询的rtindex都将加上父查询rtindex作为偏置值。
/*
- Append child RTEs (and their perminfos) to parent rtable.
*/
CombineRangeTables(&root->parse->rtable, &root->parse->rteperminfos,
rtable, subquery->rteperminfos);
{
*dst_rtable = list_concat(*dst_rtable, src_rtable); ...}
4.开始对 subquery->setOperations 进行遍历 (pull_up_union_leaf_queries 函数)，为其中的每个子查询生成一个AppendRelInfo 节点，在本例中为（ SELECT * FROM student WHERE sno = 1〕和 (SELECT * FROM student WHERE sno = sc.sno ）生成两个 AppendRelInfo 节点,这种类型的节点是记录到查询树的上下文中，在查询树中看不到。
SetOperationStmt *op = (SetOperationStmt *) setOp;
/* Recurse to reach leaf queries */
pull_up_union_leaf_queries(op->larg, root, parentRTindex, setOpQuery,
childRToffset);
pull_up_union_leaf_queries(op->rarg, root, parentRTindex, setOpQuery,
childRToffset);
appinfo = makeNode(AppendRelInfo);
appinfo->parent_relid = parentRTindex;
appinfo->child_relid = childRTindex;
appinfo->parent_reltype = InvalidOid;
appinfo->child_reltype = InvalidOid;
make_setop_translation_list(setOpQuery, childRTindex, appinfo);
appinfo->parent_reloid = InvalidOid;
root->append_rel_list = lappend(root->append_rel_list, appinfo);
简单回顾这种类型子查询流程如下图：
到此为止，还有一个需要解决的问题：子查询提升将对应的RangeTblEntry添加到了父查询的rtable中，而且过程中更新了rtindex(第4步），这个新的RangeTblEntry不会在父查询的FromExpr中出现，所以构造完ApendRelInfo后，需要对子查询构造新的RangeTblRef，填充新的rtindex, 然后执行pull_up_subqueries_recurse。
rtr = makeNode(RangeTblRef);
rtr->rtindex = childRTindex;
(void) pull_up_subqueries_recurse(root, (Node *) rtr,
NULL, appinfo);
最后就能得到优化后的查询树结构。
