Update firestore/app/src/main/java/com/google/example/firestore/DocSnippets.java

Co-authored-by: Rosário Pereira Fernandes <[email protected]>

Author: morganchen12

firestore/app/src/main/java/com/google/example/firestore/DocSnippets.java

@@ -1366,8 +1366,18 @@ public void countAggregateQuery() {
         // [START count_aggregate_query]
         Query query = db.collection("games/chess/players").whereEqualTo("online", true);
         AggregateQuery countQuery = query.count();
-        AggregateQuerySnapshot snapshot = countQuery.get(AggregateSource.SERVER).getResult();
-        Log.d(TAG, "Count: " + snapshot.getCount());
+        countQuery.get(AggregateSource.SERVER).addOnCompleteListener(new OnCompleteListener<AggregateQuerySnapshot>() {
+    @Override
+    public void onComplete(@NonNull Task<AggregateQuerySnapshot> task) {
+        if (task.isSuccessful()) {
+            // Count fetched successfully
+            AggregateQuerySnapshot snapshot = task.getResult();
+            Log.d(TAG, "Count: " + snapshot.getCount());
+        } else {
+            Log.d(TAG, "Count failed: ", task.getException());
+        }
+    }
+});;
         // [END count_aggregate_query]
     }
 }