如何在Firestore原子事务中用getCountFromServer实现条件写入？

核心结论 #

你当前的代码无法保证原子性。原因是Firestore事务的原子性依赖于对具体文档的读取跟踪，而getCountFromServer是聚合查询，它不会让事务记录对集合计数的依赖关系。即使在事务执行期间，目标集合的计数发生了变化，事务依然会提交，不会触发重试或失败，也就无法保证"计数未变才执行写入"的逻辑。

替代方案 #

方案1：维护独立的计数文档（推荐） #

为每个店铺创建一个专门的统计文档，存储对应的销售数量。事务中读取这个文档并跟踪其变化，这样就能利用Firestore的事务机制保证原子性。

const storeStatsRef = doc(db, 'storeStats', storeId); // 每个店铺对应唯一的统计文档

await runTransaction(db, async (transaction) => {
  // 事务内读取计数文档
  const statsDoc = await transaction.get(storeStatsRef);
  const currentCount = statsDoc.exists() ? statsDoc.data().salesCount : 0;

  // 执行你的计数检查逻辑
  if (currentCount < 100) { // 示例：限制销售数量不超过100
    // 写入新的销售记录
    const newSaleRef = doc(collection(db, 'sales'));
    transaction.set(newSaleRef, { 
      storeId,
      // 其他销售字段，比如createTime: serverTimestamp()
    });

    // 更新计数文档
    transaction.set(storeStatsRef, { salesCount: currentCount + 1 }, { merge: true });
  } else {
    throw new Error('已达到当前店铺的销售数量上限');
  }
});

这种方式的优势是高效，无论集合多大，事务只需要读写两个文档（统计文档+新销售文档），完全符合Firestore的事务设计模型。

方案2：事务内读取所有符合条件的文档（仅适用于小集合） #

如果必须依赖集合的实际文档数量（担心统计文档和实际数据不一致），可以在事务内直接读取所有符合条件的文档，通过snapshot.size获取计数。但这种方法只适合文档数量极少的场景，因为Firestore事务对读取的文档数量有限制（最多500个），且大量文档读取会导致事务延迟高、重试概率大。

const collectionReference = collection(db, 'sales');
const q = query(collectionReference, where("storeId", "===", storeId));

await runTransaction(db, async (transaction) => {
  // 事务内读取所有目标文档
  const snapshot = await transaction.get(q);
  const currentCount = snapshot.size;

  // 计数检查逻辑
  if (currentCount < 100) {
    // 写入新销售记录
    const newSaleRef = doc(collection(db, 'sales'));
    transaction.set(newSaleRef, { storeId /* 其他字段 */ });
  } else {
    throw new Error('已达到销售数量上限');
  }
});

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