Включите условие IF .. ELSE в plpgsql и создайте динамический запрос - PullRequest
Новая
       53

Включите условие IF .. ELSE в plpgsql и создайте динамический запрос

0 голосов
/ 04 декабря 2018

Привет всем, у меня есть 2 таблицы dcrhd (который содержит текущие данные) и dcrhd_arc (который содержит исторические данные), и я создал функцию для получения некоторых данных из этих таблиц. Но эта функция удовлетворяет только половину моего требования (это проверкаданные только из таблицы dcrhd) я поделюсь своей функцией здесь .. 

CREATE OR REPLACE FUNCTION dcr_report( --fin_year_flag,
 finid integer, prdid integer, comp_cd CHARACTER varying, divid integer, fsid integer) RETURNS refcursor LANGUAGE 'plpgsql' AS $BODY$
    DECLARE
    ref refcursor;
    BEGIN
    open ref for SELECT hd.report_no,
       hd.dcr_date,
       coalesce(pr2.para_descr,' ') work_type,
       coalesce(pr1.para_descr,' ') hq_type,
       coalesce(rm.route_name,' ') route_name,
       coalesce(hd.doctor_visits,0) doctor_visits,
       coalesce(hd.stockist_visits,0) stockist_visits,
       coalesce(hd.retailer_visits,0) retailer_visits,
       hd.dcr_id,
       fm.fs_name,
       hd.fstaff_id,
       CASE hd.status
            WHEN 'A'   THEN 'APPROVED'
            WHEN 'D'   THEN 'DISCARDED'
            WHEN 'F'   THEN 'FORWARDED'
            WHEN 'E'   THEN 'DRAFT'
       END
  status,
       zsm.fs_name report1,
       rsm.fs_name report2,
       fm.geog_lvl1_hq,
       fm.level_code,
       coalesce(pm.para_descr,'SELF') joint_work,
       fm.fs_code,
       fm.emp_code,
       coalesce(hd.doc_other,0) doc_other
FROM dcrhd hd
LEFT OUTER JOIN parameters pm ON hd.jfw = pm.para_code AND pm.para_type = 'JFW'
LEFT OUTER JOIN route_master rm ON rm.fstaff_id = hd.fstaff_id AND rm.route_id = hd.route_id AND rm.company_cd
= comp_cd
LEFT OUTER JOIN parameters pr1 ON pr1.para_code = hd.hq_exhq AND pr1.para_type = 'HQ_',
     parameters pr2,
     field_master fm,
     field_master zsm,
     field_master rsm
WHERE hd.period_id = prdid AND hd.fin_year_id = finid AND hd.fstaff_id = fm.fs_id AND fm.mgr_level4 =
zsm.fs_id AND fm.mgr_level3 = rsm.fs_id AND fm.fs_id =
     CASE
          WHEN fsid = 0  THEN fm.fs_id
          ELSE fsid
     END
      AND fm.div_id =
     CASE
          WHEN divid = 0 THEN fm.div_id
          ELSE divid
     END
      AND fm.fs_id = hd.fstaff_id AND fm.level_code = '005' AND pr2.para_code = hd.work_type AND pr2.
      para_type = 'WTP' AND hd.company = comp_cd AND fm.company_cd = comp_cd
ORDER BY fm.fs_name,
         dcr_date;
    RETURN REF;
    END;
    $BODY$;

Мое требование - я просто хочу добавить новый параметр с именем 'fin_year_flag' и соответственно выбрать основную таблицу (например, if fin_year_flag='current')затем перейдите к dcrhd иначе перейдите к dcrhd_arc Могу ли я добиться этого? Ребята, не могли бы вы поделиться своими идеями по этому вопросу? И есть ли другой способ полностью удовлетворить мои требования? Я новичок в PostgreSQL, который погуглил многихраз в интернете, но не смог найти ничего полезного ..

1 Ответ

0 голосов
/ 04 декабря 2018

Код, который вы опубликовали, огромен, поэтому я покажу вам, как динамически использовать имя таблицы и вернуть CURSOR для него, упростив его.

Я создаю две таблицы с примером строки. 

create table dcrhd     as select 'CURRENT' ::TEXT  as col;
create table dcrhd_arc as select 'ARCHIVED'::TEXT  as col;

Эта функция использует OPEN <refcursor> FOR EXECUTE для динамически генерируемого запроса.Вам нужно экранировать одинарные кавычки в вашем основном SQL, используя другую кавычку, или использовать долларовое цитирование .

Имя таблицы задается с помощью fin_year_flag из выражения CASE. 

CREATE OR REPLACE FUNCTION dcr_report( fin_year_flag TEXT) 
RETURNS refcursor LANGUAGE plpgsql 
AS $BODY$
    DECLARE
    ref refcursor;
    v_table_name TEXT := CASE fin_year_flag
                                   WHEN 'current' THEN 'dcrhd'
                                 ELSE 'dcrhd_arc' END;

    v_sql  text := format('select col from %s',v_table_name );
    BEGIN
    open ref for EXECUTE v_sql ;
    RETURN REF;
    END;
$BODY$;
...