PostgreSQL 17新特性之MERGE语句增强

数据库 PostgreSQL
PostgreSQL 15 提供了 MERGE 语句,它可以基于源表或者查询结果更新目标表中的数据。MERGE 可以在单个语句中实现 INSERT、UPDATE 以及 DELETE 操作。

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 语句新增数据时的操作:

MERGE INTO test t
USING (VALUES ('pg17')) AS s(tag)
ON t.tag = s.tag
WHEN MATCHED THEN
    UPDATE SET posts = posts + 1
WHEN NOT MATCHED THEN
    INSERT (tag, posts) VALUES (s.tag, 1)
RETURNING *;

  tag   | id | tag      | posts 
--------+----+----------+-------------
 pg17   |  1 | pg17     |           1

对于 INSERT 操作,RETURNING 返回了插入之后的数据。同时,RETURNING * 还返回了源表中的数据。

接下来是更新数据的操作:

MERGE INTO test t
USING (VALUES ('pg17')) AS s(tag)
ON t.tag = s.tag
WHEN MATCHED THEN
    UPDATE SET posts = posts + 1
WHEN NOT MATCHED THEN
    INSERT (tag, posts) VALUES (s.tag, 1)
RETURNING *;
  tag   | id | tag      | posts 
--------+----+----------+-------------
 pg17   |  1 | pg17     |           2

对于 UPDATE 操作,RETURNING 返回了更新之后的数据。

如果只需要返回目标表中更新后的数据,可以使用 RETURNING t.*:

MERGE INTO test t
USING (VALUES ('pg17')) AS s(tag)
ON t.tag = s.tag
WHEN MATCHED THEN
    UPDATE SET posts = posts + 1
WHEN 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。例如:

MERGE INTO test t
USING (VALUES ('sql'),('pg17')) AS s(tag)
ON t.tag = s.tag
WHEN MATCHED THEN
    UPDATE SET posts = posts + 1
WHEN NOT MATCHED THEN
    INSERT (tag, posts) VALUES (s.tag, 1)
RETURNING t.*, merge_action();
 id | tag      | posts       | merge_action 
----+----------+-------------+--------------
  2 | sql      |           1 | INSERT
  1 | pg17     |           4 | UPDATE

源表提供了 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 t
USING (VALUES ('pg17')) AS s(tag)
ON t.tag = s.tag
WHEN MATCHED THEN
    UPDATE SET posts = posts + 1
WHEN 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 语句解决增删改问题,完美!


责任编辑:华轩 来源: SQL编程思想
相关推荐

2023-06-07 07:30:34

iPadOS 17苹果

2012-05-18 14:36:50

Fedora 17桌面环境

2009-06-16 11:24:00

HTTP增强Java SE 6新特

2021-05-21 09:34:40

React React 17前端

2009-02-09 09:38:41

新特性MySQL 6.0MySQL

2022-09-22 07:03:15

Java 18LTSOracle

2011-08-19 10:40:27

SQL Server Merge命令

2012-05-21 14:16:36

Fedora17新特性

2024-02-04 15:58:53

C++ 17编程代码

2010-01-11 10:08:14

Oracle 11g

2012-01-12 16:16:20

Fedora 17特性规划

2009-08-18 09:37:42

ASP.NET 4.0

2009-03-10 10:31:26

DB2XMLpureXML

2013-05-20 10:25:45

vSphere 5.1vMotion

2018-09-19 16:15:18

MySQL直方图数据库

2009-08-03 18:34:43

TuplizersNHibernate

2017-01-09 16:25:55

Android Shortcuts系统

2011-05-20 09:35:22

JDK7

2009-07-08 09:47:49

Scala 2.8Scala

2012-07-02 10:43:49

JVMGroovyJava
点赞
收藏

51CTO技术栈公众号