需求描述
碰到需求,需要往表里插入5万条数据, 打算使用 存储过程 ,但是postgres 数据库没有建存储过程的SQL, 所以使用函数来实现.
表数据结构完整性要求一次插入两条记录, 两条记录相互外键约束, record1 的 partner_id 字段值是 record2 的主键id的值, record2 的 partner_id 字段值是 record1 的主键id的值.
实现
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
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
|
1 2 3 4 5 6 7 8 9 |
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; |
执行函数
|
1 |
select * from loopCreate() as tab; |
好了,完成,10万条数数据秒插, 2.1秒.
PS:CSDN的markdown编辑器真的很难用, 文字稍微长一点就卡, 而且换行经常自动调跳回上一行, 无奈, 现在都是直接在别的地方写好粘贴回来…
补充：postgresql 存储过程中遍历的一个小问题
问题
想实现这种功能，就是 for r in 后面的sql语句是一个变量，要把以下代码修改一下
|
1 2 3 4 5 |
"sqltext" = 'select "ID","ZONENAME" from "ZONE_INFO" where "ID"<>0' ; for r in "sqltext" loop return next r; end loop; |
解决方法：
|
1 2 3 4 5 |
sqltext = 'select "ID","ZONENAME" from "ZONE_INFO" where "ID" <>0' ; for r in execute sqltext loop return next r; end loop; |
以上为个人经验，希望能给大家一个参考，也希望大家多多支持服务器之家。如有错误或未考虑完全的地方，望不吝赐教。
原文链接：https://blog.csdn.net/posse_wang/article/details/76848922
查看更多关于PostgreSQL存储过程循环调用方式的详细内容...