Merge pull request #15 from jonhealy1/fix_limit

add limited and count

Author: jonhealy1

stac_fastapi/mongo/stac_fastapi/mongo/core.py

@@ -103,12 +103,19 @@ def item_collection(
         base_url = str(kwargs["request"].base_url)
 
         with self.client.start_session() as session:
-            collection_children = self.item_table.find(
-                {"collection": collection_id}, session=session
-            ).sort(
-                [("properties.datetime", pymongo.ASCENDING), ("id", pymongo.ASCENDING)]
+            collection_children = (
+                self.item_table.find({"collection": collection_id}, session=session)
+                .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 +127,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 +303,8 @@ def post_search(
                 .sort(sort_list)
             )
 
+            matched = self.item_table.count_documents(queries)
+
             results = []
             links = []
 
@@ -329,7 +338,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(