PostgreSQL存儲過程循環調用方式

Posted on by

需求描述

碰到需求,需要往表裡插入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循環的外面, id的值不會隨著循環再賦值, 主鍵沖突.

辦法

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

再寫一個下面函數循環執行函數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;

執行函數

select * 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;

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方，望不吝賜教。

推薦閱讀：