postgresql - 处理异常并从函数返回结果

Question

这是我的代码

CREATE OR REPLACE FUNCTION test_excep (arg INTEGER) RETURNS INTEGER 
AS $$
    DECLARE res INTEGER;
    BEGIN
        res := 100 / arg;

        BEGIN
            EXCEPTION
            WHEN division_by_zero 
            THEN  RETURN 999;
        END;


        RETURN res;
    END;



$$
LANGUAGE plpgsql;

也就是说，如果发生被零除，我需要返回“999”，但是：SELECT test_excep(0)

返回错误： division by zero CONTEXT: PL/pgSQL function test_excep(integer) line 4 at assignment

我的代码有什么问题？

Answer 1

该EXCEPTION子句需要与异常位于同一块中。

例如：

CREATE OR REPLACE FUNCTION test_excep (arg integer)
  RETURNS integer
AS
$func$
DECLARE
   res INTEGER;
BEGIN

res := 100 / arg;

RETURN res;

EXCEPTION
    WHEN division_by_zero 
    THEN  RETURN 999;

END
$func$
LANGUAGE plpgsql;

Answer 2

这是一个正确的函数：

CREATE OR REPLACE FUNCTION test_excep (arg INTEGER) RETURNS INTEGER 
AS $$
    DECLARE res INTEGER;
    BEGIN
        res := 100 / arg;
        RETURN res;
    EXCEPTION
        WHEN division_by_zero 
        THEN  RETURN 999;
    END;
$$
LANGUAGE plpgsql;

EXCEPTION部分必须在异常所在的块内，也必须是块的最后一部分。