如何在Firestore事务中使用getCountFromServer（前端实现）

方案一：使用原子批量写入 + FieldValue.increment（推荐） #

这是最简洁高效的方案，完全满足你的需求：原子性操作、仅按写入计费、无并发不一致问题，无需依赖事务或getCountFromServer。

核心思路是把「添加待办文档」和「更新用户计数」放在同一个批量写入中，利用Firebase的FieldValue.increment()实现服务器端原子计数更新，确保多并发场景下计数准确。

import { getFirestore, writeBatch, doc, collection, FieldValue } from "firebase/firestore";

const db = getFirestore();

async function addTodoAndUpdateCount(userId, todoData) {
  const batch = writeBatch(db);
  
  // 1. 创建新待办文档的引用并加入批量操作
  const newTodoDoc = doc(collection(db, "todos"));
  batch.set(newTodoDoc, { ...todoData, userId });
  
  // 2. 更新用户文档中的待办计数字段
  const userDoc = doc(db, "users", userId);
  batch.update(userDoc, {
    todoCount: FieldValue.increment(1)
  });
  
  // 提交批量操作，原子执行（要么全成功，要么全失败）
  await batch.commit();
}

优势 #

• 原子性：批量写入是原子操作，避免添加待办成功但计数更新失败的情况
• 无并发问题：increment是服务器端原子操作，多用户同时操作也不会出现计数偏差
• 计费友好：仅产生2次写入费用（添加待办+更新计数），无额外读取成本
• 实现简单：无需复杂的事务逻辑，前端直接执行即可

方案二：基于实际集合计数的事务实现（特殊需求场景） #

如果你的业务必须依赖集合的实际文档数量来更新计数（比如担心历史计数存在偏差），可以通过「锁文档+事务+两次getCountFromServer」的方式实现，但流程相对复杂，且存在锁竞争风险。

核心思路是用一个临时锁文档确保同一时间只有一个操作能更新计数，在事务内再次获取集合计数验证一致性，避免并发操作导致的计数错误。

import { getFirestore, runTransaction, doc, collection, getCountFromServer, deleteDoc } from "firebase/firestore";

const db = getFirestore();

async function addTodoAndUpdateCountWithActualCount(userId, todoData) {
  const lockDocRef = doc(db, "locks", `user_todo_lock_${userId}`);
  
  try {
    await runTransaction(db, async (transaction) => {
      // 检查锁是否存在，防止并发操作
      const lockSnapshot = await transaction.get(lockDocRef);
      if (lockSnapshot.exists()) {
        throw new Error("当前有其他操作正在进行，请稍后重试");
      }
      
      // 创建锁文档
      transaction.set(lockDocRef, { locked: true });
      
      // 获取当前用户的待办文档实际数量
      const todosCol = collection(db, "todos");
      const countSnapshot = await getCountFromServer(todosCol, {
        where: ["userId", "==", userId]
      });
      const currentCount = countSnapshot.data().count;
      
      // 更新用户计数
      const userDoc = doc(db, "users", userId);
      transaction.update(userDoc, { todoCount: currentCount + 1 });
      
      // 添加新待办文档
      const newTodoDoc = doc(todosCol);
      transaction.set(newTodoDoc, { ...todoData, userId });
      
      // 事务提交前删除锁
      transaction.delete(lockDocRef);
    });
  } catch (error) {
    console.error("操作失败：", error);
    // 事务失败后清理锁，防止死锁
    await deleteDoc(lockDocRef).catch(() => {});
    throw error;
  }
}

注意事项 #

• 锁文档会额外产生写入/删除费用
• 高并发场景下可能出现操作失败（锁竞争），需要前端处理重试逻辑
• 仅适用于必须保证计数与实际文档数量完全一致的特殊场景，大部分情况下方案一足够满足需求

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