database - 如何从 PostgreSQL 中的另一个函数调用表函数？

Question

我是 PostgreSQL 新手。

我有一个函数

CREATE OR REPLACE FUNCTION func1 ( a_person_id integer) RETURNS TABLE (granted_group_id varchar(10)) AS $body$

          WITH v_granted_group_list AS (
          SELECT DISTINCT a.group_id 
                FROM a_table a
               )
        SELECT v.group_id FROM v_granted_group_list v;

$body$
LANGUAGE sql;

我想在其他函数 func2 中使用它的输出。

我尝试按照以下代码执行此操作，但出现错误。

我们可以用什么来代替 v_access_groups 数组？

CREATE OR REPLACE FUNCTION func2 ( a_person_id IN integer,mhrc_emp_no IN varchar(50)) 
 RETURNS TABLE (granted_group_id varchar(10)) AS $body$
DECLARE

   v_access_groups varchar[];
BEGIN

      v_access_groups := func1(a_person_id);
      ---- this gives error
      ---ERROR:  malformed array literal: "LCCHG"
      ----DETAIL:  Array value must start with "{" or dimension information.
      --------CONTEXT:  PL/pgSQL function func2(integer,character varying) line 14 at SQL statement

      -- what can we you in place of v_access_groups array
    RETURN v_access_groups;
  END;
$body$
LANGUAGE PLPGSQL;

我想在选择查询的另一个函数中使用 func2 的结果

CREATE OR REPLACE FUNCTION func3 ( a_blurb_id integer, a_person_id integer,mhrc_emp_no varchar(20)) RETURNS boolean AS $body$
DECLARE

     v_acc_count         numeric;
     v_accessAllowed     boolean:=FALSE;
BEGIN
    SELECT COUNT(*) INTO v_acc_count
    FROM table1 agfa
    where  agfa.group_id IN (
        SELECT TO_CHAR(column_value) AS group_id
        FROM TABLE(func2(a_person_id,mhrc_emp_no))
    );
    RETURN TRUE;
  END;
$body$
LANGUAGE PLPGSQL;

我如何在 PostgreSQL 中实现这一点？

Answer 1

您可以像使用表一样在查询中使用集合返回函数(表函数)。

因此，要将结果分配给名为 v_access_groups 的 text[]，您可以按如下方式进行:

SELECT array_agg(granted_group_id) INTO v_access_groups FROM func1(a_person_id);

但是你的 func2 应该只包含:

RETURN QUERY SELECT * FROM func1(a_person_id);

如果您省略 TABLE(...) 表达式，您的 func3 应该可以工作——只需将函数视为一个表，无需额外的语法。