This commit is contained in:
2025-05-19 22:05:38 +02:00
parent 294a57338e
commit 126dfa8b30
2 changed files with 62 additions and 3 deletions

View File

@@ -0,0 +1,59 @@
from typing import Dict, Any
from mongodb_migrations.base import BaseMigration
class Migration(BaseMigration):
def upgrade(self):
index_information_users: Dict[str, Any] = self.db.users.index_information()
index_information_events: Dict[str, Any] = self.db.events.index_information()
index_information_stages: Dict[str, Any] = self.db.stages.index_information()
# Update users collection
if "user_id" in index_information_users:
self.db.users.drop_index("user_id")
if "user_id-guild_id" not in index_information_users:
self.db.users.create_index(["id", "guild_id"], name="user_id-guild_id", unique=True)
# Update events collection
if "guild_id" in index_information_events:
self.db.events.drop_index("guild_id")
if "event_name-guild_id" not in index_information_events:
self.db.events.create_index(["name", "guild_id"], name="event_name-guild_id", unique=False)
# Update stages collection
if "event_id-and-guild_id" in index_information_stages:
self.db.stages.drop_index("event_id-and-guild_id")
if "event_id-guild_id" not in index_information_stages:
self.db.stages.create_index(["event_id", "guild_id"], name="event_id-guild_id", unique=False)
def downgrade(self):
index_information_users: Dict[str, Any] = self.db.users.index_information()
index_information_events: Dict[str, Any] = self.db.events.index_information()
index_information_stages: Dict[str, Any] = self.db.stages.index_information()
# Update users collection
if "user_id-guild_id" in index_information_users:
self.db.users.drop_index("user_id-guild_id")
if "user_id" not in index_information_users:
self.db.users.create_index("id", name="user_id", unique=True)
# Update events collection
if "event_name-guild_id" in index_information_events:
self.db.events.drop_index("event_name-guild_id")
if "guild_id" not in index_information_events:
self.db.events.create_index("guild_id", name="guild_id", unique=False)
# Update stages collection
if "event_id-guild_id" in index_information_stages:
self.db.stages.drop_index("event_id-guild_id")
if "event_id-and-guild_id" not in index_information_stages:
self.db.stages.create_index(
["event_id", "guild_id"], name="event_id-and-guild_id", unique=False
)

View File

@@ -32,7 +32,7 @@ col_stages: AsyncCollection = db.get_collection("stages")
# Update indexes
async def _update_database_indexes() -> None:
await col_users.create_index("id", name="user_id", unique=True)
await col_users.create_index(["id", "guild_id"], name="user_id-guild_id", unique=True)
await col_guilds.create_index("id", name="guild_id", unique=True)
await col_events.create_index("guild_id", name="guild_id", unique=False)
await col_stages.create_index(["event_id", "guild_id"], name="event_id-and-guild_id", unique=False)
await col_events.create_index(["name", "guild_id"], name="event_name-guild_id", unique=False)
await col_stages.create_index(["event_id", "guild_id"], name="event_id-guild_id", unique=False)