java - 如何统计Firestore中某个集合下的文档数量？

Question

我正在尝试获取CollectionReference Cloud Firestore 上存在的计数，我尝试通过以下方式获取它:

FirebaseFirestore db = FirebaseFirestore.getInstance();
    final CollectionReference postsCollection = db.collection("Posts");

    final TaskCompletionSource<Integer> source = new TaskCompletionSource<>();
    new Thread(new Runnable() {
        @Override
        public void run() {
            int fromWhereToStart = postsCollection.get().getResult().size();
            source.setResult(fromWhereToStart);
        }
    }).start();

    Task<Integer> task = source.getTask();
    task.addOnCompleteListener(new OnCompleteListener<Integer>() {
        @Override
        public void onComplete(@NonNull Task<Integer> task) {
            Log.e("Z_fromWhereToStart", "= " + task.getResult());
        }
    });

但不幸的是，我得到:

java.lang.IllegalStateException: Task is not yet complete

是否有另一种方法来获取另一种方法的计数来修复IllegalStateException？

Answer 1

编辑:2022 年 10 月 20 日

从现在开始，计算集合中的文档或查询返回的文档实际上是可能的，而无需保留计数器。因此，您可以使用新的 count() 对文档进行计数方法如下:

Returns a query that counts the documents in the result set of this query.

这项新功能是在今年的 Firebase 峰会上宣布的。请记住，此功能不会读取实际文档。所以根据official documentation :

For aggregation queries such as count(), you are charged one document read for each batch of up to 1000 index entries matched by the query. For aggregation queries that match 0 index entries, there is a minimum charge of one document read.

For example, count() operations that match between 0 and 1000 index entries are billed for one document read. For A count() operation that matches 1500 index entries, you are billed 2 document reads.

<小时/>

编辑:2021 年 7 月 10 日

最近，Firebase 添加了一个名为 Distributed Counter 的新扩展:

Use this extension to add a highly scalable counter service to your app. This is ideal for applications that count viral actions or any very high-velocity action such as views, likes, or shares.

使用此扩展，您还可以突破每秒一次写入操作的最大限制。

这里还有一篇您可能感兴趣的文章:

• How to count the number of documents in a Firestore collection?

<小时/>

由于 Firebase Realtime 数据库中没有 getDocumentCount() 方法，因此需要使用 getChildrenCount() 方法来实际计算 Cloud Firestore 中 Posts 集合下所有文档的数量，因此请使用以下代码:

db.collection("Posts").get().addOnCompleteListener(new OnCompleteListener<QuerySnapshot>() {
    @Override
    public void onComplete(@NonNull Task<QuerySnapshot> task) {
        if (task.isSuccessful()) {
            int count = 0;
            for (DocumentSnapshot document : task.getResult()) {
                count++;
            }
            Log.d("TAG", count + "");
        } else {
            Log.d(TAG, "Error getting documents: ", task.getException());
        }
    }
});

或

db.collection("Posts").get().addOnCompleteListener(new OnCompleteListener<QuerySnapshot>() {
    @Override
    public void onComplete(@NonNull Task<QuerySnapshot> task) {
        if (task.isSuccessful()) {
            Log.d("TAG", task.getResult().size() + "");
        } else {
            Log.d(TAG, "Error getting documents: ", task.getException());
        }
    }
});

上述示例对于小型数据集来说足够有效，但如果数据集较大，则不起作用。但是，还有两种方法可以实现同样的目的。

一种方法是使用 Cloud Functions每次从 Posts 集合中添加或删除文档时更新计数器。该技术也适用于大数据集。但需要注意的是，这种情况下，文档的增删改查只能以小于等于1次/秒的速率进行，如Cloud Firestore Quotas and Limits中所述。 。这是一个需要阅读的文档，但它几乎可以立即显示当前计数。

如果您需要超出此限制，则需要实现分布式计数器，如 official documentation of distributed counters 中所述。 .

As a personal hint, don't store this kind of counter in Cloud Firestore, because every time you increase or decrease the counter will cost you a read or a write operation. Host this counter in the Firebase Realtime database almost at no cost.

第二种方法是，不使用云功能，而是在客户端使用事务，在添加或删除文档的同时更新计数器。这样，你的计数器也会很准确，因为它是同时更新的。但在这种情况下，最重要的是您需要确保在添加或删除文档的任何位置都包含此逻辑。在这种情况下，您也可以免费使用 Firebase Realtime 数据库。

总之，第一个代码用于小型数据集，第二个代码使用 Cloud Functions，因为这是写入时尽力而为，第三个代码使用我上面向您解释的最后一个选项。