新遇到一个业务需求,往postgresql中插入数据的时候需满足:如果数据库存在这条记录,则进行修改;不存在则进行插入。

Oracle中可以用merge来解决。Pg中之前没做过类似操作,百度查了一下,PostgreSQL 9.5 版本带来了一个新特性:UPSERTUPSERTINSERT, ON CONFLICT UPDATE的简写,简而言之就是:插入数据,正常时写入,主键冲突时更新。

示例来看一下:

一张测试表,结构如下:

yqm=# \d student;

           Table "public.student"

 Column |         Type          | Modifiers

--------+-----------------------+-----------

 id     | integer               |

 name   | character varying(20) |

Indexes:

    "id_cons" UNIQUE CONSTRAINT, btree (id)

目前数据如下:

yqm=# select * from student;

 id | name

----+------

  1 | a

  2 | b

  3 | c

(3 rows)

要插入的表要有一条唯一性约束,不然会报如下错:

yqm=# insert into student values(4,'d') on conflict(id) do update set name='as';

ERROR:  there is no unique or exclusion constraint matching the ON CONFLICT specification

如果存在这种报错,需要创建一条唯一性约束

yqm=# alter table student add constraint id_cons unique(id);

ALTER TABLE

再次插入

yqm=# insert into student values(4,'d') on conflict(id) do update set name='as';

INSERT 0 1

查看结果,已经新增了

yqm=# select * from student;

 id | name

----+------

  1 | a

  2 | b

  3 | c

  4 | d

(4 rows)

再次执行,会做更改操作

yqm=# insert into student values(4,'d') on conflict(id) do update set name='as';

INSERT 0 1

yqm=# select * from student;

 id | name

----+------

  1 | a

  2 | b

  3 | c

  4 | as

(4 rows)