From 788580a64f756c97cfccdbd23bc8eeedc5a270a3 Mon Sep 17 00:00:00 2001
From: Paul <paulmakles@gmail.com>
Date: Tue, 23 Nov 2021 08:57:17 +0000
Subject: [PATCH] fix: broken index migration feat: add new index

---
 src/database/migrations/init.rs    | 11 +++++++++--
 src/database/migrations/scripts.rs | 29 ++++++++++++++++++++++++++---
 2 files changed, 35 insertions(+), 5 deletions(-)

diff --git a/src/database/migrations/init.rs b/src/database/migrations/init.rs
index 2b7c1b73..0570c8f8 100644
--- a/src/database/migrations/init.rs
+++ b/src/database/migrations/init.rs
@@ -108,6 +108,13 @@ pub async fn create_database() {
                         "channel": 1
                     },
                     "name": "channel"
+                },
+                {
+                    "key": {
+                        "channel": 1,
+                        "_id": 1
+                    },
+                    "name": "channel_id_compound"
                 }
             ]
         },
@@ -119,7 +126,7 @@ pub async fn create_database() {
     get_db()
     .run_command(
         doc! {
-            "createIndexes": "messages",
+            "createIndexes": "channel_unreads",
             "indexes": [
                 {
                     "key": {
@@ -144,7 +151,7 @@ pub async fn create_database() {
     get_db()
     .run_command(
         doc! {
-            "createIndexes": "messages",
+            "createIndexes": "server_members",
             "indexes": [
                 {
                     "key": {
diff --git a/src/database/migrations/scripts.rs b/src/database/migrations/scripts.rs
index eeb12d6d..bae05bfd 100644
--- a/src/database/migrations/scripts.rs
+++ b/src/database/migrations/scripts.rs
@@ -11,7 +11,7 @@ struct MigrationInfo {
     revision: i32,
 }
 
-pub const LATEST_REVISION: i32 = 12;
+pub const LATEST_REVISION: i32 = 13;
 
 pub async fn migrate_database() {
     let migrations = get_collection("migrations");
@@ -396,7 +396,7 @@ pub async fn run_migrations(revision: i32) -> i32 {
         get_db()
         .run_command(
             doc! {
-                "createIndexes": "messages",
+                "createIndexes": "channel_unreads",
                 "indexes": [
                     {
                         "key": {
@@ -421,7 +421,7 @@ pub async fn run_migrations(revision: i32) -> i32 {
         get_db()
         .run_command(
             doc! {
-                "createIndexes": "messages",
+                "createIndexes": "server_members",
                 "indexes": [
                     {
                         "key": {
@@ -444,6 +444,29 @@ pub async fn run_migrations(revision: i32) -> i32 {
         .expect("Failed to create server_members index.");
     }
 
+    if revision <= 12 {
+        info!("Running migration [revision 12 / 2021-11-21]: Add indexes to database.");
+
+        get_db()
+        .run_command(
+            doc! {
+                "createIndexes": "messages",
+                "indexes": [
+                    {
+                        "key": {
+                            "channel": 1,
+                            "_id": 1
+                        },
+                        "name": "channel_id_compound"
+                    }
+                ]
+            },
+            None,
+        )
+        .await
+        .expect("Failed to create message index.");
+    }
+
     // Need to migrate fields on attachments, change `user_id`, `object_id`, etc to `parent`.
 
     // Reminder to update LATEST_REVISION when adding new migrations.