在存储过程中调用存储过程

Question

我正在尝试在 postgres 9.3 上使用 sql 在函数中调用函数。

这个问题与another post by me有关。

我写了下面的函数。到目前为止，我还没有合并任何类型的保存输出 (COPY) 语句，所以我试图通过创建一个嵌套函数打印输出函数来解决这个问题。

CREATE FUNCTION retrieve_info(TEXT, TEXT) RETURNS SETOF   
retrieve_info_tbl AS $$
 SELECT tblA.id, tblA.method, tblA.species, tblA.location
 FROM tblA
 WHERE method=$1 AND species=$2
 GROUP BY id, method, species
 ORDER BY location
$$ LANGUAGE 'sql';

上述功能有效。

尝试创建嵌套函数。

CREATE FUNCTION print_out(TEXT, TEXT) RETURNS void AS $$
 COPY (SELECT * FROM retrieve_info($1, $2)) TO 'myfilepath/test.csv'    
 WITH CSV HEADER;
$$ LANGUAGE 'sql';

调用嵌套函数。

SELECT * FROM print_out('mtd1','sp1');

输出

上面给出了这个ERROR: column "$1" does not exist SQL state: 42P02 Context: SQL function "print_out" statement 1。但是，当用 'mtd1','sp1' 替换 print_out() 中的 arg1、arg2 时，正确的输出会打印到 test.csv（如下所示）

id | method | ind | location
----------------------------
1a | mtd1   | sp3 | locA
1d | mtd1   | sp3 | locB

如何让retrieve_info() 的arg1、arg2 在print_out() 中正确调用arg1、arg2？

我完全被困住了。不胜感激，谢谢

Answer 1

x 和 y 是故意引用的吗？

COPY (SELECT * FROM retrieve_info('x','y')) TO 'myfilepath/test.csv'

您没有将print_out 的x 和y 参数发送到retrieve_info - 而是发送字符串'x' 和'y'。假设您没有 method='x' AND species='y' 的记录，难怪您没有得到任何结果。

试试这个：

COPY (SELECT * FROM retrieve_info(x,y)) TO 'myfilepath/test.csv'

Answer 2

COPY 有点奇怪，因为它在某种程度上将其 query 参数视为字符串，即使它没有写为字符串。结果是query:

SELECT * FROM retrieve_info($1, $2)

不在函数的上下文中执行，而是在 COPY 本身的上下文中执行。即使你说：

copy (select * from t) ...

它更像是你写的：

copy 'select * from t' ...

所以到执行查询时，函数参数不再具有任何意义，COPY 的 query 参数可能看起来像其他语言中的闭包，但实际上并非如此，它更像传递给 eval 的字符串。

您可以使用常用的 Kludge of Last Resort 来解决这种奇怪现象：dynamic SQL。如果您编写函数以使用字符串整理和执行，您应该会得到更好的结果：

create or replace function print_out(text, text) returns void as $$
begin
    execute 'copy ('
         || 'select * from retrieve_info'
         ||     '(' || quote_literal($1) || ',' || quote_literal($2) || ')'
         || ') to ''myfilepath/test.csv'' with csv header;';
end;
$$ language plpgsql;