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

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

需求描述

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

表數據結構完整性要求一次插入兩條記錄, 兩條記錄相互外鍵約束, record1 的 partner_id 字段值是 record2 的主鍵id的值, record2 的 partner_id 字段值是 record1 的主鍵id的值.

實現

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)再賦值, 主鍵沖突.

辦法

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

再寫一個下面函數循環(huán)執(zhí)行函數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í)行函數

1select * from loopCreate() as tab;

好了,完成,10萬條數數據秒插, 2.1秒.

PS:CSDN的markdown編輯器真的很難用, 文字稍微長一點就卡, 而且換行經常自動調跳回上一行, 無奈, 現在都是直接在別的地方寫好粘貼回來…

補充：postgresql 存儲過程中遍歷的一個小問題

問題

想實現這種功能，就是 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è)新機遇！