是否可以在Postgres中执行以下操作:
SELECT column_name FROM information_schema WHERE table_name = 'somereport' AND data_type = 'integer';
SELECT SUM(coulmn_name[0]),SUM(coulmn_name[1]) ,SUM(coulmn_name[3]) FROM somereport;
换句话说,我需要根据某些条件从表中选择一组列,然后对表中的每一列求和。
我知道我可以在循环中执行此操作,因此我可以独立计算每个列,但显然需要查询从信息架构查询返回的每个列。例如:
FOR r IN select column_name from information_schema where report_view_name = 'somereport' and data_type = 'integer';
LOOP
SELECT SUM(r.column_name) FROM somereport;
END
答案 0 :(得分:4)
此查询创建您正在使用的完整DML语句:
WITH x AS (
SELECT 'public'::text AS _schema -- provide schema name ..
,'somereport'::text AS _tbl -- .. and table name once
)
SELECT 'SELECT ' || string_agg('sum(' || quote_ident(column_name)
|| ') AS sum_' || quote_ident(column_name), ', ')
|| E'\nFROM ' || quote_ident(x._schema) || '.' || quote_ident(x._tbl)
FROM x, information_schema.columns
WHERE table_schema = _schema
AND table_name = _tbl
AND data_type = 'integer'
GROUP BY x._schema, x._tbl;
您可以单独执行它或将此查询包装在plpgsql函数中,并使用EXECUTE
自动运行查询:
使用PostgreSQL 9.1.4进行测试
CREATE OR REPLACE FUNCTION f_get_sums(_schema text, _tbl text)
RETURNS TABLE(names text[], sums bigint[]) AS
$BODY$
BEGIN
RETURN QUERY EXECUTE (
SELECT 'SELECT ''{'
|| string_agg(quote_ident(c.column_name), ', ' ORDER BY c.column_name)
|| '}''::text[],
ARRAY['
|| string_agg('sum(' || quote_ident(c.column_name) || ')'
, ', ' ORDER BY c.column_name)
|| ']
FROM '
|| quote_ident(_schema) || '.' || quote_ident(_tbl)
FROM information_schema.columns c
WHERE table_schema = _schema
AND table_name = _tbl
AND data_type = 'integer'
);
END;
$BODY$
LANGUAGE plpgsql;
呼叫:
SELECT unnest(names) AS name, unnest (sums) AS col_sum
FROM f_get_sums('public', 'somereport');
返回:
name | col_sum
---------------+---------
int_col1 | 6614
other_int_col | 8364
third_int_col | 2720642
难点是为函数定义RETURN
类型,而返回的列的数量和名称将有所不同。一个细节有点帮助:您只需要integer
列。
我通过形成一个bigint
(sum(int_col)
返回bigint
)数组来解决这个问题。另外,我返回一个列名称数组。两者都按列名称按字母顺序排序。
在函数调用中,我将这些数组与unnest()
分开,得到显示的漂亮格式。
动态创建和执行的查询是高级内容。不要被多层引号弄糊涂。基本上你有EXECUTE
,它接受一个包含要执行的SQL查询的文本参数。反过来,此文本由辅助SQL查询提供,该查询构建主查询的查询字符串。
如果这一点太多或者plpgsql
对你来说是新的,请从this related answer开始,在那里我解释一下处理更简单功能的基础知识,并提供主要功能手册的链接
如果性能是必不可少的,请直接查询Postgres目录(pg_catalog.pg_attributes
),而不是使用标准化(但速度慢)information_schema.columns
。这是simple example with pg_attributes
。