mirror of
https://github.com/postgres/postgres.git
synced 2025-11-04 20:11:56 +03:00
Improve representation of PlanRowMark.
This patch fixes two inadequacies of the PlanRowMark representation. First, that the original LockingClauseStrength isn't stored (and cannot be inferred for foreign tables, which always get ROW_MARK_COPY). Since some PlanRowMarks are created out of whole cloth and don't actually have an ancestral RowMarkClause, this requires adding a dummy LCS_NONE value to enum LockingClauseStrength, which is fairly annoying but the alternatives seem worse. This fix allows getting rid of the use of get_parse_rowmark() in FDWs (as per the discussion around commits462bd95705and8ec8760fc8), and it simplifies some things elsewhere. Second, that the representation assumed that all child tables in an inheritance hierarchy would use the same RowMarkType. That's true today but will soon not be true. We add an "allMarkTypes" field that identifies the union of mark types used in all a parent table's children, and use that where appropriate (currently, only in preprocess_targetlist()). In passing fix a couple of minor infelicities left over from the SKIP LOCKED patch, notably that _outPlanRowMark still thought waitPolicy is a bool. Catversion bump is required because the numeric values of enum LockingClauseStrength can appear in on-disk rules. Extracted from a much larger patch to support foreign table inheritance; it seemed worth breaking this out, since it's a separable concern. Shigeru Hanada and Etsuro Fujita, somewhat modified by me
This commit is contained in:
@@ -73,8 +73,8 @@ expand_security_quals(PlannerInfo *root, List *tlist)
|
||||
rt_index = 0;
|
||||
foreach(cell, parse->rtable)
|
||||
{
|
||||
bool targetRelation = false;
|
||||
RangeTblEntry *rte = (RangeTblEntry *) lfirst(cell);
|
||||
bool targetRelation = false;
|
||||
RangeTblEntry *rte = (RangeTblEntry *) lfirst(cell);
|
||||
|
||||
rt_index++;
|
||||
|
||||
@@ -241,30 +241,10 @@ expand_security_qual(PlannerInfo *root, List *tlist, int rt_index,
|
||||
rc = get_plan_rowmark(root->rowMarks, rt_index);
|
||||
if (rc != NULL)
|
||||
{
|
||||
switch (rc->markType)
|
||||
{
|
||||
case ROW_MARK_EXCLUSIVE:
|
||||
applyLockingClause(subquery, 1, LCS_FORUPDATE,
|
||||
rc->waitPolicy, false);
|
||||
break;
|
||||
case ROW_MARK_NOKEYEXCLUSIVE:
|
||||
applyLockingClause(subquery, 1, LCS_FORNOKEYUPDATE,
|
||||
rc->waitPolicy, false);
|
||||
break;
|
||||
case ROW_MARK_SHARE:
|
||||
applyLockingClause(subquery, 1, LCS_FORSHARE,
|
||||
rc->waitPolicy, false);
|
||||
break;
|
||||
case ROW_MARK_KEYSHARE:
|
||||
applyLockingClause(subquery, 1, LCS_FORKEYSHARE,
|
||||
rc->waitPolicy, false);
|
||||
break;
|
||||
case ROW_MARK_REFERENCE:
|
||||
case ROW_MARK_COPY:
|
||||
/* No locking needed */
|
||||
break;
|
||||
}
|
||||
root->rowMarks = list_delete(root->rowMarks, rc);
|
||||
if (rc->strength != LCS_NONE)
|
||||
applyLockingClause(subquery, 1, rc->strength,
|
||||
rc->waitPolicy, false);
|
||||
root->rowMarks = list_delete_ptr(root->rowMarks, rc);
|
||||
}
|
||||
|
||||
/*
|
||||
@@ -276,6 +256,7 @@ expand_security_qual(PlannerInfo *root, List *tlist, int rt_index,
|
||||
if (targetRelation)
|
||||
applyLockingClause(subquery, 1, LCS_FORUPDATE,
|
||||
LockWaitBlock, false);
|
||||
|
||||
/*
|
||||
* Replace any variables in the outer query that refer to the
|
||||
* original relation RTE with references to columns that we will
|
||||
|
||||
@@ -92,9 +92,9 @@ preprocess_targetlist(PlannerInfo *root, List *tlist)
|
||||
if (rc->rti != rc->prti)
|
||||
continue;
|
||||
|
||||
if (rc->markType != ROW_MARK_COPY)
|
||||
if (rc->allMarkTypes & ~(1 << ROW_MARK_COPY))
|
||||
{
|
||||
/* It's a regular table, so fetch its TID */
|
||||
/* Need to fetch TID */
|
||||
var = makeVar(rc->rti,
|
||||
SelfItemPointerAttributeNumber,
|
||||
TIDOID,
|
||||
@@ -125,9 +125,9 @@ preprocess_targetlist(PlannerInfo *root, List *tlist)
|
||||
tlist = lappend(tlist, tle);
|
||||
}
|
||||
}
|
||||
else
|
||||
if (rc->allMarkTypes & (1 << ROW_MARK_COPY))
|
||||
{
|
||||
/* Not a table, so we need the whole row as a junk var */
|
||||
/* Need the whole row as a junk var */
|
||||
var = makeWholeRowVar(rt_fetch(rc->rti, range_table),
|
||||
rc->rti,
|
||||
0,
|
||||
|
||||
@@ -1389,9 +1389,14 @@ expand_inherited_rtentry(PlannerInfo *root, RangeTblEntry *rte, Index rti)
|
||||
newrc->prti = rti;
|
||||
newrc->rowmarkId = oldrc->rowmarkId;
|
||||
newrc->markType = oldrc->markType;
|
||||
newrc->allMarkTypes = (1 << newrc->markType);
|
||||
newrc->strength = oldrc->strength;
|
||||
newrc->waitPolicy = oldrc->waitPolicy;
|
||||
newrc->isParent = false;
|
||||
|
||||
/* Include child's rowmark type in parent's allMarkTypes */
|
||||
oldrc->allMarkTypes |= newrc->allMarkTypes;
|
||||
|
||||
root->rowMarks = lappend(root->rowMarks, newrc);
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user