码迷,mamicode.com
首页 > 数据库 > 详细

用第三方语言编写PostgreSQL 存储函数

时间:2014-05-13 06:47:08      阅读:370      评论:0      收藏:0      [点我收藏+]

标签:postgresql   plpgsql   function   python   

在PostgreSQL里,所有的存储函数需求都可以用PLPGSQL来实现。同时也支持用第三方语言来编写,这个就得看自己哪个方面熟练了。
不过要注意的一点是 PLPGSQL的效率怎么着都比其他第三方语言来的高效。
比如,简单的插入表的存储函数:


CREATE OR REPLACE FUNCTION ytt.insert_plpgsql(f_num integer)
 RETURNS void
 LANGUAGE plpgsql
AS $ytt$
declare i int := 0;
              v_rank int := 0;
	      v_log_time timestamp;
begin
while i < f_num 
loop
    v_rank = ceil(random()*100);
    v_log_time = now() - ‘1 day‘::interval*ceil(random()*50);
    insert into t1 (rank,log_time) values (v_rank,v_log_time);
    i = i + 1;
end loop;
end;
$ytt$;


现在来插入100W条记录,花费时间大概为27秒。


t_girl=# select insert_plpgsql(1000000);
 insert_plpgsql 
----------------
 
(1 row)


Time: 27286.668 ms



我们改用python 来实现
在编写python 脚本前,确保系统已经载入了plpythonu扩展。
t_girl=# \dx plpythonu
                        List of installed extensions
   Name    | Version |   Schema   |               Description                
-----------+---------+------------+------------------------------------------
 plpythonu | 1.0     | pg_catalog | PL/PythonU untrusted procedural language
(1 row)



以下是函数体:


CREATE OR REPLACE FUNCTION ytt.insert_py(f_num integer)
 RETURNS void
 LANGUAGE plpythonu
AS $ytt$
import datetime
import random
i = 0
while i < f_num:
    v_rank = random.randrange(0,100)
    v_log_time = datetime.datetime.now() - datetime.timedelta(days=random.randrange(0,50))
    query0 = "insert into ytt.t1 (rank,log_time) values (" + str(v_rank) + ",‘" + str(v_log_time)+ "‘)"
    plpy.execute(query0)
    i += 1
$ytt$;




清空表t1.
同样插入100W条记录, 这时候,时间上的差别显而易见了,用python 写的程序效率比数据库内部的语言慢了3倍。
t_girl=# select insert_py(1000000);
 insert_py 
-----------
 
(1 row)


Time: 86061.558 ms




那可以修改以上python 程序,让其效率来的高效一些,接近系统一些。 我们改用insert ... values ..()...() 的方式。
下面是函数体:


CREATE OR REPLACE FUNCTION ytt.insert_multi_py(f_num integer, f_values integer)
 RETURNS text
 LANGUAGE plpythonu
AS $ytt$
import datetime
import random
i = 0
j = 0
query0 = "insert into ytt.t1(rank,log_time) values "
data0 = ‘‘
if (f_num/f_values)*f_values < f_num:
    return ‘Parameters should be times relation.(f_num:1000,f_values:10)‘
else:
    while i < int(f_num/f_values):
        j = 0
        while j < f_values:
            v_rank = random.randrange(0,100)
            v_log_time = datetime.datetime.now() - datetime.timedelta(days=random.randrange(0,50))
            data0 = data0 + ",(" + str(v_rank) + ",‘" + str(v_log_time)+ "‘)"
            j += 1
        result0 = query0 + data0[1:len(data0)]
        plpy.execute(result0)
        data0 = ‘‘
        i += 1
    return ‘Inserting ‘ + str(f_num) + ‘ rows‘
$ytt$;




清空表t1.
继续插入100W条数据, 这时,插入时间和原始PLPGSQL的时间一致了。


t_girl=# select insert_multi_py(1000000,20);
    insert_multi_py     
------------------------
 Inserting 1000000 rows
(1 row)


Time: 27587.715 ms
t_girl=# 


用第三方语言编写PostgreSQL 存储函数,布布扣,bubuko.com

用第三方语言编写PostgreSQL 存储函数

标签:postgresql   plpgsql   function   python   

原文地址:http://blog.csdn.net/yueliangdao0608/article/details/25642771

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!