Cloud Firestore collection count

后端 未结 17 1313
庸人自扰
庸人自扰 2020-11-22 09:25

Is it possible to count how many items a collection has using the new Firebase database, Cloud Firestore?

If so, how do I do that?

相关标签:
17条回答
  • 2020-11-22 09:32

    Took me a while to get this working based on some of the answers above, so I thought I'd share it for others to use. I hope it's useful.

    'use strict';
    
    const functions = require('firebase-functions');
    const admin = require('firebase-admin');
    admin.initializeApp();
    const db = admin.firestore();
    
    exports.countDocumentsChange = functions.firestore.document('library/{categoryId}/documents/{documentId}').onWrite((change, context) => {
    
        const categoryId = context.params.categoryId;
        const categoryRef = db.collection('library').doc(categoryId)
        let FieldValue = require('firebase-admin').firestore.FieldValue;
    
        if (!change.before.exists) {
    
            // new document created : add one to count
            categoryRef.update({numberOfDocs: FieldValue.increment(1)});
            console.log("%s numberOfDocs incremented by 1", categoryId);
    
        } else if (change.before.exists && change.after.exists) {
    
            // updating existing document : Do nothing
    
        } else if (!change.after.exists) {
    
            // deleting document : subtract one from count
            categoryRef.update({numberOfDocs: FieldValue.increment(-1)});
            console.log("%s numberOfDocs decremented by 1", categoryId);
    
        }
    
        return 0;
    });
    
    0 讨论(0)
  • 2020-11-22 09:33

    As with many questions, the answer is - It depends.

    You should be very careful when handling large amounts of data on the front end. On top of making your front end feel sluggish, Firestore also charges you $0.60 per million reads you make.


    Small collection (less than 100 documents)

    Use with care - Frontend user experience may take a hit

    Handling this on the front end should be fine as long as you are not doing too much logic with this returned array.

    db.collection('...').get().then(snap => {
       size = snap.size // will return the collection size
    });
    

    Medium collection (100 to 1000 documents)

    Use with care - Firestore read invocations may cost a lot

    Handling this on the front end is not feasible as it has too much potential to slow down the users system. We should handle this logic server side and only return the size.

    The drawback to this method is you are still invoking firestore reads (equal to the size of your collection), which in the long run may end up costing you more than expected.

    Cloud Function:

    ...
    db.collection('...').get().then(snap => {
        res.status(200).send({length: snap.size});
    });
    

    Front End:

    yourHttpClient.post(yourCloudFunctionUrl).toPromise().then(snap => {
         size = snap.length // will return the collection size
    })
    

    Large collection (1000+ documents)

    Most scalable solution


    FieldValue.increment()

    As of April 2019 Firestore now allows incrementing counters, completely atomically, and without reading the data prior. This ensures we have correct counter values even when updating from multiple sources simultaneously (previously solved using transactions), while also reducing the number of database reads we perform.


    By listening to any document deletes or creates we can add to or remove from a count field that is sitting in the database.

    See the firestore docs - Distributed Counters Or have a look at Data Aggregation by Jeff Delaney. His guides are truly fantastic for anyone using AngularFire but his lessons should carry over to other frameworks as well.

    Cloud Function:

    export const documentWriteListener = 
        functions.firestore.document('collection/{documentUid}')
        .onWrite((change, context) => {
    
        if (!change.before.exists) {
            // New document Created : add one to count
    
            db.doc(docRef).update({numberOfDocs: FieldValue.increment(1)});
        
        } else if (change.before.exists && change.after.exists) {
            // Updating existing document : Do nothing
    
        } else if (!change.after.exists) {
            // Deleting document : subtract one from count
    
            db.doc(docRef).update({numberOfDocs: FieldValue.increment(-1)});
    
        }
    
    return;
    });
    

    Now on the frontend you can just query this numberOfDocs field to get the size of the collection.

    0 讨论(0)
  • 2020-11-22 09:36

    Simplest way to do so is to read the size of a "querySnapshot".

    db.collection("cities").get().then(function(querySnapshot) {      
        console.log(querySnapshot.size); 
    });
    

    You can also read the length of the docs array inside "querySnapshot".

    querySnapshot.docs.length;
    

    Or if a "querySnapshot" is empty by reading the empty value, which will return a boolean value.

    querySnapshot.empty;
    
    0 讨论(0)
  • 2020-11-22 09:36
    firebaseFirestore.collection("...").addSnapshotListener(new EventListener<QuerySnapshot>() {
            @Override
            public void onEvent(QuerySnapshot documentSnapshots, FirebaseFirestoreException e) {
    
                int Counter = documentSnapshots.size();
    
            }
        });
    
    0 讨论(0)
  • 2020-11-22 09:37

    No, there is no built-in support for aggregation queries right now. However there are a few things you could do.

    The first is documented here. You can use transactions or cloud functions to maintain aggregate information:

    This example shows how to use a function to keep track of the number of ratings in a subcollection, as well as the average rating.

    exports.aggregateRatings = firestore
      .document('restaurants/{restId}/ratings/{ratingId}')
      .onWrite(event => {
        // Get value of the newly added rating
        var ratingVal = event.data.get('rating');
    
        // Get a reference to the restaurant
        var restRef = db.collection('restaurants').document(event.params.restId);
    
        // Update aggregations in a transaction
        return db.transaction(transaction => {
          return transaction.get(restRef).then(restDoc => {
            // Compute new number of ratings
            var newNumRatings = restDoc.data('numRatings') + 1;
    
            // Compute new average rating
            var oldRatingTotal = restDoc.data('avgRating') * restDoc.data('numRatings');
            var newAvgRating = (oldRatingTotal + ratingVal) / newNumRatings;
    
            // Update restaurant info
            return transaction.update(restRef, {
              avgRating: newAvgRating,
              numRatings: newNumRatings
            });
          });
        });
    });
    

    The solution that jbb mentioned is also useful if you only want to count documents infrequently. Make sure to use the select() statement to avoid downloading all of each document (that's a lot of bandwidth when you only need a count). select() is only available in the server SDKs for now so that solution won't work in a mobile app.

    0 讨论(0)
  • 2020-11-22 09:41

    Be careful counting number of documents for large collections. It is a little bit complex with firestore database if you want to have a precalculated counter for every collection.

    Code like this doesn't work in this case:

    export const customerCounterListener = 
        functions.firestore.document('customers/{customerId}')
        .onWrite((change, context) => {
    
        // on create
        if (!change.before.exists && change.after.exists) {
            return firestore
                     .collection('metadatas')
                     .doc('customers')
                     .get()
                     .then(docSnap =>
                         docSnap.ref.set({
                             count: docSnap.data().count + 1
                         }))
        // on delete
        } else if (change.before.exists && !change.after.exists) {
            return firestore
                     .collection('metadatas')
                     .doc('customers')
                     .get()
                     .then(docSnap =>
                         docSnap.ref.set({
                             count: docSnap.data().count - 1
                         }))
        }
    
        return null;
    });
    

    The reason is because every cloud firestore trigger has to be idempotent, as firestore documentation say: https://firebase.google.com/docs/functions/firestore-events#limitations_and_guarantees

    Solution

    So, in order to prevent multiple executions of your code, you need to manage with events and transactions. This is my particular way to handle large collection counters:

    const executeOnce = (change, context, task) => {
        const eventRef = firestore.collection('events').doc(context.eventId);
    
        return firestore.runTransaction(t =>
            t
             .get(eventRef)
             .then(docSnap => (docSnap.exists ? null : task(t)))
             .then(() => t.set(eventRef, { processed: true }))
        );
    };
    
    const documentCounter = collectionName => (change, context) =>
        executeOnce(change, context, t => {
            // on create
            if (!change.before.exists && change.after.exists) {
                return t
                        .get(firestore.collection('metadatas')
                        .doc(collectionName))
                        .then(docSnap =>
                            t.set(docSnap.ref, {
                                count: ((docSnap.data() && docSnap.data().count) || 0) + 1
                            }));
            // on delete
            } else if (change.before.exists && !change.after.exists) {
                return t
                         .get(firestore.collection('metadatas')
                         .doc(collectionName))
                         .then(docSnap =>
                            t.set(docSnap.ref, {
                                count: docSnap.data().count - 1
                            }));
            }
    
            return null;
        });
    

    Use cases here:

    /**
     * Count documents in articles collection.
     */
    exports.articlesCounter = functions.firestore
        .document('articles/{id}')
        .onWrite(documentCounter('articles'));
    
    /**
     * Count documents in customers collection.
     */
    exports.customersCounter = functions.firestore
        .document('customers/{id}')
        .onWrite(documentCounter('customers'));
    

    As you can see, the key to prevent multiple execution is the property called eventId in the context object. If the function has been handled many times for the same event, the event id will be the same in all cases. Unfortunately, you must have "events" collection in your database.

    0 讨论(0)
提交回复
热议问题