add index to speed up filter-by-candidate queries

benhiller · Oct 25, 2020 · 3fcbaea · 3fcbaea
1 parent b8202b3
commit 3fcbaea
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/migrations/20201024221925_vote_index.js b/migrations/20201024221925_vote_index.js
@@ -0,0 +1,17 @@
+exports.up = function (knex) {
+  return knex.schema.table('vote', function (table) {
+    table.index(
+      ['election_id', 'tabulator_id', 'batch_id', 'record_id', 'contest_id'],
+      'vote_distinct_ballot_index',
+    );
+  });
+};
+
+exports.down = function (knex) {
+  return knex.schema.table('vote', function (table) {
+    table.dropIndex(
+      ['election_id', 'tabulator_id', 'batch_id', 'record_id', 'contest_id'],
+      'vote_distinct_ballot_index',
+    );
+  });
+};
diff --git a/scripts/import_data.js b/scripts/import_data.js
@@ -222,6 +222,7 @@ const importVotes = async (
     'precinct_portion_id',
     'contest_id',
     'party_id',
+    'distinct_ballot',
   ];
   for (const idx of indexes) {
     try {
@@ -324,7 +325,13 @@ const importVotes = async (
   }
 
   for (const idx of indexes) {
-    await knex.raw(`CREATE INDEX vote_${idx}_index ON vote (${idx})`);
+    if (idx === 'distinct_ballot') {
+      await knex.raw(
+        `CREATE INDEX vote_${idx}_index ON vote (election_id, tabulator_id, batch_id, record_id, contest_id)`,
+      );
+    } else {
+      await knex.raw(`CREATE INDEX vote_${idx}_index ON vote (${idx})`);
+    }
   }
 
   await knex.raw('ANALYZE');