浅谈 DWS 函数出参方式

2023-03-17
广东
本文字数：1115 字
阅读完需：约 4 分钟

本文分享自华为云社区《GassDB(DWS)功能 -- 函数出参 #【玩转PB级数仓GaussDB(DWS)】》，作者：譡里个檔。

DWS 的 PL/pgSQL 函数/存储过程中有一个特殊的语法 PERFORM 语法，用于执行语句但是丢弃执行结果的场景，常用于一些状态判断的场景。但是客户往往会不当使用 PERFORM 语法，导致业务逻辑出错，最常见的就是使用 PERFORM 执行带有出参的函数。

已知函数 inner 定义如下：

CREATE OR REPLACE FUNCTION public.inner(    IN a1 integer,     IN b1 integer,     OUT a integer,     OUT b integer) RETURNS record LANGUAGE plpgsql NOT FENCED NOT SHIPPABLEAS $function$ DECLAREBEGIN    a := a1;    b := b1;END$function$;

复制代码

函数 f_outer 定义如下，函数体中调用函数 inner，把函数的出参赋值给变量 a, b

CREATE OR REPLACE FUNCTION public.f_outer(IN i_a int, IN i_b int) RETURNS void LANGUAGE plpgsql NOT FENCED NOT SHIPPABLEAS $function$ DECLARE    a int;    b int;BEGIN    PERFORM public.inner(i_a, i_b, a, b);    RAISE INFO 'a = %, b = %', a, b;END$function$;

复制代码

但是实际执行的时候发现函数 inner 的出参没有正确赋值(预期值为 a = 1 b = 11)。

postgres=# CALL f_outer(1, 11);INFO:  a = <NULL>, b = <NULL>SQLSTATE: 00000 f_outer---------
(1 row)
Time: 1.086 ms

复制代码

出现这种问题的原因是 PERFORM 语法会执行 SQL 语句，但是会抛弃执行结果，导致函数出参没有赋值

DWS 中常用的带出函数出参的方式有以下三种：

方式 1：函数出参的方式赋值

CREATE OR REPLACE FUNCTION public.test1(IN i_a int, IN i_b int) RETURNS void LANGUAGE plpgsql NOT FENCED NOT SHIPPABLEAS $function$ DECLARE    a int;    b int;BEGIN    public.inner(i_a, i_b, a, b);    RAISE info 'a = %, b = %', a, b;END$function$;

复制代码

方式 2：动态查询语句方式赋值

CREATE OR REPLACE FUNCTION public.test2(IN i_a int, IN i_b int) RETURNS void LANGUAGE plpgsql NOT FENCED NOT SHIPPABLEAS $function$ DECLARE    a int;    b int;BEGIN    EXECUTE IMMEDIATE 'SELECT * from public.inner(:1, :2)' UNSING INTO a, b USING IN i_a, i_b;    RAISE INFO 'a = %, b = %', a, b;END$function$;

复制代码

方式 3：SELECT .. INTO 赋值

CREATE OR REPLACE FUNCTION public.test3(IN i_a int, IN i_b int) RETURNS void LANGUAGE plpgsql NOT FENCED NOT SHIPPABLEAS $function$ DECLARE    a int := 0;    b int := 0;BEGIN    SELECT * INTO a, b FROM public.inner(i_a, i_b);    RAISE INFO 'a = %, b = %', a, b;END$function$;