Next.js中MongoDB自增用户ID重复问题的替代方案咨询

我用MongoDB存储用户数据，注册时想生成自增的用户ID（比如1、2、3……），写了下面这段代码，users是存储用户数据的集合：

// generate new user id
let uid;
const collections = await db.listCollections().toArray();
const collectionNames = collections.map(collection => collection.name);

if(collectionNames.indexOf("users") == -1){
    uid = 1;
}
else{
    const newest_user = await db.collection("users").find({}).sort({"_id":-1}).limit(1).toArray();
    uid = newest_user[0]["_id"] + 1;
}
user._id = uid;


// add and save user
db.collection("users").insertOne(user).catch((error)=>{
    throw error;
});

现在担心两个用户同时注册时，会拿到相同的最大ID，生成重复的新ID。本来想用线程锁，但Node.js和Next.js不支持多线程，想问问有什么替代方案。另外，我打算把_id作为用户ID字段，想知道会不会有问题（已经知道_id不可重复）。

一、解决并发生成重复ID的方案 #

1. 使用MongoDB原子操作（推荐） #

创建一个专门的counters集合存储自增序列，利用findOneAndUpdate的原子性来获取并递增ID，从根源避免并发冲突：

// 初始化计数器（首次运行或集合不存在时执行）
await db.collection("counters").updateOne(
  { _id: "userid" },
  { $setOnInsert: { sequence_value: 0 } },
  { upsert: true }
);

// 原子性获取自增ID
const counter = await db.collection("counters").findOneAndUpdate(
  { _id: "userid" },
  { $inc: { sequence_value: 1 } },
  { returnDocument: "after" }
);

user._id = counter.value.sequence_value;
await db.collection("users").insertOne(user);

findOneAndUpdate是原子操作，多个并发请求会被MongoDB依次处理，每个请求都能拿到唯一的递增ID，完全解决重复问题。

2. 唯一约束+重试机制 #

保留原有的ID生成逻辑，利用MongoDB_id的唯一约束，当插入时触发重复键错误就重试生成ID：

async function generateAndSaveUser(user) {
  let uid;
  const collections = await db.listCollections().toArray();
  const collectionNames = collections.map(c => c.name);

  if (collectionNames.indexOf("users") === -1) {
    uid = 1;
  } else {
    const newestUser = await db.collection("users").find({}).sort({ _id: -1 }).limit(1).toArray();
    uid = newestUser[0]._id + 1;
  }
  user._id = uid;

  try {
    await db.collection("users").insertOne(user);
    return user;
  } catch (error) {
    // 捕获重复键错误（错误码11000）
    if (error.code === 11000) {
      // 重试一次
      return generateAndSaveUser(user);
    }
    throw error;
  }
}

这个方案实现简单，但高并发下可能出现多次重试，适合并发量较低的场景。

3. 应用层全局锁（单进程场景） #

利用Node.js单线程事件循环的特性，用全局变量做锁标记，让ID生成请求排队处理：

let isGeneratingId = false;
const idQueue = [];

async function getNextUserId() {
  return new Promise((resolve) => {
    // 当前正在生成ID，加入队列等待
    if (isGeneratingId) {
      idQueue.push(resolve);
      return;
    }

    isGeneratingId = true;
    (async () => {
      let uid;
      const collections = await db.listCollections().toArray();
      const collectionNames = collections.map(c => c.name);

      if (collectionNames.indexOf("users") === -1) {
        uid = 1;
      } else {
        const newestUser = await db.collection("users").find({}).sort({ _id: -1 }).limit(1).toArray();
        uid = newestUser[0]._id + 1;
      }

      isGeneratingId = false;
      resolve(uid);

      // 处理队列中的等待请求
      while (idQueue.length > 0) {
        const nextResolve = idQueue.shift();
        uid += 1;
        nextResolve(uid);
      }
    })();
  });
}

// 使用示例
const uid = await getNextUserId();
user._id = uid;
await db.collection("users").insertOne(user);

注意这个方案只适用于单进程部署，如果你用pm2启动多个Next.js进程，锁会失效。

二、关于用_id作为用户ID的问题 #

完全可以把_id作为用户ID字段，没有问题：

• MongoDB默认给_id字段加了唯一索引，天然保证唯一性，正好符合用户ID的要求；
• 自增数字ID比MongoDB默认的ObjectID更直观，方便用户记忆和业务场景使用；
• 注意不要尝试修改文档的_id值，MongoDB不允许更新_id字段。

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