這篇文章主要介紹了PostgreSQL存儲過程循環(huán)調(diào)用方式,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
需求描述
碰到需求,需要往表里插入5萬條數(shù)據(jù), 打算使用存儲過程,但是postgres 數(shù)據(jù)庫沒有建存儲過程的SQL, 所以使用函數(shù)來實現(xiàn).
表數(shù)據(jù)結(jié)構(gòu)完整性要求一次插入兩條記錄, 兩條記錄相互外鍵約束, record1 的 partner_id 字段值是 record2 的主鍵id的值, record2 的 partner_id 字段值是 record1 的主鍵id的值.
實現(xiàn)
create
or replace function creatData() returns boolean as $BODY$
declare ii integer;
declare id1 integer;
declare id2 integer;
begin
ii = 1;
id1 = nextval('seq_table');
id2 = nextval('seq_table');
FOR ii IN 1..50000 LOOP
insert
into
table1
values(
id1,
10,
10250,
5001,
'2017-08-07 14:00:00',
'2017-08-07 15:00:00',
id2,
true,
864,
16950,
0,
0,
0,
null,
20,
null,
18050,
'2017-08-07 13:55:08',
18051,
'2017-08-07 13:57:28',
false,
401,
10,
null,
null,
null,
'DA-HZ001000003',
'2017-08-07 13:54:08',
'2017-08-07 13:57:28',
10251
);
insert
into
table1
values(
id2,
10,
10251,
5001,
'2017-08-07 14:00:00',
'2017-08-07 15:00:00',
id1,
true,
864,
16950,
0,
0,
0,
null,
20,
null,
18050,
'2017-08-07 13:55:08',
18051,
'2017-08-07 13:57:28',
false,
401,
10,
null,
null,
null,
'DA-HZ001000003',
'2017-08-07 13:54:08',
'2017-08-07 13:57:28',
10250
);
end LOOP;
return true;
end;
$BODY$ LANGUAGE plpgsql;
問題
這樣子插入只能插入一次, 因為取得序列值的地方在for循環(huán)的外面, id的值不會隨著循環(huán)再賦值, 主鍵沖突.
辦法
想到可以再對函數(shù)進行循環(huán), 于是再寫一個函數(shù)循環(huán)執(zhí)行上一個函數(shù), 去掉上個函數(shù)中的for 循環(huán)語句FOR i IN 1..500000 LOOP 和 end LOOP;
再寫一個下面函數(shù)循環(huán)執(zhí)行函數(shù)1
create or replace function loopCreate()
returns void as
$BODY$
begin for i in 1..50000 LOOP
PERFORM creatData();
end LOOP;
end;
$BODY$
LANGUAGE 'plpgsql' VOLATILE;
執(zhí)行函數(shù)
1select * from loopCreate() as tab;
好了,完成,10萬條數(shù)數(shù)據(jù)秒插, 2.1秒.
PS:CSDN的markdown編輯器真的很難用, 文字稍微長一點就卡, 而且換行經(jīng)常自動調(diào)跳回上一行, 無奈, 現(xiàn)在都是直接在別的地方寫好粘貼回來…
補充:postgresql 存儲過程中遍歷的一個小問題
問題
想實現(xiàn)這種功能,就是 for r in 后面的sql語句是一個變量,要把以下代碼修改一下
"sqltext" = 'select "ID","ZONENAME" from "ZONE_INFO" where "ID"<>0';
for r in "sqltext"
loop
return next r;
end loop;
解決方法:
sqltext = 'select "ID","ZONENAME" from "ZONE_INFO" where "ID" <>0';
for r in execute sqltext
loop
return next r;
end loop;
文章來源:腳本之家
來源地址:https://www.jb51.net/article/204917.htm
申請創(chuàng)業(yè)報道,分享創(chuàng)業(yè)好點子。點擊此處,共同探討創(chuàng)業(yè)新機遇!