database: add pipeline

I'm too lazy to read docs, if this doesn't work, might as well fallback to regular method of fetch -> update
Itz-fork · Jan 2, 2024 · d22eaa0 · d22eaa0
1 parent 567b887
commit d22eaa0
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/megadl/helpers/database.py b/megadl/helpers/database.py
@@ -42,15 +42,22 @@ async def plus_fl_count(
             await self.mongoc.update_async(
                 self.coll_users,
                 {"_id": user_id},
-                {"$inc": {"total_downloads": downloads}},
+                [
+                    {"$match": {"_id": user_id}},
+                    {"$inc": {"total_downloads": downloads}},
+                    {"$replaceWith": {"$mergeObjects": ["$$ROOT", "$$CURRENT"]}},
+                ],
                 use_given=True,
-                upsert=False,
             )
         elif uploads:
             await self.mongoc.update_async(
                 self.coll_users,
                 {"_id": user_id},
-                {"$inc": {"total_uploads": uploads}},
+                [
+                    {"$match": {"_id": user_id}},
+                    {"$inc": {"total_uploads": uploads}},
+                    {"$replaceWith": {"$mergeObjects": ["$$ROOT", "$$CURRENT"]}},
+                ],
                 use_given=True,
                 upsert=False,
             )