postgres 模拟merge 插入或者对已存在进行更新

最新推荐文章于 2024-04-23 21:00:00 发布

weixin_34310369

最新推荐文章于 2024-04-23 21:00:00 发布

阅读量141

点赞数

文章标签：数据库 python

原文链接：https://my.oschina.net/u/1161519/blog/183856

版权

2019独角兽企业重金招聘Python工程师标准>>>

QQ群里这几天一直有人要做类似merge或者replace的操作，嚷嚷表字段多用function进行insert or update写起来麻烦。OK，下面贴一个触发器进行replace的demo

写个触发器插入之前执行触发器

-- 创建一个测试表

create table test(id int primary key , name varchar(50));

-- 触发器插入前ID如果已经存在则替换name的值
CREATE OR REPLACE function _replace() RETURNS TRIGGER AS $INSERT$

declare

_has int ;

BEGIN

-- 判断ID 是否已经存在
select 1 from test where id = NEW.id into _has;

raise notice 'ddd:%' , _has;

if _has > 0 then
-- 存在则更新然后返回null
update test set name = NEW.name where id = NEW.id;

RETURN null;

end if;
-- 不存在则返回让执行该做的插入操作
return NEW;

END;

$INSERT$

LANGUAGE PLPGSQL;

-- 给表加上触发器只针对insert
CREATE TRIGGER tbefore BEFORE INSERT ON test FOR EACH ROW EXECUTE PROCEDURE _replace();

-- 插入两个值
insert into test(id , name) values(1,'1');
insert into test(id , name) values(1,'6');

--查询

select * from test;

结果：

pumpkin=> select * from test;
id | name
----+------
1 | 6
(1 行记录)

时间：1.474 ms

转载于:https://my.oschina.net/u/1161519/blog/183856