欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

postgres模拟merge 博客分类: postgresql posgresqlmergepg 

程序员文章站 2024-02-25 08:53:28
...

群里这几天一直有人要做类似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 行记录)