Postgres:从存储函数返回结果或错误

时间:2018-05-07 11:02:19

标签: postgresql stored-procedures plpgsql stored-functions

我正在努力弄清楚如何从Postgres存储函数中最好地处理结果或错误返回到我的应用程序。

考虑以下设计的psudeocode示例:

app.get_resource(_username text)
    RETURNS <???>

BEGIN

    IF ([ ..user exists.. ] = FALSE) THEN
        RETURN 'ERR_USER_NOT_FOUND';
    END IF;

    IF ([ ..user has permission.. ] = FALSE) THEN
        RETURN 'ERR_NO_PERMISSION';
    END IF;

    -- Return the full user object.
    RETURN QUERY( SELECT 1 
        FROM app.resources
        WHERE app.resources.owner = _username);

END

该函数可能因特定错误而失败或成功并返回0或更多资源。

首先,我尝试创建一个自定义类型,以便始终在eachh函数中用作标准返回类型:

CREATE TYPE app.appresult AS (
  success boolean,
  error   text,
  result  anyelement
);

Postgres不允许这样做:

[42P16] ERROR: column "result" has pseudo-type anyelement

然后我发现了OUT参数并尝试了以下用途:

CREATE OR REPLACE FUNCTION app.get_resource(
    IN      _username   text,
    OUT     _result app.appresult -- Custom type 
                                  -- {success bool, error text}
)

RETURNS SETOF record
AS
$$
BEGIN

  IF 1 = 1 THEN -- just a test
    _result.success = false;
    _result.error   = 'ERROR_ERROR';
    RETURN NULL;
  END IF;

  RETURN QUERY(SELECT * FROM app.resources);

END;
$$
  LANGUAGE 'plpgsql' VOLATILE;

Postgres也不喜欢这样:

[42P13] ERROR: function result type must be app.appresult because of OUT parameters

还尝试了类似的功能但反过来了:返回自定义app.appresult对象并将OUT参数设置为&#34; SETOF RECORD&#34;。这也是不允许的。

最后,我使用

查看了Postgres异常处理
RAISE EXCEPTION 'ERR_MY_ERROR';

所以在示例函数中,我只是引发了这个错误并返回。 这导致驱动程序将错误发送回:

"ERROR:  ERR_MY_ERROR\nCONTEXT:  PL/pgSQL function app.test(text) line 6 at RAISE\n(P0001)"

这很容易解析,但以这种方式做事感觉不对。

解决此问题的最佳方法是什么? 是否可以拥有一个我可以返回的自定义AppResult对象?

类似的东西:

{ success bool, error text, result <whatever type> }

//编辑1 //

我想我更倾向于@Laurenz Albe解决方案。

我的主要目标很简单:调用存储过程,该存储过程可以返回错误或某些数据。

使用RAISE似乎可以实现这一点,C ++驱动程序可以轻松检查查询返回的错误情况。

if ([error code returned from the query] == 90100)
{
    // 1. Parse out my overly verbose error from the raw driver
    //    error string.
    // 2. Handle the error.    
}

我还想知道如何使用自定义SQLSTATE代码而不是解析驱动程序字符串。

投掷&#39; __ 404&#39;可能意味着在我的SP执行过程中,它无法继续,因为找不到所需的记录。

从我的应用程序调用sql函数时,我大致了解它失败了什么&#39; __ 404&#39;意味着如何处理它。这避免了解析驱动程序错误字符串的额外步骤。

我也可以看到这可能是一个坏主意。

睡前阅读: https://www.postgresql.org/docs/current/static/errcodes-appendix.html

2 个答案:

答案 0 :(得分:1)

这是基于意见的,但我认为抛出错误是最好和最优雅的解决方案。这就是错误!

要区分各种错误消息,可以使用以6,8或9开头的SQLSTATE(不使用这些消息),然后您不必依赖于错误消息的措辞。

您可以使用

引发此类错误
RAISE EXCEPTION SQLSTATE '90001' USING MESSAGE = 'my own error';

答案 1 :(得分:0)

我们会做类似于您尝试做的事情,但我们使用TEXT而不是ANYELEMENT,因为(差不多?)任何类型都可以转换为TEXT并返回。所以我们的类型看起来像:

(errors our_error_type[], result TEXT)

返回此函数会在errors数组中存储错误(它只是一些自定义错误类型),并且可以在result字段中存储结果(强制转换为文本)。

调用函数知道它所期望的类型,因此它可以首先检查errors数组以查看是否返回了任何错误,如果不是,它可以将result值转换为预期的返回类型

作为一般观察,我认为异常更优雅(可能是因为我来自c#背景)。唯一的问题是plpgsql异常处理是(相对)慢,所以它取决于上下文 - 如果你在循环中多次运行,我宁愿一个不使用异常处理的解决方案;如果它是一个单一的呼叫,和/或特别是当你想让它中止时,我更喜欢提出异常。在实践中,我们在整个调用堆栈的不同点使用它们。

正如Laurenz Albe指出的那样,你不是要“解析”异常,而是在特定字段中使用特定值引发异常,然后捕获异常的函数可以直接提取并作用。< / p>

举个例子:

设定:

CREATE TABLE my_table (id INTEGER, txt TEXT);
INSERT INTO my_table VALUES (1,'blah');

CREATE TYPE my_type AS (result TEXT);

CREATE OR REPLACE FUNCTION my_func()
RETURNS my_type AS
$BODY$
DECLARE
    m my_type;
BEGIN
    SELECT my_table::TEXT
        INTO m.result
    FROM my_table;

    RETURN m;
END
$BODY$
LANGUAGE plpgsql STABLE;

执行命令

SELECT (m.result::my_table).*
FROM my_func() AS m

结果:

| id | txt  |
-------------
| 1  | blah |