add limited and count

Author: jonhealy1

stac_fastapi/mongo/stac_fastapi/mongo/core.py

@@ -105,10 +105,12 @@ def item_collection(
         with self.client.start_session() as session:
             collection_children = self.item_table.find(
                 {"collection": collection_id}, session=session
-            ).sort(
+            ).limit(limit).sort(
                 [("properties.datetime", pymongo.ASCENDING), ("id", pymongo.ASCENDING)]
             )
 
+            matched = self.item_table.count_documents({"collection": collection_id})
+
             for item in collection_children:
                 response_features.append(
                     self.item_serializer.db_to_stac(item, base_url=base_url)
@@ -120,7 +122,7 @@ def item_collection(
             context_obj = {
                 "returned": count if count <= 10 else limit,
                 "limit": limit,
-                "matched": len(response_features) or None,
+                "matched": matched or None,
             }
 
         return ItemCollection(
@@ -296,6 +298,8 @@ def post_search(
                 .sort(sort_list)
             )
 
+            matched = self.item_table.count_documents(queries)
+
             results = []
             links = []
 
@@ -329,7 +333,7 @@ def post_search(
             context_obj = {
                 "returned": count if count <= 10 else search_request.limit,
                 "limit": search_request.limit,
-                "matched": len(results) or None,
+                "matched": matched or None,
             }
 
         return ItemCollection(