阿里云優(yōu)惠券 先領(lǐng)券再下單

這篇文章主要介紹了PostgreSQL存儲過程循環(huán)調(diào)用方式，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧

需求描述

碰到需求,需要往表里插入5萬條數(shù)據(jù), 打算使用存儲過程,但是postgres 數(shù)據(jù)庫沒有建存儲過程的SQL, 所以使用函數(shù)來實(shí)現(xiàn).

表數(shù)據(jù)結(jié)構(gòu)完整性要求一次插入兩條記錄, 兩條記錄相互外鍵約束, record1 的 partner_id 字段值是 record2 的主鍵id的值, record2 的 partner_id 字段值是 record1 的主鍵id的值.

實(shí)現(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;

問題

這樣子插入只能插入一次, 因?yàn)槿〉眯蛄兄档牡胤皆趂or循環(huán)的外面, id的值不會隨著循環(huán)再賦值, 主鍵沖突.

辦法

想到可以再對函數(shù)進(jìn)行循環(huán), 于是再寫一個(gè)函數(shù)循環(huán)執(zhí)行上一個(gè)函數(shù), 去掉上個(gè)函數(shù)中的for 循環(huán)語句FOR i IN 1..500000 LOOP 和 end LOOP;

再寫一個(gè)下面函數(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編輯器真的很難用, 文字稍微長一點(diǎn)就卡, 而且換行經(jīng)常自動調(diào)跳回上一行, 無奈, 現(xiàn)在都是直接在別的地方寫好粘貼回來…

補(bǔ)充：postgresql 存儲過程中遍歷的一個(gè)小問題

問題

想實(shí)現(xiàn)這種功能，就是 for r in 后面的sql語句是一個(gè)變量，要把以下代碼修改一下

"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è)報(bào)道，分享創(chuàng)業(yè)好點(diǎn)子。點(diǎn)擊此處，共同探討創(chuàng)業(yè)新機(jī)遇！