在 Postgresql 中创建函数，将数据插入到已更新的表行中

Question

我一直试图弄清楚如何创建一个触发器/函数，该触发器/函数将在更新表行时计算一个值。这是我正在创建的 DVD 出租数据库。我有一些数据库经验，但我才开始使用触发器和函数。我想要发生的是，当员工插入电影归还日期时，它将采用租借日期并将其从归还日期中减去，然后将天差乘以每天 1.99 美元的租金成本。我有计算它的表达式。我想要一个触发器来运行表达式并将结果插入到该表中。

例如：

客户返回电影，员工在 Rent 表中输入返回日期，当更新此行时，触发器将运行一个计算欠款金额的表达式，并将结果插入到 Rent 表行中的 amount_owed 列。

这是我一直在尝试的一些事情，触发器执行了该函数，但之后我得到了一个错误。

功能代码：

CREATE OR REPLACE FUNCTION update_cus_balance(integer)
  RETURNS void AS
$BODY$
BEGIN

UPDATE rent
SET amount_owed=subquery.balance

FROM (SELECT date_part('day'::text, rent.return_date - rent.rent_date) *         2.99::double precision AS balance

from rent) AS subquery
WHERE rent.rent_id=rent_id;
END
$BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100;
ALTER FUNCTION update_cus_balance(integer)
  OWNER TO postgres;

触发代码：

CREATE TRIGGER update_balance
AFTER UPDATE
ON rent
FOR EACH ROW
EXECUTE PROCEDURE amount_owed();

当我尝试添加 return_date 并保存更改时，PGadmin III 只会说“发生错误”。

Answer 1

就像 Jakub 指出的那样，您需要一个 BEFORE UPDATE 触发器来完成这项工作。这样的触发器有两个重要特征：

1. 您有一个OLD 和一个NEW 变量，它们保存更新前行的值以及更新后的样子。您可以操作 NEW 变量，然后将这些更改保存到表中。
2. 如果某些更改不合适，您可以中止整个更新。

对于您的情况，触发器函数看起来有点像这样：

CREATE OR REPLACE FUNCTION update_cus_balance(integer)
  RETURNS void trigger AS -- A trigger function should ALWAYS return trigger
$BODY$
BEGIN
  -- Calculate rent if the return date is set or changed
  IF NEW.return_date <> OLD.return_date THEN 
    IF NEW.return_date = NEW.rent_date THEN -- Same day returns
      NEW.amount_owed := 2.99;
    ELSE
      NEW.amount_owed := (NEW.return_date - NEW.rent_date) * 2.99;
    END IF;
  END IF;

  -- Trigger function should ALWAYS return NEW to succeed.
  -- Return NULL to make the update fail.
  RETURN NEW; 
END
$BODY$ LANGUAGE plpgsql;

触发器应该是：

CREATE TRIGGER update_balance
BEFORE UPDATE ON rent
FOR EACH ROW EXECUTE PROCEDURE update_cus_balance();

请注意，我假设 rent_date 和 return_date 的类型为 date（顾名思义）。如果它们是 timestamps，您可以创建两个变量和 EXTRACT() 租赁日期并返回到这些变量中。

PostgreSQL 中的触发器和函数不是很复杂，但请仔细阅读excellent manual。

Answer 2

rent 表上的 update 会触发 trigger，触发器会更新租用表。租金表上的更新触发了一个触发器，一个触发器......

这就是您收到stack depth limit exceeded 错误的原因。

可以轻松修复添加

AND amount_owed<>subquery.balance

到触发器中查询的WHERE 子句。但是，将触发器更改为 BEFORE UPDATE 会更好，为了保持一致性，它甚至应该是检查 return_date 是否已填充的 BEFORE INSERT OR UPDATE 触发器。

这看起来像是一个练习练习，所以你自己试试之前的触发器。