MikroORM中OneToMany关系更新的陷阱与解决方案

问题背景

在使用MikroORM进行数据库操作时，开发者可能会遇到一个关于OneToMany关系更新的意外行为。当尝试更新一个未预先加载(未populate)的OneToMany关系时，系统会错误地删除所有其他相关实体的关联关系。

问题现象

考虑以下两个实体模型：

@Entity({ tableName: 'users' })
export class User {
    @PrimaryKey()
    id!: number;
    
    @ManyToOne({ entity: () => Role, nullable: true })
    role!: Rel<Role>;
}

@Entity({ tableName: 'roles' })
export class Role {
    @PrimaryKey()
    id!: number;
    @OneToMany(() => User, user => user.role)
    users = new Collection<User>(this);
}

当执行以下操作时：

const role = await em.findOne(Role, roleId);
const users = await em.find(User, { id: { $in: [2]} });
role.users.set(users);

MikroORM会生成两条SQL语句：

update `users` set `role_id` = 11 where `id` = 2
update `users` set `role_id` = NULL where (`id`) not in ((2))

第二条SQL语句会错误地将所有不属于当前集合的用户角色关系设置为NULL，这显然不是开发者预期的行为。

问题根源

这个问题的根本原因在于MikroORM的工作机制。当OneToMany关系未被预先加载(populate)时，ORM无法知道该集合中已经存在哪些元素。因此，当开发者调用set()方法设置新的集合时，ORM会认为开发者想要完全替换原有集合，从而生成删除其他关系的SQL语句。

解决方案

方案一：预先加载关系

最直接的解决方案是在查询角色时预先加载users集合：

const role = await em.findOne(Role, roleId, { 
   populate: ['users']
});

这样ORM就能知道当前集合中已经存在的元素，在更新时只会处理新增和删除的元素，而不会错误地清除所有其他关系。

方案二：使用其他更新方法

除了使用set()方法外，开发者还可以考虑使用其他方法来更新关系：

1. 直接修改用户实体：

const users = await em.find(User, { id: { $in: [2]} });
users.forEach(user => user.role = role);

2. 使用集合的add()方法：

role.users.add(users);

3. 使用原生查询：

await em.nativeUpdate(User, { id: 2 }, { role: role });

最佳实践建议

1. 始终考虑关系状态：在操作关系集合前，明确是否需要预先加载现有关系。

2. 选择合适的方法：根据业务场景选择最适合的关系更新方法，set()、add()或直接修改实体属性各有适用场景。

3. 事务处理：对于关键的关系更新操作，建议使用事务来保证数据一致性。

4. 测试验证：编写单元测试验证关系更新的行为是否符合预期。

总结

MikroORM的这一行为虽然可能让开发者感到意外，但从ORM的设计角度来看是合理的。理解这一机制有助于开发者编写更健壮的代码。关键在于记住：在操作未加载的关系集合时，ORM无法知道集合的当前状态，因此会采取最保守的策略。通过预先加载关系或选择适当的更新方法，可以避免这种意外行为。