一、为什么需要触发器？
在日常开发中，我们经常会遇到这样的需求：当数据库中的某张表发生数据变动时，自动执行一些额外的逻辑。比如，订单表新增一条记录时，自动更新库存数量；用户修改个人信息时，自动记录变更日志。如果每次都手动写代码处理这些逻辑，不仅繁琐，还容易遗漏。
这时候，触发器（Trigger）就派上用场了。触发器是数据库提供的一种机制，它能在特定的数据库操作（如 INSERT、UPDATE、DELETE）前后自动执行预定义的 SQL 或存储过程。PostgreSQL 的触发器功能非常强大，支持多种触发时机和条件，能够帮助我们实现复杂的业务自动化。
二、PostgreSQL 触发器基础
1. 触发器的基本语法
在 PostgreSQL 中，创建触发器主要分为两步：
- 先定义一个触发器函数（Trigger Function），它决定了触发器被激活时要执行的逻辑。
- 再创建触发器（Trigger），绑定到具体的表，并指定触发时机（BEFORE/AFTER）和触发事件（INSERT/UPDATE/DELETE）。
下面是一个最简单的触发器示例：
-- 1. 创建一个触发器函数，在插入数据时自动记录日志
CREATE OR REPLACE FUNCTION log_insert_trigger()
RETURNS TRIGGER AS $$
BEGIN
-- NEW 代表新插入的数据行
RAISE NOTICE '新数据插入: ID=%, 时间=%', NEW.id, NOW();
RETURN NEW; -- 必须返回 NEW 或 NULL
END;
$$ LANGUAGE plpgsql;
-- 2. 创建触发器，绑定到 users 表的 INSERT 操作
CREATE TRIGGER trg_log_user_insert
AFTER INSERT ON users
FOR EACH ROW EXECUTE FUNCTION log_insert_trigger();
2. 触发器的触发时机
PostgreSQL 支持以下几种触发时机：
- BEFORE：在操作执行前触发，适合数据校验或修改。
- AFTER：在操作执行后触发，适合记录日志或更新关联数据。
- INSTEAD OF：主要用于视图（View），替代原本的操作。
三、实战：用触发器实现自动化业务逻辑
1. 示例1：自动更新库存
假设我们有一个商品表 products 和一个订单表 orders，每次新增订单时，需要减少对应商品的库存。
-- 1. 创建触发器函数，减少库存
CREATE OR REPLACE FUNCTION update_stock_trigger()
RETURNS TRIGGER AS $$
BEGIN
-- 更新 products 表的库存
UPDATE products
SET stock = stock - NEW.quantity
WHERE id = NEW.product_id;
RETURN NEW;
END;
$$ LANGUAGE plpgsql;
-- 2. 创建触发器，在 orders 表插入数据后触发
CREATE TRIGGER trg_update_stock
AFTER INSERT ON orders
FOR EACH ROW EXECUTE FUNCTION update_stock_trigger();
2. 示例2：数据变更审计
有时候我们需要记录关键数据的变更历史，比如用户表的修改记录。
-- 1. 创建审计表
CREATE TABLE user_audit (
id SERIAL PRIMARY KEY,
user_id INT NOT NULL,
changed_field TEXT,
old_value TEXT,
new_value TEXT,
change_time TIMESTAMP DEFAULT NOW()
);
-- 2. 创建触发器函数，记录变更
CREATE OR REPLACE FUNCTION audit_user_changes()
RETURNS TRIGGER AS $$
BEGIN
-- 检查哪些字段被修改了
IF OLD.username <> NEW.username THEN
INSERT INTO user_audit (user_id, changed_field, old_value, new_value)
VALUES (OLD.id, 'username', OLD.username, NEW.username);
END IF;
IF OLD.email <> NEW.email THEN
INSERT INTO user_audit (user_id, changed_field, old_value, new_value)
VALUES (OLD.id, 'email', OLD.email, NEW.email);
END IF;
RETURN NEW;
END;
$$ LANGUAGE plpgsql;
-- 3. 创建触发器，在 users 表更新时触发
CREATE TRIGGER trg_audit_user_changes
AFTER UPDATE ON users
FOR EACH ROW EXECUTE FUNCTION audit_user_changes();
四、触发器的优缺点与注意事项
1. 优点
- 自动化：减少手动编码，提高开发效率。
- 数据一致性：确保业务逻辑在数据库层面强制执行，避免脏数据。
- 解耦：业务逻辑与应用程序分离，便于维护。
2. 缺点
- 调试困难：触发器逻辑隐藏在数据库中，排查问题较麻烦。
- 性能影响：频繁触发的复杂触发器可能拖慢数据库操作。
- 维护成本：随着业务变化，触发器可能变得难以管理。
3. 注意事项
- 避免递归触发：比如在触发器里更新同一张表，可能导致死循环。
- 谨慎使用 BEFORE 触发器：如果逻辑出错，可能导致原操作失败。
- 合理设计审计日志：避免记录过多无用信息，影响性能。
五、总结
PostgreSQL 触发器是一个非常强大的工具，能够帮助我们实现复杂的业务自动化逻辑。无论是库存管理、数据审计，还是实时计算，触发器都能提供可靠的解决方案。不过，使用触发器时也要注意性能和维护成本，避免过度依赖。
在实际项目中，建议结合存储过程、事件驱动架构（如 Kafka）等技术，灵活选择最适合的方案。
评论