如何从函数本身内部获取调用函数的语句？

Question

假设我有一个函数show_files(IN file text, IN suffix text, OUT statement text)。在下一步中调用该函数：

 SELECT * FROM show_files(file := 'example', suffix := '.png');

我的问题是：是否有任何解决方案可以让我从该函数内部获得调用该函数的语句？

我的意思是，在运行SELECT 后，函数(OUT statement text) 的输出应该是：'SELECT * FROM show_files(file := 'example', suffix := '.png');'，或者是否可以将此语句分配给函数内部的变量？

我需要像TG_NAME、TG_OP等在触发过程中的功能。

也许可以从 SELECT current_query FROM pg_stat_activity 中检索此语句？

当我尝试在函数中使用它时，我得到了一个空记录：

CREATE OR REPLACE FUNCTION f_snitch(text)
  RETURNS text AS
$BODY$
declare
rr text;
BEGIN
    RAISE NOTICE '.. from f_snitch.';
    -- do stuff
    SELECT current_query  into rr FROM pg_stat_activity 
    WHERE current_query ilike 'f_snitch';
    RETURN rr;
END
$BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100;

欢迎任何帮助和建议！

Answer 1

TG_NAME 和朋友是只存在于触发函数的特殊变量。常规的 plpgsql 函数没有这样的东西。我对如何在 plpgsql 中的 called 函数中获得这个想法感到新鲜。

您可以将RAISE NOTICE 添加到您的函数中，以便获得所需的信息

CREATE OR REPLACE FUNCTION f_snitch(text)
  RETURNS text LANGUAGE plpgsql AS
$func$
BEGIN
    RAISE NOTICE '.. from f_snitch.';
    -- do stuff
    RETURN 'Snitch says hi!';
END
$func$;

呼叫：

SELECT f_snitch('foo')

除了结果之外，这还返回一个通知：

NOTICE:  .. from f_snitch.

在两个方面未能取悦：

1. 来电声明不在通知中。
2. 通知中没有CONTEXT。

对于 1.，您可以改用 RAISE LOG（或者也将您的集群设置为记录 NOTICES - 我通常不会这样做，对我来说太冗长了）。使用标准设置，您会在数据库日志中获得带有 STATEMENT 的附加行：

LOG:  .. from f_snitch.
STATEMENT:  SELECT f_snitch('foo')

对于2.，请查看此related question at dba.SE。 CONTEXT 看起来像：

CONTEXT:  SQL statement "SELECT f_raise('LOG', 'My message')"
    PL/pgSQL function "f_snitch" line 5 at PERFORM

Answer 2

好的，我知道了！

CREATE OR REPLACE FUNCTION f_snitch(text)
  RETURNS setof record AS
$BODY$
BEGIN
   RETURN QUERY
    SELECT current_query 
    FROM pg_stat_activity 
    <strike>ORDER BY length(current_query) DESC LIMIT 1;</strike>
    where current_query ilike 'select * from f_snitch%';
    -- much more reliable solution

END
$BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100;

  select * from f_snitch('koper') AS (tt text);

结果如下：

这可能不是 100% 可靠的解决方案，但对于小型系统（对于少数用户而言）来说还可以。