如何在Firebase中获取嵌套子集合的文档计数？

Firebase本身不支持跨所有嵌套子集合的直接计数查询，所以要统计items/{itemid}/type下所有文档的总数，得换个思路实现，下面给你两种可行方案：

方案一：遍历所有父文档+子集合计数 #

这种是最直接的方式，先拿到所有items下的文档，再逐个查询对应的type子集合大小，最后累加总数：

实现计数流函数 #

Stream<int> getTotalTypeDocsCount() async* {
  // 先获取items集合的所有文档
  final itemsQuerySnapshot = await FirebaseFirestore.instance.collection('items').get();
  int total = 0;

  // 遍历每个item文档，查询对应的type子集合
  for (final itemDoc in itemsQuerySnapshot.docs) {
    final typeCollection = itemDoc.reference.collection('type');
    final typeSnapshot = await typeCollection.get();
    total += typeSnapshot.size;
  }

  // 返回最终总数
  yield total;
}

在UI中用StreamBuilder展示 #

StreamBuilder<int>(
  stream: getTotalTypeDocsCount(),
  builder: (context, snapshot) {
    if (snapshot.connectionState == ConnectionState.waiting) {
      return const Text('加载中...');
    }
    if (snapshot.hasError) {
      return const Text('统计失败');
    }
    return Text('所有type子集合文档总数：${snapshot.data ?? 0}');
  },
)

注意：如果items或type子集合的文档量很大，这种方式会有明显的性能问题，因为要发起多次查询，耗时较长。

方案二：维护父文档计数（推荐） #

更高效的方式是在items的每个文档里维护一个typeCount字段，每次新增/删除type子集合的文档时，同步更新这个计数，之后只需要累加所有items文档的typeCount就能得到总数。

新增type文档时同步更新计数 #

// 假设要给某个item新增type文档
final itemDocRef = FirebaseFirestore.instance.collection('items').doc('目标itemid');
final newTypeDocRef = itemDocRef.collection('type').doc();

// 用事务保证计数更新和文档新增的原子性
await FirebaseFirestore.instance.runTransaction((transaction) async {
  final itemSnapshot = await transaction.get(itemDocRef);
  // 拿到当前的type计数，默认0
  int currentCount = itemSnapshot.data()?['typeCount'] ?? 0;
  // 更新计数+1
  transaction.update(itemDocRef, {'typeCount': currentCount + 1});
  // 新增type文档
  transaction.set(newTypeDocRef, {
    // 你的文档字段
    'name': '示例类型',
    'createTime': FieldValue.serverTimestamp()
  });
});

删除type文档时同步更新计数 #

// 假设要删除某个type文档
final typeDocRef = FirebaseFirestore.instance.collection('items').doc('目标itemid').collection('type').doc('要删除的typeid');
final itemDocRef = typeDocRef.parent.parent!; // 拿到对应的item文档引用

await FirebaseFirestore.instance.runTransaction((transaction) async {
  final itemSnapshot = await transaction.get(itemDocRef);
  int currentCount = itemSnapshot.data()?['typeCount'] ?? 0;
  // 计数-1，不能小于0
  transaction.update(itemDocRef, {'typeCount': max(currentCount - 1, 0)});
  // 删除type文档
  transaction.delete(typeDocRef);
});

快速统计总数 #

Stream<int> getTotalTypeDocsCount() async* {
  final itemsSnapshot = await FirebaseFirestore.instance.collection('items').get();
  int total = 0;
  for (final doc in itemsSnapshot.docs) {
    total += doc.data()?['typeCount'] ?? 0;
  }
  yield total;
}

这种方案只需要一次查询所有items文档，性能比方案一好很多，适合数据量较大的场景。

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