Redshift中无需创建存储过程使用PL/pgSQL特性的可行性咨询

无需创建存储过程即可使用PL/pgSQL核心特性 #

可以。Amazon Redshift支持匿名PL/pgSQL代码块（DO块），无需创建永久存储过程，就能直接使用变量、循环、IF判断、动态SQL等PL/pgSQL特性。

运行临时PL/pgSQL代码块（无永久对象） #

匿名DO块完全满足这个需求：代码块仅在当前会话中执行，执行完毕后不会在数据库中留下任何永久对象，逻辑完全临时。

示例DO块代码：

DO $$DECLARE
    v_record_count INT;
    v_loop_idx INT;
BEGIN
    -- 变量赋值
    SELECT COUNT(*) INTO v_record_count FROM your_target_table;
    
    -- IF条件判断
    IF v_record_count > 50 THEN
        RAISE NOTICE '目标表记录数大于50，当前数量: %', v_record_count;
    ELSE
        RAISE NOTICE '目标表记录数小于等于50，当前数量: %', v_record_count;
    END IF;
    
    -- FOR循环示例
    FOR v_loop_idx IN 1..3 LOOP
        RAISE NOTICE '循环迭代次数: %', v_loop_idx;
    END LOOP;
    
    -- 动态SQL执行
    EXECUTE 'SELECT MAX(id) FROM your_target_table' INTO v_record_count;
    RAISE NOTICE '目标表最大ID值: %', v_record_count;
END$$;

会话级临时存储逻辑的替代方案 #

Redshift目前不支持创建会话级临时存储过程，但可以通过以下两种方式实现类似效果：

• 继续使用匿名DO块：每次需要执行逻辑时直接运行代码块即可
• 创建临时PL/pgSQL函数：该函数仅在当前会话中存在，会话结束后自动被删除，可封装可复用的逻辑

临时函数示例：

CREATE TEMP FUNCTION get_table_record_count(p_table_name VARCHAR) RETURNS INT
LANGUAGE plpgsql
AS $$
DECLARE
    v_count INT;
BEGIN
    -- 动态SQL查询指定表的记录数
    EXECUTE format('SELECT COUNT(*) FROM %I', p_table_name) INTO v_count;
    RETURN v_count;
END$$;

-- 调用临时函数
SELECT get_table_record_count('your_target_table');

内容的提问来源于stack exchange，提问作者jeremy