检测 PostgreSQL SQL 函数（不是 PL/pgSQL）中的 UPDATE 失败

Question

为了模仿 MySQL-REPLACE 语句（又名 UPSERT），我需要尝试更新记录，如果失败 - 插入它。但是我如何检测 SQL 过程中的 UPDATE 失败呢？

begin transaction;

create table pref_users (
       id varchar(32) primary key,
       first_name varchar(32),
       last_name varchar(32),
       female boolean,
       avatar varchar(128),
       city varchar(32),
       lat real check (-90 <= lat and lat <= 90),
       lng real check (-90 <= lng and lng <= 90),
       last_login timestamp default current_timestamp,
       last_ip inet,
       medals smallint check (medals > 0)
);
create table pref_rate (
       obj varchar(32) references pref_users(id),
       subj varchar(32) references pref_users(id),
       good boolean,
       fair boolean,
       nice boolean,
       about varchar(256),
       last_rated timestamp default current_timestamp
);

create table pref_money (
       id varchar(32) references pref_users,
       yw char(7) default to_char(current_timestamp, 'YYYY-WW'),
       money real
);
create index pref_money_yw_index on pref_money(yw);

create or replace function update_pref_users(id varchar,
       first_name varchar, last_name varchar, female boolean,
       avatar varchar, city varchar, last_ip inet) returns void as $$

       update pref_users set
            first_name = $2,
            last_name = $3,
            female = $4,
            avatar = $5,
            city = $6,
            last_ip = $7
        where id = $1;

        -- XXX how to detect failure here? XXX

       insert into pref_users(id, first_name, last_name,
            female, avatar, city, last_ip)
            values ($1, $2, $3, $4, $5, $6, $7);
$$ language sql;

commit;

我的 update_pref_users SQL 函数中是否需要第二对 BEGIN/COMMIT ？

Answer 1

你不能使用 SQL 作为一种语言，你需要 pl/pgsql，因为 SQL 中缺乏 if-else 结构。

在 pl/pgsql 中，您可以使用特殊变量 FOUND 查看查询是否找到了某些内容。

UPDATE ...;
IF NOT FOUND THEN -- UPDATE didn't touch anything
  INSERT ...;
END IF;

不需要额外的 SELECT 语句。

Answer 2

IF EXISTS(<query to select record required for update>)
    UPDATE ...
ELSE
    INSERT ...