PostgreSQL含UPDATE语句的触发器函数语法错误排查

问题根源 #

从你给出的错误提示（SET SET quantity_available）可以明确：你在编写触发器函数时，UPDATE语句里重复写了两次SET关键字，这是直接导致语法错误的原因。虽然你提供的函数代码里是正确的，但实际执行时应该是手滑多输入了一个SET。

修正后的触发器函数 #

同时我们可以优化函数逻辑，直接在UPDATE语句中计算库存变化，减少不必要的变量声明，并且完善INSERT和UPDATE两种场景的处理：

CREATE OR REPLACE FUNCTION update_available_quantity_func() 
RETURNS TRIGGER 
LANGUAGE plpgsql
AS $$
BEGIN
    -- 处理INSERT：新增租赁记录，库存减少对应数量
    IF TG_OP = 'INSERT' THEN
        -- 先检查库存是否足够
        IF (SELECT quantity_available FROM equipment WHERE equipment_id = NEW.equipment_id) < NEW.quantity THEN
            RAISE EXCEPTION '库存不足，equipment_id = %，请求租赁数量 = %，可用库存 = %', 
                NEW.equipment_id, 
                NEW.quantity, 
                (SELECT quantity_available FROM equipment WHERE equipment_id = NEW.equipment_id);
        END IF;
        -- 更新库存
        UPDATE equipment
        SET quantity_available = quantity_available - NEW.quantity
        WHERE equipment_id = NEW.equipment_id;
    -- 处理UPDATE：修改租赁数量，计算新旧数量差调整库存
    ELSIF TG_OP = 'UPDATE' THEN
        -- 计算数量变化：旧数量 - 新数量 = 库存需要增加的量（如果新数量比旧的多，差值为负，即库存减少）
        DECLARE
            quantity_diff INT := OLD.quantity - NEW.quantity;
            new_available INT;
        BEGIN
            new_available := (SELECT quantity_available + quantity_diff FROM equipment WHERE equipment_id = NEW.equipment_id);
            IF new_available < 0 THEN
                RAISE EXCEPTION '调整后库存为负，equipment_id = %，调整后库存 = %', NEW.equipment_id, new_available;
            END IF;
            UPDATE equipment
            SET quantity_available = new_available
            WHERE equipment_id = NEW.equipment_id;
        END;
    END IF;

    -- 输出更新通知
    RAISE NOTICE 'equipment_id = %，当前可用库存 = %', 
        NEW.equipment_id, 
        (SELECT quantity_available FROM equipment WHERE equipment_id = NEW.equipment_id);

    RETURN NEW;
END;
$$;

修正后的触发器 #

你的原触发器只处理了INSERT场景，需要扩展为同时触发INSERT和UPDATE操作，并且建议使用BEFORE触发时机（可以在操作执行前检查库存，避免无效的插入/更新操作）：

CREATE TRIGGER adjust_equipment_available_trigger
BEFORE INSERT OR UPDATE ON equipment_rented
FOR EACH ROW 
EXECUTE PROCEDURE update_available_quantity_func();

关键说明 #

1. 语法错误修复：确保UPDATE语句只有一个SET关键字，这是解决你当前报错的核心。
2. 触发时机调整：使用BEFORE替代AFTER，可以在租赁记录写入前验证库存，避免出现"租赁记录已创建但库存更新失败"的不一致情况。
3. 完整场景覆盖：同时处理INSERT（新增租赁）和UPDATE（修改租赁数量）两种情况，当租赁数量减少时，库存会自动回补。
4. 库存校验：在修改库存前先检查是否会出现负数，提前抛出异常终止操作，保证数据一致性。

内容的提问来源于stack exchange，提问作者bananen