输入参数数量可变的函数

Question

我正在 PostgreSQL 数据库中创建一个存储过程（函数），它根据输入更新表。为了创建可变数量的参数函数，我创建了一个名为 mode 的额外输入参数，用于控制在更新查询中使用哪些参数。

CREATE OR REPLACE FUNCTION update_site(
    mode integer,
    name character varying,
    city character varying,
    telephone integer,
)
RETURNS integer AS
$$
BEGIN
IF mode = 0 THEN
BEGIN
    UPDATE "Sites" SET 
    ("City","Telephone") = (city,telephone)
    WHERE "SiteName" = name;
    RETURN 1;
    EXCEPTION WHEN others THEN
    RAISE NOTICE 'Error on site update: %, %',SQLERRM,SQLSTATE;
    RETURN 0;
END;
ELSIF mode = 1 THEN
BEGIN
    UPDATE "Sites" SET "City" = city
    WHERE "SiteName" = name;
    RETURN 1;
    EXCEPTION WHEN others THEN
    RAISE NOTICE 'Error on site update: %, %',SQLERRM,SQLSTATE;
    RETURN 0;
END;
    ELSIF mode = 2 THEN
BEGIN
    UPDATE "Sites" SET "Telephone" = telephone
    WHERE "SiteName" = name;
    RETURN 1;
    EXCEPTION WHEN others THEN
    RAISE NOTICE 'Error on site update: %, %',SQLERRM,SQLSTATE;
    RETURN 0;
END;
    ELSE
            RAISE NOTICE 'Error on site update: %, %',SQLERRM,SQLSTATE;
    RETURN 0;
    END IF;
END;
$$ LANGUAGE plpgsql;

什么是最好的？要创建一个函数update_site(<all the columns of table>) 和一个单独的函数update_site(id integer, <varchar column to update>)，还是使用一个函数中的模式来定义区别？哪个选项更有效？一种独特的功能还是针对每种用途的不同功能？

Answer 1

像VARIADIC 甚至polymorphic input types 和动态SQL 等高级功能非常强大。此答案的最后一章提供了一个高级示例：

• Refactor a PL/pgSQL function to return the output of various SELECT queries

但是对于像您这样的简单情况，您可以将默认值用于函数参数。这完全取决于确切的要求。
如果所讨论的列都定义为NOT NULL，这可能会更简单、更快：

CREATE OR REPLACE FUNCTION update_site(_name      text    -- always required
                                     , _city      text    DEFAULT NULL
                                     , _telephone integer DEFAULT NULL)
  RETURNS integer AS
$func$
BEGIN
   IF _city IS NULL AND _telephone IS NULL THEN
      RAISE WARNING 'At least one value to update required!';
      RETURN;  -- nothing to update
   END IF;

   UPDATE "Sites"
   SET    "City"      = COALESCE(_city, "City")
        , "Telephone" = COALESCE(_telephone, "Telephone")
   WHERE  "SiteName"  = _name;
END
$func$  LANGUAGE plpgsql;

了解default values in the manual!

为了避免参数和列名之间的命名冲突，我习惯于在输入参数前加上_。这是品味和风格的问题。

• 第一个参数name 没有默认值，因为它始终是必需的。
• 其他参数可以省略。
• 至少需要一个，否则会引发 WARNING 并且不会发生其他任何事情。
• UPDATE 只会更改给定参数的列。
• 可以轻松扩展N个参数。

函数调用

自 Postgres 9.5 起：

简单的方法是使用positional notation 作为参数。这只允许省略最右边的参数：

SELECT update_site('foo', 'New York');  -- no telephone

Named notation 允许省略 any 具有默认值的参数：

SELECT update_site(name => 'foo', _telephone => 123);  -- no city

两者都可以组合在mixed notation:

SELECT update_site('foo', _telephone => 123);  -- still no city

在 Postgres 9.4 或更早版本中，:= 用于调用中的分配：

SELECT update_site(name := 'foo', _telephone := 123);
SELECT update_site('foo', _telephone := 123);

为了向后兼容在 Postgres 12 中仍然有效，但要使用现代符号。

Answer 2

您需要研究以下几点：

• 使用format 函数及其%I 和%L 说明符动态构建SQL，然后使用EXECUTE ... USING 执行它；和

• 使用VARIADIC 参数将可变数量的参数传递给函数，但需要注意的是它们都必须是相同的数据类型。