TypeORM中复合主键更新时的并发问题解析

在使用TypeORM进行数据库操作时，开发人员可能会遇到一个关于复合主键更新的并发问题。本文将深入分析这一问题的成因、表现及解决方案。

问题现象

当使用TypeORM的save()方法更新具有复合主键的实体时，如果多个保存操作同时执行且未正确处理异步流程，可能会抛出"duplicate key value violates unique constraint"错误。这种错误表明数据库检测到了主键冲突，而实际上开发者的意图可能是更新现有记录而非插入新记录。

问题复现

考虑以下实体定义：

@Entity()
export class Example extends BaseEntity {
  @PrimaryColumn()
  siteId: number;

  @PrimaryColumn()
  year: number;

  @PrimaryColumn()
  month: number;

  // 其他字段...
}

当执行以下操作时可能触发问题：

const e1 = new Example();
// 设置复合主键和其他字段值...
repository.save(e1); // 未使用await

const e2 = new Example();
// 设置相同的复合主键值...
repository.save(e2); // 未使用await

问题根源

1. 异步操作未等待：TypeORM的save()方法是异步的，如果不使用await，操作会被放入事件循环队列，可能导致多个保存操作并发执行。

2. 复合主键处理机制：对于复合主键，TypeORM需要检查记录是否存在以决定是执行INSERT还是UPDATE。在并发情况下，多个操作可能同时判断记录不存在，都尝试执行INSERT。

3. 数据库约束：PostgreSQL等数据库会严格执行主键唯一性约束，当检测到重复主键时会立即抛出错误。

解决方案

1. 正确使用异步/等待

确保每次保存操作都使用await：

await repository.save(e1);
await repository.save(e2);

2. 使用事务处理

对于需要原子性保证的操作，使用事务：

await datasource.transaction(async manager => {
  await manager.save(e1);
  await manager.save(e2);
});

3. 显式指定操作类型

如果明确知道操作类型，可以使用insert或update替代save：

// 明确插入
await repository.insert(e1);
// 明确更新
await repository.update(
  { siteId: e2.siteId, year: e2.year, month: e2.month },
  e2
);

4. 配置ESLint规则

添加@typescript-eslint/no-floating-promises规则，防止忘记处理Promise：

{
  "rules": {
    "@typescript-eslint/no-floating-promises": "error"
  }
}

最佳实践建议

1. 始终处理异步操作的完成状态，使用await或.then()
2. 对于关键数据操作，使用事务保证一致性
3. 考虑在应用层添加锁机制处理高并发场景
4. 在团队中建立代码规范，确保异步操作被正确处理

通过理解TypeORM的保存机制和数据库约束行为，开发者可以避免这类并发问题，编写出更健壮的数据访问代码。