NestJS+Mongoose实现MongoDB一对一关系关联新用户与邀请的方法

针对你的需求，有两种实现路径，对应不同的数据库操作次数： #

方案1：沿用当前引用式Schema设计 #

你现在的设计是将邀请数据作为独立集合、通过ObjectId和用户关联，这种情况下必须执行两次数据库写请求，无法单次完成，因为两个独立集合的文档必须分别创建后再绑定关联。
具体实现代码如下：

async create(createUserDto: CreateUserDto) {
    const userExists = await this.userModel.findOne({ email: createUserDto.email });
    if (userExists) {
      this.logger.error('User already exists');
      throw new NotAcceptableException('User already exists');
    }
    // 第一步：先创建默认邀请文档
    const createdInvite = await new this.inviteModel({
      status: 'not_sent'
    }).save();
    // 第二步：把邀请ID绑定到用户字段上，再保存用户
    const createdUser = await new this.userModel({
      ...createUserDto,
      invite: createdInvite._id
    }).save();
    // 第三步：关联查询邀请数据，返回符合要求的结构
    await createdUser.populate('invite');
    return toCreateUserDto(createdUser);
}

如果需要保证数据一致性，建议添加数据库事务，避免邀请创建成功但用户创建失败导致的脏数据。 #

方案2：改为内嵌式一对一设计，单次请求完成 #

如果你的邀请数据没有独立查询、独立批量操作的需求，完全符合一对一关系内嵌的最佳实践，调整Schema后就可以单次数据库请求完成整个流程：

调整后的Schema代码 #

export type UserModelDocument = UserModel & Document;

// 邀请不需要再作为独立集合的Schema，直接作为用户Schema的子文档
@Schema()
export class InviteSubSchema {
  @Prop()
  status: string;
}

@Schema()
export class UserModel {
  @Prop()
  email: string;

  @Prop({ type: InviteSubSchema })
  invite: InviteSubSchema;
}

export const UserSchema = SchemaFactory.createForClass(UserModel);

调整后的创建逻辑 #

async create(createUserDto: CreateUserDto) {
    const userExists = await this.userModel.findOne({ email: createUserDto.email });
    if (userExists) {
      this.logger.error('User already exists');
      throw new NotAcceptableException('User already exists');
    }
    // 单次保存即可完成用户和内嵌邀请数据的写入，不需要额外操作
    const createdUser = await new this.userModel({
      ...createUserDto,
      invite: {
        status: 'not_sent'
      }
    }).save();
    return toCreateUserDto(createdUser);
}

这种方案不仅减少了数据库请求次数，查询的时候也不需要额外关联查询，性能更优。

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