New index to enable distinct country searching AT SCALE #2559

globaldothealth · Mar 15, 2022 · aff5275 · aff5275
1 parent d096007
commit aff5275
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/data-serving/scripts/setup-db/migrations/20220315104910-distinct_country_index.js b/data-serving/scripts/setup-db/migrations/20220315104910-distinct_country_index.js
@@ -0,0 +1,37 @@
+const indexes = [
+  {
+    name: 'locationCountry',
+    key: {
+        'location.country': -1,
+    },
+    collation: {
+        locale: 'en_US',
+        strength: 2,
+    },
+  },
+  {
+    name: 'travelHistoryCountry',
+    key: {
+        'travelHistory.travel.location.country': -1,
+    },
+    collation: {
+        locale: 'en_US',
+        strength: 2,
+    },
+  },
+];
+module.exports = {
+  async up(db, client) {
+      await db.command({
+          createIndexes: 'cases',
+          indexes: indexes,
+      });
+  },
+
+  async down(db, client) {
+      await db.command({
+          dropIndexes: 'cases',
+          index: ['locationCountry', 'travelHistoryCountry'],
+      });
+  },
+};