postgres模拟merge 博客分类: postgresql posgresqlmergepg
群里这几天一直有人要做类似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 行记录)