Question

我试过了：

CREATE or replace FUNCTION create_patient(_name text, _email text, _phone text
     , _password text
     , _field1 text, _field2 text, _field3 timestamp, _field4 text
     , OUT _pid integer, OUT _id integer
    ) RETURNS record AS
$$
DECLARE
    _id integer;
    _type text;
    _pid integer;
BEGIN
    _type := 'patient';

    INSERT into patients (name, email, phone, field1, field2, field3)
    VALUES (_name, _email, _phone, _field1, _field2, _field3)
    RETURNING id into _pid;

    INSERT into users (username, password, type, pid, phone, language)
    VALUES (_email, _password, _type, _pid, _phone, _field4)
    RETURNING id into _id;
END;
$$ LANGUAGE plpgsql;

但是有很多实例我不希望指定一些field1 / field2 / field3 / field4并希望未指定的字段使用表中的默认值。目前这是不可能的，因为要调用此函数我需要指定所有字段。

是否有一种简单的方法可以在PL / pgSQL中为INSERT创建包装程序，我可以在其中指定要插入的字段？

Answer 1

使用default values作为功能参数和dynamic SQL ...

演示功能

CREATE OR REPLACE FUNCTION create_patient(
     _name     text      = NULL  -- always updated, NULL if not provided
    ,_email    text      = NULL
    ,_phone    text      = NULL
    ,_password text      = NULL
    ,_field1   text      = NULL  -- variable parameters
    ,_field2   text      = NULL
    ,_field3   timestamp = NULL
    ,_language text      = NULL
    ,_cols     text[]    = '{field1,field2,field3,language}'
    ,OUT _pid  text
    ,OUT _id   int)
  RETURNS record AS
$func$
BEGIN
   EXECUTE format(
   'INSERT INTO patients (field1, field2, field3, name, email, phone)
    VALUES               (%s,     %s,     %s,     $4,   $5,    $6   )
    RETURNING id'
       ,CASE WHEN 'field1' = ANY(_cols) THEN '$1' ELSE 'DEFAULT' END
       ,CASE WHEN 'field2' = ANY(_cols) THEN '$2' ELSE 'DEFAULT' END
       ,CASE WHEN 'field3' = ANY(_cols) THEN '$3' ELSE 'DEFAULT' END)
    INTO  _pid                       -- return value, also used in 2nd insert
    USING _field1, _field2, _field3, _name, _email, _phone;

   EXECUTE format(
   'INSERT INTO users (username, password, type,        pid, phone, language)
    VALUES            ($1,       $2,       $$patient$$, $3,  $4,    %s      )
    RETURNING id'
      ,CASE WHEN 'language' = ANY(_cols) THEN '$4' ELSE 'DEFAULT' END)
    INTO  _id                        -- return value
    USING _email, _password, _pid, _phone, _language;
END
$func$  LANGUAGE plpgsql;

呼叫

SELECT * FROM create_patient('myname','myemail','myphone','mypassword'
,'myfield1','myfield2',NULL,'English','{field2,language,field1}'::text[]);

由于该函数使用命名参数并且每个参数都有一个默认值，您甚至可以这样调用它：

SELECT * FROM create_patient(_name := 'myname');

如果需要某些非空值，则可能无法为您的表工作，但是为了证明您提供命名参数后可以省略任何带有默认值的参数。省略的参数采用声明的默认值（不要与列默认值混淆）。更多相关答案：
Functions with variable number of input parameters

重点

使用DEFAULT keyword of the INSERT command。它使系统插入表的列默认值另一种方法是仅列出INSERT行中获取VALUES行中相应项目的列。
您必须使用dynamic SQL and EXECUTE来操纵语句本身，而不仅仅是值。
“Swing columns”为field1到field3和language，其余按照定义硬连线。根据需要变化。
我的功能适用于所有情况，您甚至可以提供NULL值而不是列默认值。这需要一个参数_cols，提供要插入哪些列的信息。
如果所有涉及的列都已声明为NOT NULL - 尚未澄清 - 您可以简化：为应该获取列默认值的任何列传递NULL并调整{{1} }陈述。
如果省略CASE，则会插入所有字段。由于_cols是最后一个_cols参数并且具有默认值，因此您可以始终省略它。
我使用USING clause for EXECUTE将参数作为值传递，并使用动态构建的查询字符串阻止SQL注入。
我使用IN来简化语句汇编并避免多次分配。 PL / pgSQL便宜。
请勿在函数体中format() DECLARE和_id，因为它们在标头中由_pid参数声明并自动返回。
您可以直接在OUT语句中插入type的常量值。这样您就不需要任何变量并保存其他分配。
使用PostgreSQL 9.1 进行测试，但自8.4以来应该适用于所有版本 - 除了9.1引入的format()。

用于INSERT多个表中的动态列列表的函数

1 个答案:

演示功能

呼叫

重点