如何设计嵌套函数事务：仅全部子函数提交才生效？能否简化异常处理？

需求背景 #

需要设计嵌套函数的数据库事务逻辑：主函数调用多个子函数，仅当所有子函数的操作都成功时，整个事务才提交；只要任一子函数操作失败，所有操作都回滚。

示例主函数unfollow调用delete_relationship和decrement_counter两个子函数，期望二者均成功时事务才提交：

CREATE OR REPLACE FUNCTION unfollow(
    follower_id uuid,
    following_id uuid
)
    RETURNS void
    LANGUAGE plpgsql
    SECURITY INVOKER
AS
$$
BEGIN
    PERFORM delete_relationship(follower_id, following_id);
    PERFORM decrement_counter(following_id);
END;
$$;

现有实现的问题 #

当前实现需要在每个子函数中手动检查CRUD操作的执行结果，未找到目标行时手动抛出异常回滚，代码冗余。例如delete_relationship函数：

CREATE OR REPLACE FUNCTION delete_relationship(
    follower_id uuid,
    following_id uuid
)
    RETURNS void
    LANGUAGE plpgsql
    SECURITY INVOKER
AS
$$
BEGIN
    DELETE
    FROM
        following_profiles
    WHERE
          follower = follower_id
      AND following = following_id;

    IF NOT FOUND THEN
        ROLLBACK;
        RAISE EXCEPTION 'Error in unfollow_transaction: %', SQLERRM;
    END IF;
END;
$$;

如果省略异常抛出逻辑，即使子函数操作失败，主事务仍会提交，导致数据不一致。

优化方案 #

可以通过以下方式简化实现，避免重复编写异常处理逻辑：

1. 用ASSERT语句简化结果检查 #

PostgreSQL的ASSERT语句可以在条件不满足时自动抛出异常，替代手动的IF NOT FOUND判断。只需确保数据库参数plpgsql.check_asserts已开启（可全局设置或在函数内临时开启）。

2. 利用PostgreSQL自动事务回滚机制 #

PL/pgSQL函数默认运行在调用者的事务上下文中，只要任一子函数抛出异常，整个事务会自动进入中止状态并回滚，无需手动调用ROLLBACK。

优化后的代码示例 #

优化后的子函数delete_relationship #

CREATE OR REPLACE FUNCTION delete_relationship(
    follower_id uuid,
    following_id uuid
)
    RETURNS void
    LANGUAGE plpgsql
    SECURITY INVOKER
AS
$$
BEGIN
    -- 临时开启断言（若全局未开启）
    SET LOCAL plpgsql.check_asserts = on;

    DELETE FROM following_profiles
    WHERE follower = follower_id AND following = following_id;

    -- 未找到目标行时自动抛出异常
    ASSERT FOUND, '未找到要删除的关注关系，操作失败';
END;
$$;

优化后的子函数decrement_counter #

CREATE OR REPLACE FUNCTION decrement_counter(
    following_id uuid
)
    RETURNS void
    LANGUAGE plpgsql
    SECURITY INVOKER
AS
$$
BEGIN
    SET LOCAL plpgsql.check_asserts = on;

    UPDATE user_profiles
    SET follower_count = follower_count - 1
    WHERE user_id = following_id;

    ASSERT FOUND, '未找到目标用户，无法更新关注计数';
END;
$$;

主函数无需额外修改 #

主函数unfollow保持原有逻辑即可，只要任一子函数抛出异常，整个事务会自动回滚：

CREATE OR REPLACE FUNCTION unfollow(
    follower_id uuid,
    following_id uuid
)
    RETURNS void
    LANGUAGE plpgsql
    SECURITY INVOKER
AS
$$
BEGIN
    PERFORM delete_relationship(follower_id, following_id);
    PERFORM decrement_counter(following_id);
END;
$$;

优势说明 #

• 代码更简洁：用一行ASSERT替代手动的条件判断和异常抛出，减少重复代码。
• 事务一致性保障：依赖PostgreSQL原生事务机制，任一子函数失败都会触发全局回滚，避免数据不一致。
• 异常信息更清晰：自定义的断言信息便于快速定位问题，比默认错误信息更直观。

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