Doctrine ORM 中 PostgreSQL 触发器导致 lastval() 值异常的解决方案

问题背景

在使用 Doctrine ORM 与 PostgreSQL 数据库交互时，开发者可能会遇到一个特殊的问题：当在实体表上设置了 AFTER INSERT 触发器后，多对多关联关系的持久化操作会失败，并抛出外键约束违反异常。

这个问题的核心在于 PostgreSQL 的序列值获取机制与 Doctrine ORM 的 ID 生成策略之间的微妙交互。具体表现为，当触发器向日志表插入记录时，Doctrine 错误地获取了日志表的序列值，而非目标实体表的序列值。

问题重现场景

假设我们有以下数据结构：

1. 两个实体 EntityA 和 EntityB，它们之间存在多对多关联关系
2. 关联通过中间表 tableA_tableB 实现
3. 两个实体表的主键都使用 PostgreSQL 的 serial 类型（自动递增）
4. 在 tableA 表上设置了 AFTER INSERT 触发器，用于记录操作日志到 log_audit 表

当同时持久化新的 EntityA 和 EntityB 实例时，Doctrine 会抛出如下异常：

SQLSTATE[23503]: Foreign key violation: 7 ERROR: insert or update on table "tableA_tableB" violates foreign key constraint
DETAIL: Key (tableA_id)=(18) is not present in table "tableA".

问题根源分析

这个问题的根本原因在于 PostgreSQL 的 lastval() 函数行为和 Doctrine ORM 的 ID 生成机制：

1. Doctrine 使用 IdentityGenerator 为 serial 类型的主键生成 ID
2. 生成过程中会调用 PDO::lastInsertId() 获取最后插入的序列值
3. PostgreSQL 的 lastval() 函数返回当前会话中最近一次调用的序列值
4. 触发器中的插入操作会"偷走"序列值，导致 Doctrine 获取到的是日志表的序列值，而非实体表的序列值

解决方案

方案一：修改日志表设计（推荐）

最简单的解决方案是修改日志表的结构，不使用 serial 类型的主键：

ALTER TABLE log_audit DROP COLUMN id;
ALTER TABLE log_audit ADD COLUMN id uuid PRIMARY KEY DEFAULT gen_random_uuid();

使用 UUID 或其他非序列类型的主键可以避免触发器影响实体表的序列值获取。

方案二：调整触发器设计

另一种方法是修改触发器，使其不影响序列值：

1. 将触发器改为行级触发器（FOR EACH ROW）而非语句级触发器
2. 在触发器中使用 RETURN NEW 而非 RETURN NULL，确保原操作继续执行
3. 显式指定日志记录的 ID，避免自动生成

CREATE OR REPLACE FUNCTION tableA_ait() RETURNS TRIGGER 
LANGUAGE plpgsql AS $$
BEGIN
  INSERT INTO log_audit (id, log_data, log_table, operation) 
  VALUES (gen_random_uuid(), row_to_json(NEW.*), 'tableA', TG_OP);	
  RETURN NEW;
END;
$$;

CREATE TRIGGER tableA_ait AFTER INSERT ON tableA
FOR EACH ROW EXECUTE PROCEDURE tableA_ait();

方案三：自定义 ID 生成策略

对于高级用例，可以创建自定义的 ID 生成器，绕过 PostgreSQL 的 lastval() 问题：

use Doctrine\ORM\Id\AbstractIdGenerator;

class CustomPostgresGenerator extends AbstractIdGenerator
{
    public function generateId(EntityManager $em, $entity)
    {
        $conn = $em->getConnection();
        $sequenceName = $em->getClassMetadata(get_class($entity))->getSequenceName($conn->getDatabasePlatform());
        
        return $conn->fetchOne("SELECT nextval('".$sequenceName."')");
    }
}

然后在实体映射中指定使用此生成器：

/**
 * @Entity
 * @Table(name="tableA")
 */
class EntityA
{
    /**
     * @Id
     * @Column(type="integer")
     * @GeneratedValue(strategy="CUSTOM")
     * @CustomIdGenerator(class="CustomPostgresGenerator")
     */
    private $id;
    
    // ...
}

最佳实践建议

1. 在 PostgreSQL 中使用触发器时，尽量避免影响序列值
2. 考虑使用 UUID 作为日志表的主键类型
3. 对于关键业务表，建议进行充分的集成测试，特别是涉及复杂触发器和关联关系时
4. 在 Doctrine 中，了解不同数据库的 ID 生成策略差异很重要

通过理解 PostgreSQL 的序列机制和 Doctrine ORM 的工作原理，开发者可以避免这类隐蔽的问题，构建更健壮的数据库应用。