插入一个 COALESCE(NULL,default)

Question

我有使用 UUID 的表。我希望能够插入带有或不带有 UUID 的新行，因为有时客户端会生成 UUID，而有时它不会。

每张桌子的核心都是这个：

CREATE TABLE IF NOT EXISTS person (
    id UUID PRIMARY KEY DEFAULT gen_random_uuid()
);

我正在尝试使用函数来插入行。我希望能够提供一个 NULL id 并获得一个默认值（生成的 UUID）。我有这样的事情：

CREATE OR REPLACE FUNCTION create_person(
    id UUID
) RETURNS BOOLEAN LANGUAGE plpgsql SECURITY DEFINER AS $$
BEGIN
    INSERT INTO person( id )
    VALUES (
        COALESCE(id,default)
    );
    RETURN FOUND;
END;
$$;

我试过了：

INSERT INTO person ( id ) VALUES (
    COALESCE(id, default),
);

还有这个：

INSERT INTO person ( id ) VALUES (
  CASE WHEN id IS NULL THEN default ELSE id END
);

这可行，但它重复了 gen_random_uuid() 代码：

INSERT INTO person ( id ) VALUES (
  COALESCE(id, gen_random_uuid()),
);

同样，这也有效，但有同样的问题：

INSERT INTO person ( id ) VALUES (
    CASE WHEN id IS NULL THEN gen_random_uuid() ELSE id END
);

有没有办法让我不必重复 gen_random_uuid() 代码？

使用plpgsql 会更好吗？

Answer 1

无法重新使用列上定义的默认值。默认值仅用于定义INSERT 未指定值时会发生什么。根据这个定义，null 值仍然是“指定的”，因此不能使用默认值。

您对某人可能不会使用该函数的评论表明触发器比简单函数更符合您的要求。

https://www.postgresql.org/docs/current/static/plpgsql-trigger.html

CREATE OR REPLACE FUNCTION default_id() RETURNS TRIGGER AS $default_id$
    BEGIN       
        IF (NEW.id IS NULL) THEN
            NEW.id := gen_random_uuid();
        END IF;
        RETURN NEW;
    END;
$default_id$ LANGUAGE plpgsql;

CREATE TRIGGER default_id_trigger
BEFORE INSERT OR UPDATE ON person
    FOR EACH ROW EXECUTE PROCEDURE default_id();

---

如果您确实想使用函数执行此操作，那么最简单的方法就是在插入之前分配值：

CREATE OR REPLACE FUNCTION create_person(
    id UUID
) RETURNS BOOLEAN LANGUAGE plpgsql SECURITY DEFINER AS $$
BEGIN
    IF id IS NULL THEN
        id := gen_random_uuid();
    END IF;
    -- OR
    -- id := coalesce(id, gen_random_uuid());
    INSERT INTO person( id )
    VALUES (id);
    RETURN FOUND;
END;
$$;

Answer 2

核心问题是附加到INSERT 的VALUES 表达式中的关键字DEFAULT 的特殊性质。 Per documentation:

在出现在INSERT 顶层的VALUES 列表中， 表达式可以替换为DEFAULT，表示目的地 应该插入列的默认值。 DEFAULT 不能在以下情况下使用 VALUES 出现在其他上下文中。

我的大胆强调。具体来说，DEFAULT 不能作为函数的参数：

<strike>COALESCE(<i>function_parameter</i>, DEFAULT)</strike>  -- not possible

可能的解决方案

有多种方法，具体取决于确切要求。

这个函数不需要知道person.id的实际默认值——这似乎是你所追求的：

CREATE OR REPLACE FUNCTION create_person(_id UUID)
  RETURNS boolean LANGUAGE plpgsql SECURITY DEFINER AS
$func$
BEGIN
   IF _id IS NULL THEN                 -- no UUID provided
      INSERT INTO myschema.person(id)  -- see below about schema name
      VALUES (DEFAULT);                -- use the key word DEFAULT
   ELSE                                -- UUID provided
      INSERT INTO myschema.person(id)
      VALUES (_id);
   END IF;

   RETURN FOUND;                       -- (return value pointless so far)
END
$func$;

避免对参数和涉及的表列使用相同的名称。由于函数参数在函数体中的每个 SQL 命令中都是可见的，因此可能会导致非常混乱的命名冲突（即使 INSERT 的目标列在现代 Postgres 中不受此限制）。我改用_id 作为参数名。

INSERT 中未提及的其他列的默认值自动填写。我使用关键字DEFAULT，因为我们需要为INSERT 列出至少一个目标列。

boolean 返回值在此演示中毫无意义，因为它总是 true（除非您有可能跳过该行的触发器）。

可能的替代方案和大量解释的相关答案：

• Generate DEFAULT values in a CTE UPSERT using PostgreSQL 9.3

另外：您应该在 SECURITY DEFINER 函数中对 all 函数和表名进行模式限定 - 或（如果您不确定，可能会更好）明确设置search_path 以防御可能的攻击。更多：

• How does the search_path influence identifier resolution and the "current schema"

Answer 3

重载gen_random_uuid函数：

create or replace function gen_random_uuid()
returns uuid as $$
    select 'a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11'::uuid;
$$ language sql;


create or replace function gen_random_uuid(_id uuid)
returns uuid as $$
    select coalesce(_id, gen_random_uuid());
$$ language sql;

照常创建表格：

create table if not exists person (
    id uuid default gen_random_uuid()
);

在create_person函数内部调用gen_random_uuid函数：

create or replace function create_person(_id uuid)
returns boolean as $$
    insert into person (id) values (gen_random_uuid(_id))
    returning true;
$$ language sql;

然后所有插入变体都将起作用：

insert into person (id) values (default);
select create_person(null);
select create_person('b1eebc99-9c0b-4ef8-bb6d-6bb9bd380a22'::uuid);