在Firebase中实现短随机唯一编号替代文档ID的最优方案咨询

你的方案可行但并非最优，存在几个关键问题：

• 并发冲突风险：多个请求同时生成相同随机数时，都会通过数组检查（因为还没完成写入），最终导致重复编号被存入。
• 性能损耗：随着已用编号增多，存储编号的数组会越来越大，每次校验都要读取整个数组做比对，效率会逐步下降。
• 维护成本：单独维护一个存储已用编号的文档，后续清理、查询操作都不够灵活。

更优方案1：预生成所有可用编号并按需分配 #

因为4位数字总共只有10000种组合，完全可以提前生成所有未使用的编号，存入Firebase集合（比如availableCodes），每个文档对应一个编号，用字段标记状态（如unused/used）。

• 分配逻辑：用事务操作查询第一个unused状态的文档，将其状态改为used，同时关联到目标业务文档。
• 优势：彻底避免并发冲突，分配操作原子性强，查询效率高，无需重复生成和校验。
• 示例代码（JavaScript）：

async function assignUniqueCode() {
  const db = firebase.firestore();
  const batch = db.batch();
  
  // 查询首个未使用的编号文档
  const querySnapshot = await db.collection('availableCodes')
    .where('status', '==', 'unused')
    .limit(1)
    .get();
  
  if (querySnapshot.empty) {
    throw new Error('所有4位编号已用尽');
  }
  
  const codeDoc = querySnapshot.docs[0];
  const code = codeDoc.id; // 文档ID直接设为编号
  
  // 标记编号为已使用
  batch.update(codeDoc.ref, { status: 'used' });
  
  // 同时将编号写入目标业务文档（按需调整）
  // batch.set(db.collection('businessDocs').doc(), { code, ...otherFields });
  
  await batch.commit();
  return code;
}

更优方案2：原子事务优化原生成逻辑 #

如果不想预生成编号，可以用Firebase的事务改造你的原方案，确保检查和写入的原子性，解决并发冲突问题：

• 核心逻辑：在事务中读取已用编号数组，生成随机数后校验是否存在，不存在则将其加入数组；事务会自动处理并发冲突，失败则重试。
• 示例代码（JavaScript）：

async function generateUniqueCode() {
  const db = firebase.firestore();
  const usedCodesRef = db.collection('meta').doc('usedCodes');
  
  return db.runTransaction(async (transaction) => {
    const doc = await transaction.get(usedCodesRef);
    let usedCodes = doc.data()?.codes || [];
    
    let newCode;
    do {
      // 生成4位随机数（自动补前导0）
      newCode = Math.floor(1000 + Math.random() * 9000).toString();
    } while (usedCodes.includes(newCode));
    
    // 更新已用编号数组
    transaction.update(usedCodesRef, { codes: [...usedCodes, newCode] });
    return newCode;
  });
}

• 优势：解决了原方案的并发问题，无需预生成，但随着已用编号增多，生成随机数的循环次数可能增加，不过在1万种组合用尽前效率都可接受。

总结 #

如果业务场景允许预生成，方案1是最优选择，性能和可靠性都最高；如果不想预生成，方案2是原方案的针对性优化，解决了核心的并发冲突问题。

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