如何在NodeJS中实现用户注册时生成递增的4位及以上随机ID？

核心思路 #

• 初始ID从1000开始（满足至少4位的最小要求）
• 用独立的计数器集合维护ID序列，通过MongoDB的原子操作findOneAndUpdate避免并发注册时的ID冲突
• 利用Mongoose预保存中间件，在用户保存前自动生成递增ID

代码实现 #

1. 创建计数器模型 #

// models/counter.model.js
const mongoose = require('mongoose');

const counterSchema = new mongoose.Schema({
  _id: { type: String, required: true },
  seq: { type: Number, default: 1000 }
});

// 封装获取下一个序列的静态方法
counterSchema.statics.getNextSeq = async function(counterName) {
  const counter = await this.findOneAndUpdate(
    { _id: counterName },
    { $inc: { seq: 1 } },
    { new: true, upsert: true } // 不存在则创建，返回更新后的数据
  );
  return counter.seq;
};

module.exports = mongoose.model('Counter', counterSchema);

2. 修改用户Schema #

// models/user.model.js
const mongoose = require('mongoose');
const Counter = require('./counter.model');

const userSchema = new mongoose.Schema({
  customId: { type: Number, unique: true, required: true }, // 新增的递增ID字段
  // 保留你原有的用户字段，示例如下
  username: { type: String, required: true },
  password: { type: String, required: true },
  email: { type: String }
});

// 预保存中间件：仅在新增用户时生成customId
userSchema.pre('save', async function(next) {
  if (!this.isNew) return next();

  try {
    this.customId = await Counter.getNextSeq('userCustomId');
    next();
  } catch (err) {
    next(err);
  }
});

module.exports = mongoose.model('User', userSchema);

3. 更新注册控制器 #

// controllers/user.controller.js
const User = require('../models/user.model');

exports.register = async (req, res) => {
  try {
    const { username, password, email } = req.body;
    const newUser = new User({ username, password, email });
    await newUser.save();

    res.status(201).json({
      message: '注册成功',
      userInfo: {
        customId: newUser.customId,
        username: newUser.username,
        email: newUser.email
      }
    });
  } catch (err) {
    // 处理唯一索引冲突（极端并发场景下的兜底）
    if (err.code === 11000) {
      return res.status(400).json({ message: '当前注册请求过于频繁，请稍后重试' });
    }
    res.status(500).json({ message: '注册失败', error: err.message });
  }
};

4. 路由模块示例（如果你的user.module是路由文件） #

// routes/user.module.js
const express = require('express');
const userController = require('../controllers/user.controller');
const router = express.Router();

router.post('/register', userController.register);

module.exports = router;

关键说明 #

• 计数器集合的原子操作是核心：findOneAndUpdate保证了在多请求同时触发时，每次seq的递增都是原子性的，不会出现重复ID
• 预保存中间件自动处理ID生成，业务代码无需手动干预
• 给customId添加unique索引，即使极端情况出现冲突，也能通过MongoDB的索引约束拦截，避免脏数据

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