mirror of
				https://github.com/postgres/postgres.git
				synced 2025-10-29 22:49:41 +03:00 
			
		
		
		
	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:
		| @@ -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, | ||||||
|   | |||||||
| @@ -131,7 +131,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, | ||||||
|   | |||||||
| @@ -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; | ||||||
|   | |||||||
| @@ -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"; | ||||||
| 	} | 	} | ||||||
|   | |||||||
| @@ -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, | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user