MongoDB中基于存在性校验生成唯一短URL字符串的最佳方法

你当前的循环查重逻辑存在几个致命缺陷，根本达不到防重复的效果：

• URL.findOne()调用前没有加await，dbQuery拿到的是未resolve的Promise对象，后续判断dbQuery.fullURL永远是undefined，查重逻辑完全失效
• 无上限的while(true)循环在极端场景（比如短串生成规则冲突率飙升、数据库连接异常）会持续占用事件循环，直接拖垮服务
• 存在竞态风险：高并发下两个请求可能同时查到同一个未被占用的短码，先后写入时还是会出现重复数据
• 自定义Model命名为URL，和Node.js全局内置的URL构造函数重名，很容易引发难以排查的类型错误

核心思路是把防重复的兜底逻辑交给数据库唯一索引，不要在应用层做前置查重，既减少一次数据库查询开销，又能彻底解决并发竞态问题。

第一步：给短码字段加唯一索引 #

修改Mongoose Schema定义，从数据库层面强制短码唯一：

const mongoose = require('mongoose');

const urlSchema = new mongoose.Schema({
  fullURL: {
    type: String,
    required: true
  },
  shortURL: {
    type: String,
    required: true,
    unique: true // 数据库层面强制唯一，重复写入会直接抛出11000错误
  }
});

// 开发环境可以开autoIndex，生产环境建议手动建索引避免启动卡顿
const URLModel = mongoose.model('URL', urlSchema);

第二步：用有限重试+捕获重复错误的逻辑替代无限循环查重 #

不需要提前查库判断短码是否存在，直接尝试写入，遇到唯一键冲突错误就重新生成短码重试，设置合理重试上限即可：

// 短链生成接口逻辑
const MAX_RETRY_COUNT = 5; // 6位大小写+数字混合短码的冲突概率低于百万分之一，5次重试足够覆盖极端情况
async function createShortURL(req, res, next) {
  const originalURL = req.body.inputURL;
  let retryTimes = 0;

  while (retryTimes < MAX_RETRY_COUNT) {
    try {
      const shortCode = utils.makeShortURLString();
      const newURLRecord = await URLModel.create({
        fullURL: originalURL,
        shortURL: shortCode
      });
      // 写入成功直接返回结果
      const fullShortURL = `${req.get('host')}/${newURLRecord.shortURL}`;
      return res.render('shortenedURL', {
        shortURL: fullShortURL,
        fullURL: newURLRecord.fullURL
      });
    } catch (err) {
      // 只有短码重复的错误才触发重试
      if (err.code === 11000 && err.keyPattern?.shortURL) {
        retryTimes++;
        continue;
      }
      // 其他错误交给全局错误中间件处理
      return next(err);
    }
  }

  // 重试次数耗尽的兜底返回
  res.status(500).render('error', {msg: '短链生成繁忙，请稍后再试'});
}

• 没有前置查重查询，每次生成逻辑最多一次数据库写入，性能比原方案高近一倍
• 数据库唯一索引兜底，从根本上杜绝并发写入导致的短码重复
• 有限重试避免死循环风险，错误分层处理不会吞掉非冲突类的异常
• 修正了原代码漏加await、变量命名冲突的低级bug

可选优化：如果后续业务量级很大，可以提前批量生成一批未使用的短码存在Redis缓存中，生成短链时直接从缓存取可用短码，后台异步补全缓存池，能进一步降低数据库写入的冲突概率。

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