如何修复Postgres 16中无法正常运行的函数测试脚本？

从SQL Server迁移到Postgres，最容易踩的坑就是表变量的使用逻辑差异——Postgres没有SQL Server那种可以直接INSERT的表变量，得用自定义行类型数组或者临时表来实现类似功能。结合你的问题，给你梳理正确的写法和注意事项：

1. 核心语法修正（DO块中正确处理自定义表类型变量） #

假设你的mySchema.utTable是这样定义的行类型：

CREATE TYPE mySchema.utTable AS (
    id int,
    name text
);

那在DO块中声明变量时，要声明为数组类型（因为要存多行数据），然后通过array_agg或者直接数组构造来填充数据，示例如下：

DO $$
DECLARE
    data1 mySchema.utTable[]; -- 声明为自定义行类型的数组
BEGIN
    -- 方式1：从SELECT结果生成数组（适合动态数据）
    SELECT array_agg(row(id, name)::mySchema.utTable)
    INTO data1
    FROM (
        -- 这里放你的测试数据
        VALUES (1, '测试数据1'), (2, '测试数据2')
    ) AS t(id, name);

    -- 方式2：直接构造数组（适合固定少量测试数据）
    -- data1 := ARRAY[
    --     (1, '测试数据1')::mySchema.utTable,
    --     (2, '测试数据2')::mySchema.utTable
    -- ];

    -- 调用目标函数
    PERFORM mySchema.fnDoStuff(data1);
END $$;

2. 为什么你之前的写法报错？ #

• Postgres中没有INSERT INTO @变量这种语法，变量不是物理表，不能直接用INSERT操作。
• 单独用BEGIN块没用，因为VS Code会拆分独立语句，变量作用域只在单个语句内，必须用DO块把所有逻辑包裹成一个匿名函数，保证变量作用域有效。
• SELECT ... INTO在Postgres中是用来给变量赋值的，但你之前可能没把查询结果转成对应的自定义行类型数组，导致语法不匹配。

3. 其他实用方案 #

临时表替代（更贴近SQL Server表变量习惯） #

如果觉得数组写法别扭，可以用临时表，测试完会话结束会自动销毁：

-- 创建临时表并插入测试数据
CREATE TEMP TABLE temp_data AS
SELECT * FROM (
    VALUES (1, '测试数据1'), (2, '测试数据2')
) AS t(id, name);

-- 调用函数（如果函数支持从表中读取数据，或者参数为表类型）
PERFORM mySchema.fnDoStuff((SELECT array_agg(row(id, name)::mySchema.utTable) FROM temp_data));

直接传递VALUES子句给函数 #

如果函数参数是TABLE类型（Postgres 11+支持），可以直接把测试数据作为VALUES子句传递，不用声明变量：

PERFORM mySchema.fnDoStuff(VALUES (1, '测试数据1'), (2, '测试数据2'));

4. 工具注意事项 #

你用VS Code的MySQL插件跑Postgres脚本，本身就容易出现语法解析错误——MySQL和Postgres语法差异很大，建议换成Microsoft官方的PostgreSQL插件，能更好地支持Postgres语法高亮和语句执行逻辑。

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