MongoDB实现productCode唯一自增（如p01、p02）的方案咨询及问题排查

我来帮你搞定这个自增唯一编码的问题～你之前用countDocuments的方案确实存在两个致命问题：并发场景下会生成重复编码，以及如果有产品被删除，计数回退会导致编码重复。想要实现类似MongoDB ObjectId那种自动、唯一的生成逻辑，最靠谱的方式是用原子性的计数器集合来维护序号，下面是具体的实现步骤和代码：

为什么原方案不行？ #

你的代码里'p'+productCount+1其实写法有误（应该是'p' + (productCount + 1)），但就算修正了语法，本质问题依然存在：

• countDocuments是非原子操作，当多个请求同时调用时，会拿到相同的计数，生成重复的productCode
• 如果有产品文档被删除，countDocuments的结果会减少，新生成的编码会和已删除的重复，彻底破坏唯一性

最优实现：计数器集合+原子操作 #

MongoDB本身没有内置的自增字段，但我们可以自己维护一个专门的计数器集合，用findOneAndUpdate的原子操作来保证每次获取的序号都是唯一且递增的，完全避免并发冲突。

1. 初始化计数器集合 #

首先我们需要一个counters集合，用来存储不同序列的当前值。针对产品编码，我们可以初始化一条文档（也可以让后续操作自动创建）：

// 项目启动时执行一次即可
await db.collection('counters').updateOne(
  { _id: 'productCode' }, // 标识这个计数器是给产品编码用的
  { $setOnInsert: { seq: 0 } }, // 如果不存在就初始化为0
  { upsert: true } // 不存在则插入，存在则不做操作
);

2. 编写获取下一个序号的函数 #

这个函数用原子操作findOneAndUpdate来递增序号并返回新值，确保多请求下不会冲突：

async function getNextProductSeq() {
  // 原子性地递增序号并返回更新后的值
  const counter = await db.collection('counters').findOneAndUpdate(
    { _id: 'productCode' },
    { $inc: { seq: 1 } }, // 每次调用序号+1
    { 
      returnDocument: 'after', // 返回更新后的文档
      upsert: true // 防止计数器不存在时自动创建
    }
  );
  return counter.value.seq;
}

3. 生成格式化的productCode并创建产品 #

现在在创建新产品时，先调用上面的函数拿到唯一序号，再格式化成你需要的p01、p02样式：

// 获取唯一递增序号
const seq = await getNextProductSeq();
// 格式化为带前缀+补零的编码，padStart(2, '0')表示补两位零，可根据需求调整长度
const productCode = `p${seq.toString().padStart(2, '0')}`;

// 创建新产品文档
const newProduct = new Product({ 
  productCode: productCode, 
  name: req.body.name, 
  description: req.body.description 
});
await newProduct.save();

方案优势 #

• 绝对唯一：原子操作保证了即使高并发下，每个请求都会拿到唯一的序号，不会生成重复编码
• 高效可靠：计数器集合只操作单个文档，比countDocuments扫描全集合快得多
• 不依赖文档数量：就算有产品被删除，序号依然持续递增，不会出现编码重复的问题
• 扩展性强：如果后续需要不同类型的产品编码（比如v01表示VIP产品），只需要在counters集合里新增一个_id: 'vipProductCode'的计数器即可

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