可以通过下列函数在 Perl 函数中访问数据库本身:
spi_exec_query(query [, limit]) spi_exec_query 执行 SQL 命令,并把整个行集作为 哈希引用数组的引用返回。如果指定了 limit 且其大于零,则 spi_exec_query 最多只会取回 limit 行,就像查询中包含了 LIMIT 子句一样。省略 limit 或将其指定为零时,不会限制返回行数。
只有在确定结果集会比较小时,才应使用此命令。下面是一个查询(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(command) spi_fetchrow(cursor) spi_cursor_close(cursor) 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, "cannot 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);
通常,应重复调用 spi_fetchrow,直到它返回 undef,这表示已经没有更多行可读。当 spi_fetchrow 返回 undef 时, spi_query 返回的游标会被自动释放。如果不打算读取 所有行,则应调用 spi_cursor_close 释放游标。 否则会导致内存泄漏。
spi_prepare(command, argument types) spi_query_prepared(plan, arguments) spi_exec_prepared(plan [, attributes], arguments) spi_freeplan(plan) spi_prepare, spi_query_prepared, spi_exec_prepared,以及 spi_freeplan 实现相同的功能,但用于预备查询。spi_prepare 接受包含编号参数占位符($1、$2 等)的查询字符串,以及由参数类型字符串组成的列表:
$plan = spi_prepare('SELECT * FROM test WHERE id > $1 AND name = $2',
'INTEGER', 'TEXT');
调用 spi_prepare 预备查询计划后,就可以使用该计划代替查询字符串。可以将其用于 spi_exec_prepared,其结果与以下函数返回的结果相同:spi_exec_query;也可以用于 spi_query_prepared,它返回游标,行为与 spi_query 完全相同。随后可以将该游标传给 spi_fetchrow。spi_exec_prepared 可选的第二个参数是属性哈希引用;目前唯一支持的属性是 limit,用于设置查询返回的最大行数。省略 limit 或将其指定为零,表示不限制行数。
预处理查询的优点在于,一个准备好的计划可以用于多次查询执行。当计划 不再需要时,可以用 spi_freeplan 将其释放:
CREATE OR REPLACE FUNCTION init() RETURNS VOID AS $$
$_SHARED{my_plan} = spi_prepare('SELECT (now() + $1)::date AS now',
'INTERVAL');
$$ LANGUAGE plperl;
CREATE OR REPLACE FUNCTION add_time( INTERVAL ) RETURNS TEXT AS $$
return spi_exec_prepared(
$_SHARED{my_plan},
$_[0]
)->{rows}->[0]->{now};
$$ LANGUAGE plperl;
CREATE OR REPLACE FUNCTION done() RETURNS VOID AS $$
spi_freeplan( $_SHARED{my_plan});
undef $_SHARED{my_plan};
$$ LANGUAGE plperl;
SELECT init();
SELECT add_time('1 day'), add_time('2 days'), add_time('3 days');
SELECT done();
add_time | add_time | add_time
------------+------------+------------
2005-12-10 | 2005-12-11 | 2005-12-12
请注意,spi_prepare 中的参数下标由 $1、$2、$3 等表示,因此应避免用双引号声明查询字符串,以免轻易引入难以察觉的错误。
下面的另一个示例展示了如何在 spi_exec_prepared 中使用可选参数:
CREATE TABLE hosts AS SELECT id, ('192.168.1.'||id)::inet AS address
FROM generate_series(1,3) AS id;
CREATE OR REPLACE FUNCTION init_hosts_query() RETURNS VOID AS $$
$_SHARED{plan} = spi_prepare('SELECT * FROM hosts
WHERE address << $1', 'inet');
$$ LANGUAGE plperl;
CREATE OR REPLACE FUNCTION query_hosts(inet) RETURNS SETOF hosts AS $$
return spi_exec_prepared(
$_SHARED{plan},
{limit => 2},
$_[0]
)->{rows};
$$ LANGUAGE plperl;
CREATE OR REPLACE FUNCTION release_hosts_query() RETURNS VOID AS $$
spi_freeplan($_SHARED{plan});
undef $_SHARED{plan};
$$ LANGUAGE plperl;
SELECT init_hosts_query();
SELECT query_hosts('192.168.1.0/30');
SELECT release_hosts_query();
query_hosts
-----------------
(1,192.168.1.1)
(2,192.168.1.2)
(2 rows)
spi_commit() spi_rollback() 提交或回滚当前事务。只能在从顶层调用的过程或匿名代码块 (DO 命令)中调用这些函数。请注意,无法通过 spi_exec_query 或类似函数执行 COMMIT 或 ROLLBACK 这类 SQL 命令;必须使用这些函数来完成。在一个事务结束后,会自动启动一个 新事务,因此不需要单独的函数来开始新事务。
这里是一个示例:
CREATE PROCEDURE transaction_test1()
LANGUAGE plperl
AS $$
foreach my $i (0..9) {
spi_exec_query("INSERT INTO test1 (a) VALUES ($i)");
if ($i % 2 == 0) {
spi_commit();
} else {
spi_rollback();
}
}
$$;
CALL transaction_test1();
elog(level, msg) 发出日志消息或错误信息。可用级别包括 DEBUG、LOG、INFO、 NOTICE、WARNING以及ERROR。 ERROR 会引发错误条件;如果周围的 Perl 代码没有 捕获它,错误就会传播到调用查询,导致当前事务或子事务被中止。这实质上 等同于 Perl 的 die 命令。其他级别只会生成不同优先级 的消息。某一优先级的消息是报告给客户端、写入服务器日志,还是两者兼有, 由配置变量 log_min_messages 和 client_min_messages 控制。详见 Chapter 19。
quote_literal(string) 返回给定字符串适当加引号后的形式,以便把它用作 SQL 语句字符串中的 字符串字面量。嵌入的单引号和反斜线会被正确地双写。注意,对于 undef 输入,quote_literal 会返回 undef; 如果参数可能为 undef,quote_nullable 往往更合适。
quote_nullable(string) 返回给定字符串适当加引号后的形式,以便把它用作 SQL 语句字符串中的 字符串字面量;如果参数为 undef,则返回未加引号的字符串 "NULL"。嵌入的单引号和反斜线会被正确地双写。
quote_ident(string) 返回给定字符串适当加引号后的形式,以便把它用作 SQL 语句字符串中的 标识符。只有在必要时才会添加引号,也就是当字符串包含非标识符字符, 或者会发生大小写折叠时。嵌入的引号会被正确地双写。
decode_bytea(string) 返回由给定字符串的内容表示的未转义二进制数据,该字符串应为 bytea 编码形式。
encode_bytea(string) 返回给定字符串中二进制数据内容的 bytea 编码形式。
encode_array_literal(array) encode_array_literal(array, delimiter)将引用数组的内容以数组字面量格式(见 Section 8.15.2)返回为字符串。若参数不是数组引用, 则原样返回该参数值。如果未指定分隔符,或者分隔符为 undef,则数组 字面量元素之间默认使用 ", " 作为分隔符。
encode_typed_literal(value, typename) 将一个 Perl 变量转换为第二个参数指定的数据类型的值,并返回该值的 字符串表示。它能正确处理嵌套数组和复合类型的值。
encode_array_constructor(array) 将引用数组的内容以数组构造器格式( Section 4.2.12)返回为字符串。其中 每个值都使用 quote_nullable 加引号。如果参数 不是数组引用,则返回使用 quote_nullable 加引号后的参数值。
looks_like_number(string) 如果按照 Perl 的规则看,给定字符串的内容像数字,则返回真值, 否则返回假值。如果参数为 undef,则返回 undef。前导和尾随空格 会被忽略。Inf 和 Infinity 被视为数字。
is_array_ref(argument) 如果给定参数可以作为数组引用处理,即参数的 ref 为 ARRAY 或 PostgreSQL::InServer::ARRAY,则返回真值;否则返回假值。