pgsql: Fix planner's handling of RETURNING lists in writable CTEs. - Mailing list pgsql-committers

From Tom Lane
Subject pgsql: Fix planner's handling of RETURNING lists in writable CTEs.
Date
Msg-id E1SNCSG-0008NT-Vu@gemulon.postgresql.org
Whole thread Raw
List pgsql-committers
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.

Branch
------
master

Details
-------
http://git.postgresql.org/pg/commitdiff/9fa82c980935ef4aee18fabe8da20ae2198b052a

Modified Files
--------------
src/backend/optimizer/plan/createplan.c |   12 +----
src/backend/optimizer/plan/planner.c    |   27 ++---------
src/backend/optimizer/plan/setrefs.c    |   75 +++++++++++++++++++++++++-----
src/include/optimizer/planmain.h        |    4 --
src/test/regress/expected/with.out      |   42 +++++++++++++++++
src/test/regress/sql/with.sql           |    6 +++
6 files changed, 116 insertions(+), 50 deletions(-)


pgsql-committers by date:

Previous
From: Tom Lane
Date:
Subject: pgsql: Fix edge-case behavior of pg_next_dst_boundary().
Next
From: Tom Lane
Date:
Subject: pgsql: Fix planner's handling of RETURNING lists in writable CTEs.