PostgreSQL 15 提供了 MERGE 语句,它可以基于源表或者查询结果更新目标表中的数据。MERGE 可以在单个语句中实现 INSERT、UPDATE 以及 DELETE 操作。
PostgreSQL 17 进一步增强了该语句的功能,包括:
支持 RETURNING 子句,可以返回新增、更新或者删除的数据行;
支持 WHEN NOT MATCHED BY SOURCE 操作,用于操作源表中不存在但是目标表中存在的数据行。
RETURNING 子句
PostgreSQL 支持 INSERT、UPDATE 以及 DELETE 语句的 RETURNING 子句,用于返回新增、更新或者删除的数据行。新版本则为 MERGE 语句提供了相同的功能。
我们先创建一个测试表:
CREATE TABLE test ( id INT GENERATED ALWAYS AS IDENTITY PRIMARY KEY, tag VARCHAR(10) NOT NULL UNIQUE, posts INT NOT NULL DEFAULT 0);
然后测试一下 MERGE 语句新增数据时的操作:
对于 INSERT 操作,RETURNING 返回了插入之后的数据。同时,RETURNING * 还返回了源表中的数据。
接下来是更新数据的操作:
对于 UPDATE 操作,RETURNING 返回了更新之后的数据。
如果只需要返回目标表中更新后的数据,可以使用 RETURNING t.*:
MERGE INTO test tUSING (VALUES ('pg17')) AS s(tag)ON t.tag = s.tagWHEN MATCHED THEN UPDATE SET posts = posts + 1WHEN NOT MATCHED THEN INSERT (tag, posts) VALUES (s.tag, 1)RETURNING t.*; id | tag | posts ----+----------+------------- 1 | pg17 | 2
提示:对于 DELETE 操作,RETURNING 返回删除之前的数据。
除此之外,新版本还增加了一个 merge_action 函数,用于返回 MERGE 操作类型。函数的返回值包括 INSERT、UPDATE 以及 DELETE。例如:
源表提供了 2 条记录,其中一条在目标表中执行了 INSERT 操作,另一条执行了 UPDATE 操作。
merge_action 函数只能用于 MERGE 语句的 RETURNING 字段列表。
WHEN NOT MATCHED BY SOURCE
PostgreSQL 17 还为 MERGE 语句增加了一个新的操作选项:WHEN NOT MATCHED BY SOURCE。该选项可以针对目标表中存在、但是源表中不存在的数据行进行操作,允许的操作包括 UPDATE、DELETE 或者 DO NOTHING。
例如:
MERGE INTO test tUSING (VALUES ('pg17')) AS s(tag)ON t.tag = s.tagWHEN MATCHED THEN UPDATE SET posts = posts + 1WHEN NOT MATCHED THEN INSERT (tag, posts) VALUES (s.tag, 1)WHEN NOT MATCHED BY SOURCE THEN DELETE;SELECT * FROM test; id | username | touch_count ----+----------+------------- 1 | pg17 | 5
源表只提供了 1 条记录,目标表存在 2 条记录,我们删除了目标表中多出的一条数据( tag = ‘sql’)。
默认的 WHEN NOT MATCHED 选项等价于 WHEN NOT MATCHED BY TARGET,用于针对源表中存在、但是目标表中不存在的数据行进行操作,允许的操作包括 INSERT 或者 DO NOTHING。
以前我们做数据 ETL 时,考虑到源表数据可能被删除,这时候目标表也需要做相同的操作,需要增加额外的 DELETE 语句,新版本 PostgreSQL 只需要一个 MERGE 语句解决增删改问题,完美!
转载此文是出于传递更多信息目的。若来源标注错误或侵犯了您的合法权益,请与本站联系,我们将及时更正、删除、谢谢。
https://www.414w.com/read/389485.html