PostgreSQLのfunctionで取得する列を動的に変えるとき

Question

お世話になっております。

PostgreSQLでファンクションを作成しており、引数に応じてSELECTの内容を変えたいです。

hogeテーブルのcol1-co5はいずれもvarchar(4000)です。

create or replace function test(p_type in integer)
returns table(
col1 hoge.col1%type,
col2 hoge.col2%type,
col3 hoge.col3%type,
col4 hoge.col4%type,
col5 hoge.col5%type) as $$
declare

    v_sql text;

begin

    if p_type = 1 then
        v_sql:='select col1, col2, col3, col4, col5 from hoge order by col1';
    else
        v_sql:='select col1, col2, col3 from hoge order by col1';
    end if;

    return query execute v_sql;

end $$ language plpgsql;

このようなfunctionを定義し実行時の引数に1以外を与えたとき、下記エラーが発生します。

ERROR:  structure of query does not match function result type
DETAIL: Number of returned columns (3) does not match expected column count (5).

そこで結果が5列になるようelse以下のSQLを変更してみたのですが、やはりエラーとなります。

        v_sql:='select col1, col2, col3, null, null from hoge order by col1';
        v_sql:='select col1, col2, col3, ''a'', ''b'' from hoge order by col1';

hogeテーブルの表構造からするとエラーにはならないと思うのですが、エラーになる理由と回避方法がありましたらご教示下さい。宜しくお願い致します。

Answer 1

そもそも返却するカラム数を可変にはできませんので、1個目のエラーは当然です。

2個目のエラーは1個目とは違うんじゃないですか？
型を省略しているのでtext属性になっていると思われます。
明示的にカラムの属性を指定してみて下さい。

尚、返却がhogeテーブルの構造そのものなら、

returns setof hoge%ROWTYPE

と記述する方が簡潔です。

Comments

[dotnet]

sazi様
早速有難うございます。ご指摘のとおり、2つ目のエラーは記載漏れで失礼致しました。

ERROR: structure of query does not match function result type

ご指摘を受けて修正して正常動作を確認しました。

v_sql:='select col1, col2, col3, null::varchar, null::varchar from hoge order by col1';

%ROWTYPEの件、ご教示有難うございます。使いどころがたくさんありそうなので覚えておきます。大変助かりました。有難うございました！