在自定义函数中使用 SQL 查询作为数组参数（输入）

Question

我在 pl/pgsql 中创建了一个自定义函数，它的唯一输入是一个数组。我想将查询结果用作此自定义函数的输入，但似乎无法正确处理。我将使用一个更简单的函数作为示例，并且使用一个比我尝试用作输入的查询更简单的查询。

函数定义：

CREATE OR REPLACE FUNCTION bigger_than_ones(input_array int[])
returns SETOF int
AS
$$
DECLARE
  array_item int;
BEGIN
  FOREACH array_item in ARRAY input_array
  LOOP
    IF array_item > 1 THEN
      RETURN NEXT array_item;
    END IF;
  END LOOP;
END
$$ LANGUAGE plpgsql;

我有一个表，其列的数据类型与数组数据类型相同。在这种情况下是一个整数列：

SELECT * FROM my_table;
id | int_attribute
---+--------------
1  | 2
2  | 3
3  | 1
4  | 4
5  | 1
6  | 6
7  | 1
8  | 1
9  | 8

我想使用一个返回该数据类型的列的查询作为我的函数的输入。我已经尝试了一些变体，但没有任何运气：

SELECT * FROM bigger_than_ones(SELECT int_attribute FROM my_table);

返回：

result 
-------------
2
3
4
6
8

我使用类似结构函数的原因是因为我需要在我的代码中的多个位置计算来自多个表的合并时间范围（自定义类型）。我最好的想法是创建一个可以做到这一点的函数，无论我从哪里获取它，都将所有时间范围用作输入，这样我就不必每次都进行更复杂的查询。

对我缺少什么有什么想法吗？

Answer 1

你发现的是一个ARRAY constructor，它不是一个操作符（也不是一个函数），而是一个SQL构造（一个SQL语法元素）。引用的答案使用了错误的术语（现已修复）。差异可能很重要。

SELECT * FROM bigger_than_ones(ARRAY(SELECT int_attribute FROM my_table));

您也可以使用基本的aggregate function array_agg()，它更容易集成到更复杂的查询中 - 但对于简单的情况来说速度较慢：

SELECT * FROM bigger_than_ones((SELECT array_agg(int_attribute) FROM my_table));

相关：

• How to make an array from a SELECT returning more than one row

我假设您知道unnest()？它可以用来从根本上简化您的测试功能：

CREATE OR REPLACE FUNCTION bigger_than_ones(input_array int[])
  RETURNS SETOF int AS
$func$
   SELECT *
   FROM   unnest(input_array) elem
   WHERE  elem > 1;
$func$  LANGUAGE sql;

此外，通常有一种基于集合的优越方法来解决所有这些问题。 从一个集合构造一个数组只是为了将它传递给一个函数可能是不必要的复杂化。

Answer 2

我终于在另一个 SO 问题中找到了解决方案：Store select query's output in one array in postgres

我可以按如下方式使用 array() 运算符：

SELECT * FROM bigger_than_ones(ARRAY(SELECT int_attribute FROM my_table));

得到了预期的输出。

Answer 3

您是否尝试过创建自定义类型？有数组和表等集合类型，也有类似于单行记录的对象类型。然后，您可以创建此自定义类型的变量以用作参数。

这是一个例子：

CREATE 或 REPLACE TYPE namearray AS VARRAY(3) OF VARCHAR2(10);