在Postgres中有条件地将列设置为其默认值

时间:2010-04-20 04:38:14

标签: sql postgresql

我有一个PostgreSQL 8.4表,其中包含一个自动递增但可为空的整数列。我想更新一些列值,如果此列为NULL,则将其设置为其默认值(这是从序列自动生成的整数),但是我想要返回其值两种情况。所以我想要这样的东西:

UPDATE mytable
SET incident_id = COALESCE(incident_id, DEFAULT), other = 'somethingelse'
WHERE ...
RETURNING incident_id

不幸的是,这不起作用 - 似乎DEFAULT是特殊的,不能成为表达式的一部分。最好的方法是什么?

1 个答案:

答案 0 :(得分:4)

使用它:

update mytable set a = 
    coalesce(incidentid, 
    (
     select column_default::int 
     from information_schema.columns 
     where table_schema = 'public' 
     and table_name = 'mytable' and column_name = 'incidentid')
    )

如果你的incidentid是整数类型,请在column_default

上进行类型转换

<强> [编辑]

create or replace function get_default_value(_table_name text,_column_name text) 
returns text
as
$$
declare r record;
s text;
begin

    s = 'SELECT ' || coalesce(

    (select column_default 
    from information_schema.columns 
    where table_schema = 'public' 
    and table_name = _table_name and column_name = _column_name)

    , 'NULL') || ' as v';

    EXECUTE s into r;
    return r.v;
end;
$$
language 'plpgsql';

使用:

update mytable set a = 
    coalesce(incidentid, get_default_value('mytable','a')::int )