plpgsql 函数返回另一个函数值

Question

我有一个函数func1()，它返回整数。

我想编写另一个函数func2(mode integer)，它可以返回func1() 结果或做更多的事情。 func1() 的返回值是 INTEGER 类型。

类似这样的：

CREATE OR REPLACE FUNCTION func2(mode integer)
  RETURNS integer AS
$$
begin
     if mode=1 then
       return func1();  -- NOT plpgsql syntax
     end if;

     more stuff .....

     return 2;
end
$$
LANGUAGE plpgsql VOLATILE

我的问题是return func1(); 怎么办？

我知道我能做到：

select func1() into temp;
return temp;

但我想知道是否有更优雅的方法来做到这一点。

Answer 1

所有这些工作：

选项 1：

CREATE OR REPLACE FUNCTION func2(mode integer)
    RETURNS integer AS
$BODY$
DECLARE 
    _result integer;
BEGIN
    _result = 2;
     IF mode=1 THEN
       _result = func1();
     END IF;

     --more stuff .....

     RETURN _result;
END
$BODY$
    LANGUAGE plpgsql VOLATILE COST 100;

选项 2：

CREATE OR REPLACE FUNCTION func2(mode integer)
    RETURNS integer AS
$BODY$
BEGIN
     IF mode=1 THEN
       RETURN func1();
     END IF;

     --more stuff .....

     RETURN 2;
END
$BODY$
    LANGUAGE plpgsql VOLATILE COST 100;