如何查询Cloud Firestore指定集合中的文档数量

当然可以查询指定集合里的文档数量，这里有两种常用的实现方式，各自适用不同场景，你可以按需选择：

方法一：直接查询所有文档并计数（适合小集合） #

这种方法最简单，但要注意它会读取集合里的所有文档，如果集合很大，不仅会产生较高的读取费用，还会有性能延迟。

以JavaScript为例，实现代码如下：

import { collection, getDocs } from "firebase/firestore";
import { db } from "./your-firebase-config";

async function getCollectionDocCount(collectionName) {
  const colRef = collection(db, collectionName);
  const snapshot = await getDocs(colRef);
  return snapshot.size;
}

// 使用示例
getCollectionDocCount("users").then(count => {
  console.log(`集合中的文档数量：${count}`);
});

注意事项 #

• 当集合文档数超过1万时，这种方法的效率会明显下降，而且Firestore的单次查询有1万条文档的限制，超过的话需要分页处理。
• 每次调用都会消耗与文档数量相等的读取配额，成本较高。

方法二：维护一个计数器文档（推荐用于大集合） #

这种方法是在Firestore中单独维护一个存储计数的文档，每次添加或删除集合内的文档时，同步更新这个计数器的值。这样查询数量时只需要读取这一个文档，高效又省钱。

实现步骤 #

1. 创建一个专门存储计数器的集合（比如叫counters），里面建一个文档（比如叫usersCount），初始字段count设为0。

2. 添加文档时更新计数器（用事务保证并发安全）：

import { collection, addDoc, runTransaction, doc } from "firebase/firestore";
import { db } from "./your-firebase-config";

async function addUserAndUpdateCounter(userData) {
  try {
    await runTransaction(db, async (transaction) => {
      // 添加新用户文档
      const userRef = collection(db, "users");
      await addDoc(userRef, userData);

      // 更新计数器
      const counterRef = doc(db, "counters", "usersCount");
      const counterDoc = await transaction.get(counterRef);
      if (!counterDoc.exists()) {
        transaction.set(counterRef, { count: 1 });
      } else {
        transaction.update(counterRef, { count: counterDoc.data().count + 1 });
      }
    });
    console.log("用户添加成功，计数器已更新");
  } catch (e) {
    console.error("事务失败：", e);
  }
}

3. 删除文档时同样用事务更新计数器：

async function deleteUserAndUpdateCounter(userId) {
  try {
    await runTransaction(db, async (transaction) => {
      // 删除用户文档
      const userRef = doc(db, "users", userId);
      await transaction.delete(userRef);

      // 更新计数器
      const counterRef = doc(db, "counters", "usersCount");
      const counterDoc = await transaction.get(counterRef);
      if (counterDoc.exists() && counterDoc.data().count > 0) {
        transaction.update(counterRef, { count: counterDoc.data().count - 1 });
      }
    });
    console.log("用户删除成功，计数器已更新");
  } catch (e) {
    console.error("事务失败：", e);
  }
}

4. 查询计数器获取文档数量：

import { doc, getDoc } from "firebase/firestore";
import { db } from "./your-firebase-config";

async function getUserCount() {
  const counterRef = doc(db, "counters", "usersCount");
  const counterDoc = await getDoc(counterRef);
  if (counterDoc.exists()) {
    return counterDoc.data().count;
  } else {
    return 0;
  }
}

// 使用示例
getUserCount().then(count => {
  console.log(`用户总数：${count}`);
});

优势 #

• 只需要读取1个文档就能获取数量，性能极佳，成本极低。
• 用事务处理可以避免并发操作导致的计数错误。

补充说明 #

如果你的集合已经有大量历史文档，可以先一次性用方法一统计出初始数量，再设置到计数器文档中，之后再用事务维护更新。

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