在postgres sql函数中按参数分组并返回与参数相同的列名

Question

我有一个问题

select count(distinct id), usertype from mytable group by usertype;

我想把它包装在一个以用户类型为参数的函数中

create or replace function myfun(facet text)
RETURNS TABLE(total_accounts bigint, facet text)

LANGUAGE plpgsql
AS $$
    select count(distinct id), facet from mytable group by facet;
$$;
select * from myfun('usertype');

所以输出列名也取决于我发送的参数。

但它显示错误“参数方面被多次使用”。

当我将 'usertype' 作为参数传递时，如何使输出与原始查询相同？

Answer 1

您可以在 plpgsql 函数中使用动态查询：

create or replace function myfun(OUT total_accounts bigint, INOUT facet text)
RETURNS setof record LANGUAGE plpgsql
AS $$
BEGIN
    RETURN QUERY EXECUTE '
    select count(distinct id),' || quote_ident(facet) || ' from mytable group by ' || quote_ident(facet) ;
END ;
$$;

select * from myfun('usertype');

测试结果在dbfiddle