Fix planner's handling of RETURNING lists in writable CTEs.
authorTom Lane <tgl@sss.pgh.pa.us>
Thu, 26 Apr 2012 00:20:33 +0000 (20:20 -0400)
committerTom Lane <tgl@sss.pgh.pa.us>
Thu, 26 Apr 2012 00:20:33 +0000 (20:20 -0400)
commit9fa82c980935ef4aee18fabe8da20ae2198b052a
treec8345ab2665cc49129875f9921d6f6de1b097834
parentc62b8eaae11aaa69a2b71bc63f9f78ca72eb412c
Fix planner's handling of RETURNING lists in writable CTEs.

setrefs.c failed to do "rtoffset" adjustment of Vars in RETURNING lists,
which meant they were left with the wrong varnos when the RETURNING list
was in a subquery.  That was never possible before writable CTEs, of
course, but now it's broken.  The executor fails to notice any problem
because ExecEvalVar just references the ecxt_scantuple for any normal
varno; but EXPLAIN breaks when the varno is wrong, as illustrated in a
recent complaint from Bartosz Dmytrak.

Since the eventual rtoffset of the subquery is not known at the time
we are preparing its plan node, the previous scheme of executing
set_returning_clause_references() at that time cannot handle this
adjustment.  Fortunately, it turns out that we don't really need to do it
that way, because all the needed information is available during normal
setrefs.c execution; we just have to dig it out of the ModifyTable node.
So, do that, and get rid of the kluge of early setrefs processing of
RETURNING lists.  (This is a little bit of a cheat in the case of inherited
UPDATE/DELETE, because we are not passing a "root" struct that corresponds
exactly to what the subplan was built with.  But that doesn't matter, and
anyway this is less ugly than early setrefs processing was.)

Back-patch to 9.1, where the problem became possible to hit.
src/backend/optimizer/plan/createplan.c
src/backend/optimizer/plan/planner.c
src/backend/optimizer/plan/setrefs.c
src/include/optimizer/planmain.h
src/test/regress/expected/with.out
src/test/regress/sql/with.sql