【PostgreSQL】テーブル存在チェック，フィールド存在チェックなストアドプロシジャ[PL/pgSQL]

チョロっと書いてみた
ストアドレベルで書いておけば，取り回しも楽じゃろと？

ざっと調べて，ネットに事例もなかったんで載せておきまする

※SQLインジェクションな対応コードにはなってませんです

SQL文を文字列結合で組み立てる凶悪なコード（汗
バインドしても良かったけど，面倒くさくなったんでやめた（爆

ネット公開なサーバに本ストアド載せるときは，前段に必ずSQLインジェクション対応処理をおねがいしまする

今回のうちの案件は，完全ローカルなクローズドシステムなんでSQLインジェクションは関係ないもんで（GRB汗

毎度おなじみ，バグってたらスマソ（猛汗

—–

istable(テーブル名)がテーブルの存在チェック
isfield(テーブル名, フィールド名)がフィールドの存在チェック

どっちもtrueかfalseを返しまする
ex)
select istable(‘test_tbl’);
select isfield(‘test_tbl’, ‘mainkey’);

CREATE OR REPLACE FUNCTION public.istable(tblname text)
  RETURNS boolean AS
$BODY$
DECLARE
  sqlstr text;
  cur_max refcursor;
  colfields record;
  result boolean;
BEGIN
  -- 処理

  result := false;
  sqlstr := 'select count(tablename) as cnt from pg_tables where tablename = ''' || tblname || '''';              --SQL定義
  open cur_max for  EXECUTE  sqlstr;                   --カーソル Open
  loop
    fetch cur_max into colfields;
    if not found then
      exit;
    end if;
    if colfields.cnt > 0 then
	result := true;
    else
	result := false;
    end if;
    
    exit;
  end loop;
  close cur_max;  --カーソルクローズ

  return result;
END;
$BODY$
  LANGUAGE plpgsql;


CREATE OR REPLACE FUNCTION public.isfield(
    tblname text,
    fieldname text)
  RETURNS boolean AS
$BODY$
DECLARE
  sqlstr text;
  cur_max refcursor;
  colfields record;
  result boolean;
BEGIN
  -- 処理

  result := false;
  sqlstr := '
     select count(attname) as cnt
     from pg_attribute
     where attnum > 0
        and attrelid = (select relfilenode from pg_class where relname = ''' || tblname || ''')
        and attname = ''' || fieldname || '''';                                                  --SQL定義

  open cur_max for  EXECUTE  sqlstr;                   --カーソル Open
  loop
    fetch cur_max into colfields;
    if not found then
      exit;
    end if;
    if colfields.cnt > 0 then
	result := true;
    else
	result := false;
    end if;
    
    exit;
  end loop;
  close cur_max;  --カーソルクローズ

  return result;
END;
$BODY$
  LANGUAGE plpgsql;