将选择查询的输出存储在 postgres 的一个数组中

Question

我的代码是：

SELECT column_name
FROM information.SCHEMA.columns
WHERE table_name = 'aean'

它返回表aean的列名称。
现在我已经声明了一个数组：

DECLARE colnames text[]

如何将 select 的输出存储在 colnames 数组中。
是否需要初始化colnames？

Answer 1

有两种方法。一种是聚合：

SELECT array_agg(column_name::TEXT)
FROM information.schema.columns
WHERE table_name = 'aean'

另一种是使用数组构造函数：

SELECT ARRAY(
    SELECT column_name 
    FROM information_schema.columns 
    WHERE table_name = 'aean'
)

我假设这是针对 plpgsql 的。在这种情况下，您可以这样分配它：

colnames := ARRAY(
    SELECT column_name
    FROM information_schema.columns
    WHERE table_name='aean'
);

Answer 2

我有完全相同的问题。只需对 Denis 给出的解决方案进行另一项工作修改即可（必须指定类型）：

SELECT ARRAY(
SELECT column_name::text
FROM information_schema.columns
WHERE table_name='aean'
)

Answer 3

常规：

SELECT post_id FROM posts WHERE(poster_name='John');

output: [
  {'post_id': 1},
  {'post_id': 2},
  {'post_id': 3},
]

使用 ARRAY_AGG：

SELECT ARRAY_AGG(post_id) FROM posts WHERE(poster_name='John');

output: [
 {[1, 2, 3]}
]

Answer 4

转换为数据类型“TEXT”将确保您的查询运行没有任何问题。
在plpgsql中，当我们分配给数组变量时，我们不需要使用类型转换。我的要求是获取特定表的所有列名称的 CSV。我在 plpgsql 中使用了以下代码。

Declare col_list varchar[]:=NULL;
cols varchar:=NULL;
Begin
    col_list := ARRAY(select t.name from frm_columns t where t.tname='emp_mstr');
    cols := array_to_string(col_list,',');
    return cols;
End;

Answer 5

CREATE OR REPLACE FUNCTION f_test_array(in _colname text)
returns text as $body$
DECLARE colnames text[];
begin
colnames := ARRAY(
    SELECT column_name FROM information_schema.columns WHERE table_name='customer'
);
    if exists(select _colname = any(colnames))
    then return format('%s it exits.', _colname);
    else return format('%s not exits.', _colname);
end if;
end
$body$
LANGUAGE plpgsql;   

检查该列是否存在。
关键点：如果存在(select _colname = any(colnames))
我们还可以使用string_agg
String_agg 用法：

CREATE OR REPLACE FUNCTION f_test_array1(in _colname text)
returns text as $body$
DECLARE colnames text;
begin
colnames := (SELECT string_agg(column_name,',') FROM information_schema.columns WHERE table_name='customer')::text;
if exists(select colnames ilike '%' || quote_literal(_colname) ||'%')
    then return format('column %s  exits.', _colname);
    else return format('column %s does not exits.', _colname);
end if;
end
$body$
LANGUAGE plpgsql;

Answer 6

在函数中声明 colnames text[] 数组，然后在 begin 中编写以下查询：

SELECT column_name into colname
FROM信息.SCHEMA.列
WHERE 表名 = 'aean'