PostgreSQL 14.x迁移脚本如何用临时变量存储查询值？

你要实现的逻辑完全可以在PostgreSQL 14.x中完成，核心是使用PL/pgSQL匿名块来处理变量声明、赋值和条件判断，变量仅在脚本执行期间存在，执行后自动销毁，不会持久化。

对应代码实现 #

DO $$
DECLARE
    user1 uuid;
    user2 uuid;
    user3 uuid;
BEGIN
    -- 为变量赋值，对应SQL Server的SELECT @user1 = id逻辑
    SELECT id INTO user1 FROM users WHERE type = 'role1';
    SELECT id INTO user2 FROM users WHERE type = 'role2';
    SELECT id INTO user3 FROM users WHERE type = 'role3';

    -- 条件判断，当三个变量都不为空时执行操作
    IF user1 IS NOT NULL AND user2 IS NOT NULL AND user3 IS NOT NULL THEN
        -- 这里编写你需要执行的操作，例如插入、更新等
        -- 示例：INSERT INTO target_table (user_id, role_type) VALUES (user1, 'role1');
        RAISE NOTICE '所有角色用户均存在，开始执行操作'; -- 可选提示，可删除
    END IF;
END $$;

关键说明 #

• 变量声明：PostgreSQL中不需要@前缀，直接在DECLARE块中声明变量名和类型（uuid对应SQL Server的UUID类型）。
• 赋值方式：使用SELECT ... INTO 变量替代SQL Server的SELECT @var = 列写法，这是PostgreSQL给变量赋值的标准方式。
• 条件判断：语法与SQL Server类似，使用IF ... THEN ... END IF结构，内部操作直接写在THEN块中即可。
• 异常注意：如果users表中对应type的记录存在多条，SELECT INTO会抛出异常，若需兼容这种场景，可添加LIMIT 1：SELECT id INTO user1 FROM users WHERE type = 'role1' LIMIT 1;。

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