Wrap query for global aggregation

Author: WaVEV

django_mongodb_backend/compiler.py

@@ -38,6 +38,7 @@ def __init__(self, *args, **kwargs):
         self.subqueries = []
         # Atlas search stage.
         self.search_pipeline = []
+        self.wrap_for_global_aggregation = False
 
     def _get_group_alias_column(self, expr, annotation_group_idx):
         """Generate a dummy field for use in the ids fields in $group."""
@@ -234,11 +235,8 @@ def _build_aggregation_pipeline(self, ids, group):
         """Build the aggregation pipeline for grouping."""
         pipeline = []
         if not ids:
-            group["_id"] = None
-            pipeline.append({"$group": group})
-            # It may be a bug, $$NOW has to be called to be reachable in the rest of the pipeline.
-            pipeline.append({"$set": {"__now": "$$NOW"}})
-            pipeline.append({"$unionWith": {"pipeline": [{"$documents": [{}]}]}})
+            pipeline.append({"$group": {"_id": None, **group}})
+            self.wrap_for_global_aggregation = True
         else:
             group["_id"] = ids
             pipeline.append({"$group": group})

django_mongodb_backend/query.py

@@ -56,6 +56,7 @@ def __init__(self, compiler):
         # $lookup stage that encapsulates the pipeline for performing a nested
         # subquery.
         self.subquery_lookup = None
+        self.wrap_for_global_aggregation = compiler.wrap_for_global_aggregation
 
     def __repr__(self):
         return f"<MongoQuery: {self.match_mql!r} ORDER {self.ordering!r}>"
@@ -91,6 +92,22 @@ def get_pipeline(self):
             pipeline.append({"$match": self.match_mql})
         if self.aggregation_pipeline:
             pipeline.extend(self.aggregation_pipeline)
+        if self.wrap_for_global_aggregation:
+            pipeline = [
+                {"$collStats": {}},
+                {
+                    "$lookup": {
+                        "from": self.compiler.collection_name,
+                        "as": "wrapped",
+                        "pipeline": pipeline,
+                    }
+                },
+                {
+                    "$replaceWith": {
+                        "$cond": [{"$eq": ["$wrapped", []]}, {}, {"$first": "$wrapped"}]
+                    }
+                },
+            ]
         if self.project_fields:
             pipeline.append({"$project": self.project_fields})
         if self.combinator_pipeline: