Написать хранимую процедуру с динамическим SQL - PullRequest
0 голосов
/ 09 июля 2019

Я хочу написать хранимую процедуру, которая принимает текстовый параметр (имя_таблицы) и выполняет запрос на основе этого имени таблицы: выберите число (*) и записывает найденное значение в таблицу.

Это часть (SELECT COUNT(*) FROM VARTABLE_NAME), которую я не знаю, как реализовать в Amazon Redshift

CREATE OR REPLACE  PROCEDURE SP_LOGGING (VARSTART INT, VARRUNID INT , VARPHASE VARCHAR, VARTABLE_NAME VARCHAR)
AS $$

BEGIN
            IF VARSTART = 1 THEN
                BEGIN
                UPDATE AUDIT.RUN_TEST
                SET TABLE_NAME = VARTABLE_NAME
                ,"no_of_records_at_start" = (SELECT COUNT(*) FROM VARTABLE_NAME)
                END;

            ELSEIF VARSTART = 0 THEN    
                BEGIN
                UPDATE AUDIT.RUN_TEST
                SET TABLE_NAME = VARTABLE_NAME
                ,"no_of_records_at_end" = (SELECT COUNT(*) FROM VARTABLE_NAME)
                END;
            END IF;
END;
$$ LANGUAGE PLPGSQL;

1 Ответ

2 голосов
/ 09 июля 2019

Вам нужно использовать EXECUTE для запуска динамического SQL.

CREATE PROCEDURE sp_logging (
      varstart      INT
    , varrunid      INT 
    , varphase      VARCHAR
    , vartable_name VARCHAR ) 
AS $$
BEGIN
    IF varstart = 1 THEN
        EXECUTE 'UPDATE audit.run_test
                    SET table_name = '||vartable_name||'
                      , start_rows = (SELECT COUNT(*) FROM '||vartable_name||' )';
    ELSEIF varstart = 0 THEN    
        EXECUTE 'UPDATE audit.run_test
                    SET table_name = '||vartable_name||'
                      , end_rows = (SELECT COUNT(*) FROM '||vartable_name||')';
    ELSE 
        RAISE ERROR 'Value of `varstart` must be 1 or 0. Received `%`.',varstart;
    END IF;
END;
$$ LANGUAGE PLPGSQL;
...