Fix MERGE with DO NOTHING actions into a partitioned table.

ExecInitPartitionInfo() duplicates much of the logic in
ExecInitMerge(), except that it failed to handle DO NOTHING
actions. This would cause an "unknown action in MERGE WHEN clause"
error if a MERGE with any DO NOTHING actions attempted to insert into
a partition not already initialised by ExecInitModifyTable().

Bug: #18871
Reported-by: Alexander Lakhin <exclusion@gmail.com>
Author: Tender Wang <tndrwang@gmail.com>
Reviewed-by: Gurjeet Singh <gurjeet@singh.im>
Discussion: https://postgr.es/m/18871-b44e3c96de3bd2e8%40postgresql.org
Backpatch-through: 15
This commit is contained in:
Dean Rasheed 2025-03-29 09:58:40 +00:00
parent a0ed19e0a9
commit 8b6a0e2392
4 changed files with 34 additions and 2 deletions

View File

@ -876,7 +876,7 @@ ExecInitPartitionInfo(ModifyTableState *mtstate, EState *estate,
* reference and make copy for this relation, converting stuff that
* references attribute numbers to match this relation's.
*
* This duplicates much of the logic in ExecInitMerge(), so something
* This duplicates much of the logic in ExecInitMerge(), so if something
* changes there, look here too.
*/
if (node && node->operation == CMD_MERGE)
@ -956,6 +956,8 @@ ExecInitPartitionInfo(ModifyTableState *mtstate, EState *estate,
NULL);
break;
case CMD_DELETE:
case CMD_NOTHING:
/* Nothing to do */
break;
default:

View File

@ -3784,7 +3784,7 @@ ExecInitMerge(ModifyTableState *mtstate, EState *estate)
case CMD_NOTHING:
break;
default:
elog(ERROR, "unknown operation");
elog(ERROR, "unknown action in MERGE WHEN clause");
break;
}
}

View File

@ -2060,6 +2060,23 @@ SELECT * FROM pa_target ORDER BY tid;
15 | 1500 | initial
(8 rows)
ROLLBACK;
-- bug #18871: ExecInitPartitionInfo()'s handling of DO NOTHING actions
BEGIN;
TRUNCATE pa_target;
MERGE INTO pa_target t
USING (VALUES (10, 100)) AS s(sid, delta)
ON t.tid = s.sid
WHEN NOT MATCHED THEN
INSERT VALUES (1, 10, 'inserted by merge')
WHEN MATCHED THEN
DO NOTHING;
SELECT * FROM pa_target ORDER BY tid, val;
tid | balance | val
-----+---------+-------------------
1 | 10 | inserted by merge
(1 row)
ROLLBACK;
DROP TABLE pa_target CASCADE;
-- The target table is partitioned in the same way, but this time by attaching

View File

@ -1271,6 +1271,19 @@ MERGE INTO pa_target t
SELECT * FROM pa_target ORDER BY tid;
ROLLBACK;
-- bug #18871: ExecInitPartitionInfo()'s handling of DO NOTHING actions
BEGIN;
TRUNCATE pa_target;
MERGE INTO pa_target t
USING (VALUES (10, 100)) AS s(sid, delta)
ON t.tid = s.sid
WHEN NOT MATCHED THEN
INSERT VALUES (1, 10, 'inserted by merge')
WHEN MATCHED THEN
DO NOTHING;
SELECT * FROM pa_target ORDER BY tid, val;
ROLLBACK;
DROP TABLE pa_target CASCADE;
-- The target table is partitioned in the same way, but this time by attaching