Postgres存储函数：如何向应用返回结果或错误？附示例

嘿，针对Postgres存储函数返回结果或错误的最佳处理方式，结合你给出的场景，我整理了几个实用的方案，帮你理清思路：

这种方式最贴合Postgres的原生错误处理逻辑，相当于告诉数据库“这个情况是异常，必须中断当前流程”，应用层只需要捕获对应的SQL异常就能处理错误，不用额外判断返回值格式。

结合你的伪代码，改造后的函数示例：

CREATE OR REPLACE FUNCTION app.get_resource(_username text) 
RETURNS SETOF app.resources AS $$
BEGIN
    -- 检查用户是否存在
    IF NOT EXISTS (SELECT 1 FROM app.users WHERE username = _username) THEN
        RAISE EXCEPTION 'ERR_USER_NOT_FOUND: User % does not exist', _username;
    END IF;

    -- 检查用户权限
    IF NOT EXISTS (SELECT 1 FROM app.user_permissions up 
                   WHERE up.username = _username AND up.permission = 'access_resource') THEN
        RAISE EXCEPTION 'ERR_NO_PERMISSION: User % has no access to resources', _username;
    END IF;

    -- 权限校验通过，返回资源数据
    RETURN QUERY
        SELECT * FROM app.resources 
        WHERE owner = _username;
END;
$$ LANGUAGE plpgsql;

应用层可以通过数据库驱动捕获异常，解析错误消息里的标识（比如ERR_USER_NOT_FOUND）来执行对应的错误逻辑，大多数ORM框架都支持这种异常捕获。

如果不想抛出异常，希望函数返回一个包含“状态标识+数据”的统一结构，可以先定义一个复合类型，让函数返回这个类型：

-- 先定义复合类型，包含状态和资源数据
CREATE TYPE app.resource_result AS (
    status text,
    resource app.resources
);

CREATE OR REPLACE FUNCTION app.get_resource(_username text) 
RETURNS app.resource_result AS $$
DECLARE
    result app.resource_result;
BEGIN
    IF NOT EXISTS (SELECT 1 FROM app.users WHERE username = _username) THEN
        result.status := 'ERR_USER_NOT_FOUND';
        RETURN result;
    END IF;

    IF NOT EXISTS (SELECT 1 FROM app.user_permissions up 
                   WHERE up.username = _username AND up.permission = 'access_resource') THEN
        result.status := 'ERR_NO_PERMISSION';
        RETURN result;
    END IF;

    -- 成功获取资源，赋值并返回
    SELECT * INTO result.resource
    FROM app.resources 
    WHERE owner = _username
    LIMIT 1; -- 如果是单条资源，多条可改为返回SETOF复合类型

    result.status := 'SUCCESS';
    RETURN result;
END;
$$ LANGUAGE plpgsql;

这种方式的好处是应用层不用处理异常，直接读取返回值的status字段判断结果；缺点是需要维护额外的复合类型，批量返回资源时还要调整返回类型为SETOF app.resource_result。

如果函数需要返回多条资源，同时要携带错误状态，可以在结果集里新增一个状态列：

CREATE OR REPLACE FUNCTION app.get_resource(_username text) 
RETURNS TABLE(status text, resource app.resources) AS $$
BEGIN
    IF NOT EXISTS (SELECT 1 FROM app.users WHERE username = _username) THEN
        status := 'ERR_USER_NOT_FOUND';
        RETURN NEXT;
        RETURN; -- 终止后续执行
    END IF;

    IF NOT EXISTS (SELECT 1 FROM app.user_permissions up 
                   WHERE up.username = _username AND up.permission = 'access_resource') THEN
        status := 'ERR_NO_PERMISSION';
        RETURN NEXT;
        RETURN;
    END IF;

    -- 校验通过，返回所有符合条件的资源，状态标记为SUCCESS
    RETURN QUERY
        SELECT 'SUCCESS'::text, * 
        FROM app.resources 
        WHERE owner = _username;
END;
$$ LANGUAGE plpgsql;

应用层可以先读取第一条结果的status字段，如果是错误标识就直接处理；如果是SUCCESS，再遍历所有返回的资源数据。

• 如果错误是不可恢复、需要中断流程的（比如用户不存在、无权限），优先用抛出异常，语义清晰，符合数据库原生设计，应用层也能统一处理SQL异常。
• 如果需要在同一个返回中同时携带状态和数据，或者希望柔性处理错误不中断流程，再考虑复合类型或带状态的结果集方案。

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