编写 SQL 触发器来查找列中出现的数字是否超过 X 次？

Question

我想编写一个 Postgres SQL 触发器，它基本上会发现一个数字是否在列中出现 5 次或更多次。如果它出现第五次，我想抛出一个异常。以下是表格的外观：

create table tab(
first integer not null constraint pk_part_id primary key,
second integer constraint fk_super_part_id references bom,
price integer);

insert into tab values(1,NULL,100), (2,1,50), (3,1,30), (4,2,20), (5,2,10), (6,3,20);

以上是表格中的原始插入。我的触发器将在向表中插入更多值时发生。

基本上，如果一个数字在插入表格后出现在“第二”列中超过 4 次，我想引发异常。这是我编写触发器的尝试：

create function check() return trigger as '
begin
    if(select first, second, price
          from tab
          where second in (
            select second from tab
            group by second
            having count(second) > 4)
) then
raise exception ''Error, there are more than 5 parts.'';
end if;
return null;
end 
'language plpgsql;

create trigger check
after insert or update on tab
for each row execute procedure check();

谁能帮帮我？如果是这样那就太好了！谢谢！

Answer 1

CREATE FUNCTION trg_upbef()
  RETURN trigger as
$func$
BEGIN
IF (SELECT count(*) 
    FROM   tab
    WHERE  second = NEW.second ) > 3 THEN

   RAISE EXCEPTION 'Error: there are more than 5 parts.';
END IF;

RETURN NEW;  -- must be NEW for BEFORE trigger

END
$func$ LANGUAGE plpgsql;

CREATE TRIGGER upbef
BEFORE INSERT OR UPDATE ON tab
FOR EACH ROW EXECUTE procedure trg_upbef();

要点

• 关键字是RETURNS，而不是RETURN。

• 使用special variable NEW 来引用新插入/更新的行。

• 使用BEFORE 触发器。发生异常时最好尽早跳过。

• 不要把一切都算在你的测试中，只需要你需要的。 快得多。

• 使用dollar-quoting。让您的生活更轻松。

• 并发：
  如果您想绝对确定，则必须在计数之前对表进行排他锁。否则，并发插入/更新可能会在并发负载较重的情况下相互超越。虽然这不太可能，但有可能。