PL/pgSQLでクエリで返した戻り値を参照したい

Question

前提・実現したいこと

SQLファイル内でファンクションを実行し、複数行（クエリ）で帰ってくる戻り値をそれぞれ参照したい。

■ファイル構成
. D:\SQLTEST
. 　├ POWERSHELL
.　 │　└ TEST.ps1
.　 └ SQL
.　 　└ TEST.sql

該当のソースコード

CREATE OR REPLACE FUNCTION test_function()
RETURNS SETOF CHARACTER VARYING
AS $$
DECLARE
  return_cd    varchar;
  return_text1 varchar;

BEGIN
  -- 変数初期化
  return_cd    := '0';
  return_text1 := '';

  SELECT test_col INTO col1 FROM test_tab;
  
  IF return_cd = '0'
  THEN
    return_text1 := '正常終了';
    RETURN QUERY VALUES (return_cd),(return_text1);
  END IF;

EXCEPTION
  WHEN OTHERS THEN
    return_cd    := '1';
    return_text1 := '異常終了';
    RETURN QUERY VALUES (return_cd),(return_text1);

END;
$$
LANGUAGE 'PLPGSQL';

まず、FUNCTIONでは、上記のように正常終了・異常終了にかかわらず戻り値をクエリで設定しています。

■実行結果（正常終了）

1 | 0
2 | 正常終了

■実行結果（異常終了）

1 | 1
2 | 異常終了

DO $$
DECLARE
  wk_return_cd    varchar;
  wk_return_text1 varchar;

BEGIN
  select * from test_function()
  into wk_return_cd, wk_return_text1;  -- ここがまず上手くいかない
  
  IF wk_return_cd = '1'
  THEN
     -- なんらかのエラーを記述
  END IF;

END $$;

そして、ファンクションの予備元であるPSQL側で、ファンクションの実行結果を参照して、
return_cdが 1 なら エラー、としたいのですが、
まず into でそれぞれ変数に入れることができません。
（戻り値が 複数列ではなく複数行のため、だと思います）

実行結果の欄にあるとおり、複数行で戻り値が返ってくるため、intoがうまく動かないのですが
なにか解決策はありますでしょうか。

ご教示いただけますと幸いです。

Answer 1

別に複数行返す必要は無いのではないですか。

CREATE OR REPLACE FUNCTION test_function()
RETURNS CHARACTER VARYING[]
AS $$
DECLARE
BEGIN
  SELECT test_col INTO col1 FROM test_tab;
  RETURN array['0', '正常終了']::CHARACTER VARYING[];

EXCEPTION
  WHEN OTHERS THEN
    RETURN array['1', '異常終了']::CHARACTER VARYING[];
END;
$$
LANGUAGE 'plpgsql';

DO $$
DECLARE
  wk_return_cd    varchar;
  wk_return_text1 varchar;

BEGIN
  select result[1], result[2] from test_function() as tf(result)
  into wk_return_cd, wk_return_text1;  -- ここがまず上手くいかない

  IF wk_return_cd = '1'
  THEN
     -- なんらかのエラーを記述
  END IF;

END $$;

Comments

[L.J]

ご回答いただきありがとうございます。
なるほど、複数行ではなく配列で戻り値を設定するということですね。

実際に手元の環境で試してみて、実現したい通りの動きになることが確認できました。
PostgreSQLに不慣れなため、質問の意図がわかりにくかったかもしれませんが、とても助かりました。
ありがとうございました！

[sazi]

function の戻り値がsetofの場合、受け取る場合にはFetchが必要になりますので、質問の内容からは配列の方が扱いやすいと思います。