如何将Postgres SUM结果存入多个变量？语法报错求助

问题场景 #

尝试用以下PL/pgSQL代码将SUM函数计算结果存入两个变量：

DECLARE
    total1 float;
    total2 float;
  
SELECT
    SUM(CASE WHEN totals.valid = 0 THEN amount ELSE 0 END),
    SUM(CASE WHEN totals.valid <> 0 THEN amount ELSE 0 END)
    INTO total1, total2
FROM
    totals;

执行时触发语法错误：

ERROR: 42601: syntax error at end of input

LINE 12: SUM(CASE WHEN totals.valid = 0 THEN amount ELSE 0 END),

^

CONTEXT: invalid type name "SUM(CASE WHEN totals.valid "

移除INTO total1, total2后代码可正常运行。

错误原因 #

核心问题是代码缺少PL/pgSQL块的执行段包裹：
PL/pgSQL的代码结构要求，DECLARE声明变量后，必须用BEGIN ... END;包裹所有执行语句。你的代码直接在DECLARE后写SELECT语句，PostgreSQL会错误地将其解析为声明部分的内容（而非执行语句），把SUM(...)表达式误认为是类型定义，从而抛出"invalid type name"的语法错误。

当你移除INTO子句后，PostgreSQL会将这段代码当作普通SQL查询执行，而非PL/pgSQL代码，因此不会触发语法错误，但这并不是你想要的PL/pgSQL变量赋值逻辑。

正确写法 #

将SELECT ... INTO语句放入BEGIN ... END块中，完整的PL/pgSQL块示例如下：

DO $$
DECLARE
    total1 float;
    total2 float;
BEGIN
    SELECT
        SUM(CASE WHEN totals.valid = 0 THEN amount ELSE 0 END),
        SUM(CASE WHEN totals.valid <> 0 THEN amount ELSE 0 END)
    INTO total1, total2
    FROM totals;
    
    -- 可在此处使用变量，例如输出结果
    RAISE NOTICE '无效金额总和: %, 有效金额总和: %', total1, total2;
END $$;

如果是在自定义函数中使用，写法如下：

CREATE OR REPLACE FUNCTION get_totals()
RETURNS void AS $$
DECLARE
    total1 float;
    total2 float;
BEGIN
    SELECT
        SUM(CASE WHEN totals.valid = 0 THEN amount ELSE 0 END),
        SUM(CASE WHEN totals.valid <> 0 THEN amount ELSE 0 END)
    INTO total1, total2
    FROM totals;
    
    RAISE NOTICE '无效金额总和: %, 有效金额总和: %', total1, total2;
END $$ LANGUAGE plpgsql;

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