Fix double level-up race condition

poketwo · Oct 6, 2024 · 8d8cb36 · 8d8cb36
1 parent 9611837
commit 8d8cb36
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cogs/levels.py b/cogs/levels.py
@@ -94,10 +94,12 @@ async def on_message(self, message):
 
             roles = [message.guild.get_role(x) for x in ROLES[new_level]]
             await message.author.add_roles(*roles)
-            await self.bot.mongo.db.member.update_one(
-                {"_id": {"id": message.author.id, "guild_id": message.guild.id}},
+            u = await self.bot.mongo.db.member.update_one(
+                {"_id": {"id": message.author.id, "guild_id": message.guild.id}, "level": user.get("level", None)},
                 {"$inc": {"level": 1}},
             )
+            if not u.modified_count:
+                return
 
             msg = f"Congratulations {message.author.mention}, you are now level **{new_level}**!"
             for role in roles: