如何通过API在MongoDB插入互相关联的用户与公司文档？

要实现user和company文档互相引用对方ID的插入需求，核心是保证两个文档的关联ID一致，且操作要么全部成功要么全部失败，避免数据不一致。以下是两种可行方案：

方案一：客户端预生成ID + 事务插入（推荐） #

MongoDB的ObjectId可以在客户端直接生成，无需依赖数据库插入后返回。结合事务能确保两个插入操作的原子性，步骤如下：

1. 预生成两个ObjectId，分别作为user和company的ID
2. 构造两个文档，直接填写对方的ID
3. 在事务中执行两个集合的插入操作

代码示例（Node.js 原生MongoDB驱动） #

const { MongoClient, ObjectId } = require('mongodb');

async function createUserAndCompany() {
  const client = new MongoClient('your-mongodb-connection-string');
  
  try {
    await client.connect();
    const db = client.db('your-db-name');
    const usersCollection = db.collection('user');
    const companiesCollection = db.collection('company');

    // 预生成两个ObjectId
    const userId = new ObjectId();
    const companyId = new ObjectId();

    // 构造文档
    const userDoc = {
      _id: userId,
      companyId: companyId
    };
    const companyDoc = {
      _id: companyId,
      registeredByID: userId
    };

    // 启动事务
    const session = client.startSession();
    session.startTransaction();

    try {
      // 在事务中插入两个文档
      await usersCollection.insertOne(userDoc, { session });
      await companiesCollection.insertOne(companyDoc, { session });

      // 提交事务
      await session.commitTransaction();
      console.log('用户和公司文档插入成功');
    } catch (err) {
      // 事务失败，回滚
      await session.abortTransaction();
      throw err;
    } finally {
      session.endSession();
    }
  } catch (err) {
    console.error('插入失败:', err);
  } finally {
    await client.close();
  }
}

createUserAndCompany();

注意：该方案要求MongoDB版本≥4.0，且部署为副本集或分片集群（单节点MongoDB不支持事务）。

方案二：分步插入+更新+异常回滚（兼容单节点） #

如果你的MongoDB是单节点无法使用事务，可以采用分步操作+手动回滚的方式，步骤如下：

1. 先插入user文档（companyId暂时设为null或留空）
2. 用返回的user ID插入company文档
3. 用返回的company ID更新user文档的companyId字段
4. 若任意步骤失败，手动删除已插入的文档

代码示例（Node.js 原生MongoDB驱动） #

const { MongoClient } = require('mongodb');

async function createUserAndCompany() {
  const client = new MongoClient('your-mongodb-connection-string');
  let insertedUserId = null;
  let insertedCompanyId = null;

  try {
    await client.connect();
    const db = client.db('your-db-name');
    const usersCollection = db.collection('user');
    const companiesCollection = db.collection('company');

    // 1. 插入user文档
    const userResult = await usersCollection.insertOne({ companyId: null });
    insertedUserId = userResult.insertedId;

    // 2. 插入company文档，关联user ID
    const companyResult = await companiesCollection.insertOne({ registeredByID: insertedUserId });
    insertedCompanyId = companyResult.insertedId;

    // 3. 更新user文档，关联company ID
    await usersCollection.updateOne(
      { _id: insertedUserId },
      { $set: { companyId: insertedCompanyId } }
    );

    console.log('用户和公司文档插入成功');
  } catch (err) {
    console.error('插入失败，开始回滚:', err);
    // 回滚：删除已插入的文档
    if (insertedCompanyId) {
      await companiesCollection.deleteOne({ _id: insertedCompanyId });
    }
    if (insertedUserId) {
      await usersCollection.deleteOne({ _id: insertedUserId });
    }
  } finally {
    await client.close();
  }
}

createUserAndCompany();

注意：该方案需要手动处理回滚逻辑，若中间步骤（如更新user）失败，必须确保已插入的company被删除，否则会产生无效数据。

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