fix(bot): 🐛 Messages not getting flushed to the DB.

This commit is contained in:
Nyeki.py
2024-10-18 20:55:23 -03:00
parent faef6e7a0f
commit aac8f9da9f
+18 -1
View File
@@ -9,6 +9,8 @@ from discord.utils import find
from tortoise import transactions, exceptions
import asyncpg.exceptions
import aiohttp
import orjson
@@ -63,8 +65,23 @@ class Calls(base.Cog):
key=lambda m: m.created_at,
)
while True:
try:
await models.Message.bulk_create(messages, batch_size=500)
await models.MessageContent.bulk_create(message_contents, batch_size=500)
break
except asyncpg.exceptions.DeadlockDetectedError:
continue
while True:
try:
await models.MessageContent.bulk_create(
message_contents, batch_size=500
)
break
except asyncpg.exceptions.DeadlockDetectedError:
continue
print(
f"DONE! Flushed {len(messages):,} messages and {len(message_contents):,} message contents."