38.2. 从 PL/Perl 访问数据库 #
在 Perl 函数中访问数据库本身,可以通过下文描述的函数
spi_exec_query,也可以通过一个实验性模块
DBD::PgSPI(也可以从 CPAN 镜像站点获取)。该模块提供了一个名为
$pg_dbh 的、符合 DBI 的数据库句柄,可用普通的 DBI
语法执行查询。
PL/Perl 提供了三个附加的 Perl 命令:
spi_exec_query(query[,max-rows])spi_exec_query(command)spi_query(command)spi_fetchrow(command)spi_exec_query执行 SQL 命令,并返回整个行集;返回值是对一个数组的引用,数组元素为哈希引用。只有在确定结果集会比较小时,才应使用此命令。下面是一个查询(SELECT命令)的示例,指定了可选的最大行数:$rv = spi_exec_query('SELECT * FROM my_table', 5);这会返回最多 5 行,来自表
my_table。如果my_table包含列my_column,就可以取得结果中第$i行的该列值,方法如下:$foo = $rv->{rows}[$i]->{my_column};要取得
SELECT查询返回的总行数,可以这样做:$nrows = $rv->{processed}下面是使用另一种命令类型的示例:
$query = "INSERT INTO my_table VALUES (1, 'test')"; $rv = spi_exec_query($query);
可以这样访问命令状态(例如
SPI_OK_INSERT):$res = $rv->{status};要获取受影响的行数,可使用:
$nrows = $rv->{processed};这里是一个完整的示例:
CREATE TABLE test ( i int, v varchar ); INSERT INTO test (i, v) VALUES (1, 'first line'); INSERT INTO test (i, v) VALUES (2, 'second line'); INSERT INTO test (i, v) VALUES (3, 'third line'); INSERT INTO test (i, v) VALUES (4, 'immortal'); CREATE OR REPLACE FUNCTION test_munge() RETURNS SETOF test AS $$ my $rv = spi_exec_query('select i, v from test;'); my $status = $rv->{status}; my $nrows = $rv->{processed}; foreach my $rn (0 .. $nrows - 1) { my $row = $rv->{rows}[$rn]; $row->{i} += 200 if defined($row->{i}); $row->{v} =~ tr/A-Za-z/a-zA-Z/ if (defined($row->{v})); return_next($row); } return undef; $$ LANGUAGE plperl; SELECT * FROM test_munge();spi_query和spi_fetchrow需要配合使用,适用于结果集可能很大,或者希望在行到达时立即返回的情况。spi_fetchrow只能与spi_query一起使用。下面的示例演示了它们的配合方式:CREATE TYPE foo_type AS (the_num INTEGER, the_text TEXT); CREATE OR REPLACE FUNCTION lotsa_md5 (INTEGER) RETURNS SETOF foo_type AS $$ use Digest::MD5 qw(md5_hex); my $file = '/usr/share/dict/words'; my $t = localtime; elog(NOTICE, "opening file $file at $t" ); open my $fh, '<', $file # ooh, it's a file access! or elog(ERROR, "Can't open $file for reading: $!"); my @words = <$fh>; close $fh; $t = localtime; elog(NOTICE, "closed file $file at $t"); chomp(@words); my $row; my $sth = spi_query("SELECT * FROM generate_series(1,$_[0]) AS b(a)"); while (defined ($row = spi_fetchrow($sth))) { return_next({ the_num => $row->{a}, the_text => md5_hex($words[rand @words]) }); } return; $$ LANGUAGE plperlu; SELECT * from lotsa_md5(500);elog(level,msg)发出日志消息或错误信息。可用级别包括
DEBUG、LOG、INFO、NOTICE、WARNING以及ERROR。ERROR会引发错误条件;如果周围的 Perl 代码没有捕获它,错误就会传播到调用查询,导致当前事务或子事务被中止。这实质上等同于 Perl 的die命令。其他级别只会生成不同优先级的消息。某一优先级的消息是报告给客户端、写入服务器日志,还是两者兼有,由配置变量log_min_messages和client_min_messages控制。详见第 17 章。