本文介绍如何使用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) – 列名
- ON CONSTRAINT constraint_name – 唯一约束的名称
- WHERE predicate – 带谓词的where子句.
action
可能为下面两者之一:
DO NOTHING – 如果行已存在表中,不执行任何动作.
DO UPDATE SET column_1 = value_1, … WHERE condition – 更新表中一些字段.
注意:
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插入更新特性。