Ensure that whole-row junk Vars are always of composite type.

The EvalPlanQual machinery assumes that whole-row Vars generated for the
outputs of non-table RTEs will be of composite types.  However, for the
case where the RTE is a function call returning a scalar type, we were
doing the wrong thing, as a result of sharing code with a parser case
where the function's scalar output is wanted.  (Or at least, that's what
that case has done historically; it does seem a bit inconsistent.)

To fix, extend makeWholeRowVar's API so that it can support both use-cases.
This fixes Belinda Cussen's report of crashes during concurrent execution
of UPDATEs involving joins to the result of UNNEST() --- in READ COMMITTED
mode, we'd run the EvalPlanQual machinery after a conflicting row update
commits, and it was expecting to get a HeapTuple not a scalar datum from
the "wholerowN" variable referencing the function RTE.

Back-patch to 9.0 where the current EvalPlanQual implementation appeared.

In 9.1 and up, this patch also fixes failure to attach the correct
collation to the Var generated for a scalar-result case.  An example:
regression=# select upper(x.*) from textcat('ab', 'cd') x;
ERROR:  could not determine which collation to use for upper() function
This commit is contained in:
Tom Lane 2011-11-27 22:27:24 -05:00
parent 91572ee0a6
commit dd3bab5fd7
5 changed files with 38 additions and 27 deletions

View File

@ -121,11 +121,17 @@ makeVarFromTargetEntry(Index varno,
* with error cases, but it's not worth changing now.) The vartype indicates * with error cases, but it's not worth changing now.) The vartype indicates
* a rowtype; either a named composite type, or RECORD. This function * a rowtype; either a named composite type, or RECORD. This function
* encapsulates the logic for determining the correct rowtype OID to use. * encapsulates the logic for determining the correct rowtype OID to use.
*
* If allowScalar is true, then for the case where the RTE is a function
* returning a non-composite result type, we produce a normal Var referencing
* the function's result directly, instead of the single-column composite
* value that the whole-row notation might otherwise suggest.
*/ */
Var * Var *
makeWholeRowVar(RangeTblEntry *rte, makeWholeRowVar(RangeTblEntry *rte,
Index varno, Index varno,
Index varlevelsup) Index varlevelsup,
bool allowScalar)
{ {
Var *result; Var *result;
Oid toid; Oid toid;
@ -157,39 +163,34 @@ makeWholeRowVar(RangeTblEntry *rte,
InvalidOid, InvalidOid,
varlevelsup); varlevelsup);
} }
else else if (allowScalar)
{ {
/* /* func returns scalar; just return its output as-is */
* func returns scalar; instead of making a whole-row Var,
* just reference the function's scalar output. (XXX this
* seems a tad inconsistent, especially if "f.*" was
* explicitly written ...)
*/
result = makeVar(varno, result = makeVar(varno,
1, 1,
toid, toid,
-1, -1,
exprCollation(rte->funcexpr),
varlevelsup);
}
else
{
/* func returns scalar, but we want a composite result */
result = makeVar(varno,
InvalidAttrNumber,
RECORDOID,
-1,
InvalidOid, InvalidOid,
varlevelsup); varlevelsup);
} }
break; break;
case RTE_VALUES:
toid = RECORDOID;
/* returns composite; same as relation case */
result = makeVar(varno,
InvalidAttrNumber,
toid,
-1,
InvalidOid,
varlevelsup);
break;
default: default:
/* /*
* RTE is a join or subselect. We represent this as a whole-row * RTE is a join, subselect, or VALUES. We represent this as a
* Var of RECORD type. (Note that in most cases the Var will be * whole-row Var of RECORD type. (Note that in most cases the Var
* expanded to a RowExpr during planning, but that is not our * will be expanded to a RowExpr during planning, but that is not
* concern here.) * our concern here.)
*/ */
result = makeVar(varno, result = makeVar(varno,
InvalidAttrNumber, InvalidAttrNumber,

View File

@ -129,7 +129,8 @@ preprocess_targetlist(PlannerInfo *root, List *tlist)
/* Not a table, so we need the whole row as a junk var */ /* Not a table, so we need the whole row as a junk var */
var = makeWholeRowVar(rt_fetch(rc->rti, range_table), var = makeWholeRowVar(rt_fetch(rc->rti, range_table),
rc->rti, rc->rti,
0); 0,
false);
snprintf(resname, sizeof(resname), "wholerow%u", rc->rowmarkId); snprintf(resname, sizeof(resname), "wholerow%u", rc->rowmarkId);
tle = makeTargetEntry((Expr *) var, tle = makeTargetEntry((Expr *) var,
list_length(tlist) + 1, list_length(tlist) + 1,

View File

@ -2059,8 +2059,15 @@ transformWholeRowRef(ParseState *pstate, RangeTblEntry *rte, int location)
/* Find the RTE's rangetable location */ /* Find the RTE's rangetable location */
vnum = RTERangeTablePosn(pstate, rte, &sublevels_up); vnum = RTERangeTablePosn(pstate, rte, &sublevels_up);
/* Build the appropriate referencing node */ /*
result = makeWholeRowVar(rte, vnum, sublevels_up); * Build the appropriate referencing node. Note that if the RTE is a
* function returning scalar, we create just a plain reference to the
* function value, not a composite containing a single column. This is
* pretty inconsistent at first sight, but it's what we've done
* historically. One argument for it is that "rel" and "rel.*" mean the
* same thing for composite relations, so why not for scalar functions...
*/
result = makeWholeRowVar(rte, vnum, sublevels_up, true);
/* location is not filled in by makeWholeRowVar */ /* location is not filled in by makeWholeRowVar */
result->location = location; result->location = location;

View File

@ -1188,7 +1188,8 @@ rewriteTargetListUD(Query *parsetree, RangeTblEntry *target_rte,
*/ */
var = makeWholeRowVar(target_rte, var = makeWholeRowVar(target_rte,
parsetree->resultRelation, parsetree->resultRelation,
0); 0,
false);
attrname = "wholerow"; attrname = "wholerow";
} }

View File

@ -35,7 +35,8 @@ extern Var *makeVarFromTargetEntry(Index varno,
extern Var *makeWholeRowVar(RangeTblEntry *rte, extern Var *makeWholeRowVar(RangeTblEntry *rte,
Index varno, Index varno,
Index varlevelsup); Index varlevelsup,
bool allowScalar);
extern TargetEntry *makeTargetEntry(Expr *expr, extern TargetEntry *makeTargetEntry(Expr *expr,
AttrNumber resno, AttrNumber resno,