PostgreSQL upsert(插入更新)数据的操作详解

本文介绍如何使用PostgreSQL upsert特性插入或当被插入数据已存在则更新数据。

1. 介绍PostgreSQL upsert

在关系型数据库中，upsert是一个组合词，即当往表中插入记录，如果该记录已存在则更新，否则插入新记录。为了使用该特性需要使用INSERT ON CONFLICT语句：

INSERT INTO table_name(column_list) 
VALUES(value_list)
ON CONFLICT target action;

该语法中target可以是下面列举内容之一：

• (column_name) ?C 列名
• ON CONSTRAINT constraint_name ?C 唯一约束的名称
• WHERE predicate ?C 带谓词的where子句.

action可能为下面两者之一：

DO NOTHING ?C 如果行已存在表中，不执行任何动作.
DO UPDATE SET column_1 = value_1, … WHERE condition ?C 更新表中一些字段.

注意：ON CONFLICT子句仅从PostgreSQL 9.5版本才有效。如果需用在之前版本，需要使用其他方法实现。

2. PostgreSQL upsert示例

下面语句创建customers表，演示PostgreSQL upsert特性：

DROP TABLE IF EXISTS customers;

CREATE TABLE customers (
	customer_id serial PRIMARY KEY,
	name VARCHAR UNIQUE,
	email VARCHAR NOT NULL,
	active bool NOT NULL DEFAULT TRUE
);

customers 表包括四个字段customer_id, name, email, active，name字段有唯一约束确保名称唯一。

下面插入语句新增几条记录：

INSERT INTO 
 customers (name, email)
VALUES 
 ('IBM', 'contact@ibm.com'),
 ('Microsoft', 'contact@microsoft.com'),
 ('Intel', 'contact@intel.com');

假设现在Microsoft 修改email字段，从 contact@microsoft.com 到 hotline@microft.com。我们可以使用update更新语句，因为需要演示upsert特性，这里使用INSERT ON CONFLICT语句：

INSERT INTO customers (NAME, email)
VALUES('Microsoft','hotline@microsoft.com') 
ON CONFLICT ON CONSTRAINT customers_name_key 
DO NOTHING;

上面语句表示如果名称表中存在，则什么都不做。下面语句与上面等价，但使用name列代替唯一约束名称：

INSERT INTO customers (name, email)
VALUES('Microsoft','hotline@microsoft.com') 
ON CONFLICT (name) 
DO NOTHING;

假设当记录已存在时你需要连接新的邮箱至原邮箱，这时update动作：

INSERT INTO customers (name, email)
VALUES('Microsoft','hotline@microsoft.com') 
ON CONFLICT (name) 
DO 
 UPDATE SET email = EXCLUDED.email || ';' || customers.email;

这里使用EXCLUDED虚拟表，其包含我们要更新的记录，也就是新记录（相对于原记录customers）。等式右边字段需要表名进行区分，才不会报字段不明确错误。
读者可以自行验证结果是否如你所愿。

3. 总结

本文介绍通过INSERT ON CONFLICT实现PostgreSQL插入更新特性。