如何在交易期间从 MongoDB 创建操作中检索 ID?

How can I retrieve an id from MongoDB create operation during a transaction?

我正在尝试使用 Apollo Server 和 Mongoose 创建审计跟踪。当用户最初注册时,我会在用户集合中创建一个文档,并在历史集合中为他们提供的每条数据(用户名、密码、电子邮件等)创建一个文档。对于每个历史集合文档,我都包含用户文档的 ID 以创建关系。完美运行。

但是,当我在(见下文)中添加交易时,用户文档的 userId 返回为未定义,因此我无法将其添加到历史条目文档中。我假设在整个交易完成之前不会创建文档的 ID?

有什么想法吗?

Mutation: {
  register: async (_, { data }) => {

    // Start a mongo session & transaction
    const session = await mongoose.startSession()
    session.startTransaction()

    try {

      // Hash password and create user
      const hashedPassword = await bcrypt.hash(data.password, 12)
      const user = await User.create(
        [{ ...data, password: hashedPassword }],
        { session }
      )
      
      // Add history entries
      HistoryEntry.create([
      {
        user: user.id,
        action: 'registered'
      },
      {
        user: user.id,
        action: 'set',
        object: 'profile',
        instance: user.id,
        property: 'firstName',
        value: firstName
      },
      {
        user: user.id,
        action: 'set',
        object: 'profile',
        instance: user.id,
        property: 'lastName',
        value: lastName
      },
      {
        user: user.id,
        action: 'set',
        object: 'profile',
        instance: user.id,
        property: 'password'
      }
    ])

    if (loginType === 'email') {
      HistoryEntry.create({
        user: user.id,
        action: 'set',
        object: 'profile',
        instance: user.id,
        property: 'email',
        value: login
      })
    }

    if (loginType === 'mobile') {
      HistoryEntry.create({
        user: user.id,
        action: 'set',
        object: 'profile',
        instance: user.id,
        property: 'mobile',
        value: login
      })
    }

    // commit the changes if everything was successful
    await session.commitTransaction()
      return {
        ok: true,
        user
      }
    } catch (err) {
      // if anything fails above, rollback the changes in the transaction
      await session.abortTransaction()
      return formatErrors(err)
    } finally {
      // end the session
      session.endSession()
    }
  }
}

想一想,你还没有添加User,怎么添加一个HistoryEntry呢?这不是您目前正在做的 'history'。我相信您在这里有两个选择 - 手动在用户上设置 _id new Schema({ _id: { type: Schema.ObjectId, auto: true }}) 然后在事务中生成它:var userId = ObjectId(); 并用于用户和历史条目。

我相信第二个选项在这种情况下在语义上更正确 - 你应该附加到 post-save hook:

schema.post('save', function(doc) {
  console.log('%s has been saved', doc._id);
});

因此,无论何时创建用户,都会触发 post-save 挂钩以更新历史记录。

最近遇到了同样的问题,希望你已经解决了。我可以为未来的求职者添加这个。

  1. 以下创建函数 returns 创建的文档数组。
const user = await User.create(
        [{ ...data, password: hashedPassword }],
        { session }
);

因此访问用户 ID 为 user[0]._id

  1. 也将会话传递给 HistoryEntry.create()
HistoryEntry.create([{...},{...}], {session})

注意:在这个用例中,我个人更喜欢@marek 第二个选项来使用 post-save hook。