Postgres函数返回未知的列数

时间:2018-08-16 15:06:56

标签: json postgresql plpgsql

我一直在寻找一种创建函数的方法,该函数将json作为输入,在其中我将指定列和表名,并将这些列作为输出。

假设我有下表:

 col1 | col2  | col3  
------+-------+-------
    1 | 303.8 | 331.9
    5 | 373.2 | 765.8
    1 | 467.4 | 427.5
    5 | 139.5 | 500.5
    4 | 350.6 | 706.2

我想这样调用一个函数:

SELECT foo('{"dimensions":["col2", "col3"], "table":"test"}');

这将给出:

 col2  | col3  
-------+-------
 303.8 | 331.9
 373.2 | 765.8
 467.4 | 427.5
 139.5 | 500.5
 350.6 | 706.2

问题是我不确定我的输出应该是什么,因为它不能是“表”,因为在执行函数之前我不知道列及其类型。

CREATE OR REPLACE FUNCTION foo ( j json ) RETURNS ??? AS
$$

DECLARE
   dimensions json ARRAY;
   sql_string TEXT;
   field TEXT;
   table_name TEXT;

BEGIN
   dimensions := ARRAY(SELECT json_array_elements(j->'dimensions'));
   table_name := TEXT ( j->> 'table' );

   sql_string := 'SELECT';

   FOREACH field IN ARRAY dimensions LOOP
      sql_string:= sql_string || ' ' || field || ',';
   END LOOP;
   SELECT TRIM (trailing ',' from sql_string) INTO sql_string;

   sql_string := sql_string || ' FROM ' || table_name ;

   EXECUTE sql_string INTO ???;
   RETURN ???;
END
$$ LANGUAGE plpgsql;

谢谢!

1 个答案:

答案 0 :(得分:0)

简而言之,refcursor和动态查询。


执行您的功能RETURNS refcursor

DECLARE部分:

_str text;
_cur refcursor = 'name_of_refcursor_1';

在代码中:

_str = 'SELECT ';
FOR _column IN <array or select to make column list>
LOOP
  _str = _str || _column.name || ', '
END LOOP;
_str = TRIM(TRAILING ', ' FROM _str);
_str = _str || ' FROM my_table1';

OPEN _cur FOR
EXECUTE _str;

RETURN _cur;

功能之外:

FETCH ALL FROM "name_of_refcursor_1";

FOR _row IN FETCH ALL FROM "name_of_refcursor_1"
...