如何编写PostgreSQL存储过程生成序列并在PGAdmin中调用获取返回值？

修正后的存储过程代码 #

原代码存在语法错误（CASE结构误用、nextval调用方式错误），以下是符合需求的正确实现：

CREATE OR REPLACE PROCEDURE sp_generate_seq(
    s_seq_name VARCHAR(255),
    i_id INOUT INT
)
LANGUAGE plpgsql
AS $$
BEGIN
    -- 传入ID为NULL时，调用目标序列的nextval获取新值
    IF i_id IS NULL THEN
        -- 用动态SQL执行nextval，避免序列名作为变量时的语法错误
        EXECUTE format('SELECT nextval(%L)', s_seq_name) INTO i_id;
    END IF;
    -- ID不为NULL时，直接保留原值返回，无需额外操作
END $$;

关键修正点 #

• 替换错误的CASE结构为更简洁的IF判断
• 使用EXECUTE + format实现动态SQL调用nextval，确保序列名称作为变量时的语法合法性，同时避免SQL注入风险
• 保留INOUT参数特性，实现「传入+返回」的双向数据传递

PGAdmin中调用示例 #

方式1：用DO块捕获并打印返回值 #

执行以下代码后，在PGAdmin底部的输出面板（点击「输出」标签）查看结果：

DO $$
DECLARE
    i_val INT;
BEGIN
    -- 第一次调用：i_val为NULL，自动获取序列下一个值
    CALL sp_generate_seq('foo_seq', i_val);
    RAISE NOTICE '第一次调用返回ID: %', i_val;

    -- 第二次调用：i_val已赋值，直接返回原值，不会触发nextval
    CALL sp_generate_seq('foo_seq', i_val);
    RAISE NOTICE '第二次调用返回ID: %', i_val;
END $$;

方式2：直接调用并查看返回值 #

通过PGAdmin查询工具执行以下语句，结果会直接显示在结果面板：

-- 先确保目标序列存在（如果没有的话）
CREATE SEQUENCE IF NOT EXISTS foo_seq START 1;

-- 声明变量接收返回值
\set i_val NULL
CALL sp_generate_seq('foo_seq', :i_val);
-- 输出返回的ID
SELECT :i_val AS generated_id;

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