最近有开发人员问,有没有办法实现在pg9.5版本之前实现upsert功能,现整理如下--创建测试表,注意此处先不要主键或唯一约束create table t2 (ID int,name varchar(100));-- pg 在9.5之前实现不存在则插入-- 现在需要实现,当ID字段的值存在时,则更新name字段的值,如果ID字段的值不存在,则执行插入with upsert as (update t2 set name='rudy1' where ID=5 returning *) insert into t2 select 5,'rudy' where not exists (select 1 from upsert where ID=5);--注意使用此种方法并不能保证两个事务同时插入一条数据-- session1执行该语句,成功postgres=# begin;BEGINpostgres=# with upsert as (update t2 set name='rudy2' where ID=5 returning *) insert into t2 select 5,'rudy' where not exists (select 1 from upsert where ID=5);INSERT 0 1-- session2执行该语句,也成功postgres=# begin;BEGINpostgres=# with upsert as (update t2 set name='rudy2' where ID=5 returning *) insert into t2 select 5,'rudy' where not exists (select 1 from upsert where ID=5);INSERT 0 1--两者都提交后发现ID=5的数据有两条postgres=# select * from t2; ID | name ----+------ 5 | rudy 5 | rudy--为了保证并发,此时可以给表加上主键或唯一键 postgres=# alter table t2 add primary key(ID);ALTER table --此时session2再提交语句则会报错 postgres=# begin;BEGINpostgres=# with upsert as (update t2 set name='rudy3' where ID=5 returning *) insert into t2 select 5,'rudy' where not exists (select 1 from upsert where ID=5); ERROR: duplicate key value violates unique constraint "t2_pkey"DETAIL: Key (ID)=(5) already exists. --那有没有办法实现在表上没有主键或唯一约束时,也能保证并发呢?--有,此时需要使用pg_try_advisory_xact_lock(其是一个轻量级的锁,在事务回滚或提交后,会自动释放锁),但其接受的参数是整数,为了保证尽量唯一,可以使用md5函数 --借助于lock实现upsert,注意此sql对于记录不存在,可以保证只有一个session插入数据,对于同一条数据更新,先来的session会lock着记录,后来的session会waitwith w1 as(select ('x'||substr(md5('6'),1,16))::bit(64)::bigint as tra_ID),upsert as (update t2 set name='rudy2' where ID=6 returning *)insert into t2 select 6,'rudy' from w1 where pg_try_advisory_xact_lock(tra_ID) and not exists (select 1 from upsert where ID=6);--借助于lock实现upsert,注意此sql对于记录不存在,可以保证只有一个session插入数据,对于同一条数据更新,先来的session会更新数据,后来的session 失败with w1 as(select ('x'||substr(md5('6'),upsert as (update t2 set name='rudy2' from w1 where pg_try_advisory_xact_lock(tra_ID) and ID=6 returning *)insert into t2 select 6,'rudy' from w1 where pg_try_advisory_xact_lock(tra_ID) and not exists (select 1 from upsert where ID=6); --后记--为了保证性能,ID字段最好有索引(但不一定是主键或唯一约束时)--如果校验字段是否字段不为ID,把相应字段的替换掉ID字段则可--由md5虚拟的tra_ID并不保证一定是唯一的,但重复的概率极低-- 在pg9.5中可以直接使用upsert,注意此时要求表上有主键或唯一约束insert into t2 values(5,'rudy1') ON CONFliCT(ID) do update set name=EXCLUDED.name ;--对于MysqL可以使用insert into on duplicate key实现类似功能(其也要求有主键或唯一约束)在此不详细举例--对于oracle可以使用merge into,想想还是这个更强大,嘿嘿总结
以上是内存溢出为你收集整理的postgresql 9.5版本之前实现upsert功能全部内容,希望文章能够帮你解决postgresql 9.5版本之前实现upsert功能所遇到的程序开发问题。
如果觉得内存溢出网站内容还不错,欢迎将内存溢出网站推荐给程序员好友。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)